a0b8ec9b7f
udating the api to be able to return salesorders between a given range, as well as the client/shell to make use of this. Has resulted in a lot of duplicate code, but next patch will go through and clean up a bunch of that, and extract a few validators into either decorators, or at least helper functions. Change-Id: I59ed58c32b8dd4a75babf70df188e965e5e94a16
68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
from decorator import decorator
|
|
import flask
|
|
import itertools
|
|
import json
|
|
from artifice.models import Tenant
|
|
|
|
|
|
def _validate(data, *args, **kwargs):
|
|
for key in itertools.chain(args, kwargs.keys()):
|
|
if not key in data:
|
|
flask.abort(400, json.dumps({'error': 'missing parameter',
|
|
'param': key}))
|
|
for key, val in kwargs.iteritems():
|
|
flask.abort(400, json.dumps({'error': 'validation failed',
|
|
'param': key}))
|
|
|
|
|
|
def must(*args, **kwargs):
|
|
"""
|
|
Asserts that a given set of keys are present in the request parameters.
|
|
Also allows for keyword args to handle validation.
|
|
"""
|
|
def tester(func):
|
|
def funky(*iargs, **ikwargs):
|
|
_validate(flask.request.params, *args, **kwargs)
|
|
return func(*iargs, **ikwargs)
|
|
return decorator(funky, func)
|
|
return tester
|
|
|
|
|
|
@decorator
|
|
def returns_json(func, *args, **kwargs):
|
|
status, content = func(*args, **kwargs)
|
|
response = flask.make_response(
|
|
json.dumps(content), status)
|
|
response.headers['Content-type'] = 'application/json'
|
|
return response
|
|
|
|
|
|
def json_must(*args, **kwargs):
|
|
"""Implements a simple validation system to allow for the required
|
|
keys to be detected on a given callable."""
|
|
def unpack(func):
|
|
def dejson(f, *iargs):
|
|
if flask.request.headers.get('content-type', '') != "application/json":
|
|
flask.abort(400, json.dumps({"error": "must be in JSON format"}))
|
|
# todo -- parse_float was handled specially
|
|
_validate(flask.request.json, *args, **kwargs)
|
|
return func(*iargs)
|
|
return decorator(dejson, func)
|
|
return unpack
|
|
|
|
|
|
def validate_tenant_id(tenant_id, session):
|
|
"""Tenant ID validation that check that the id you passed is valid,
|
|
and that a tenant with this ID exists.
|
|
- returns tenant query, or a tuple if validation failure."""
|
|
if isinstance(tenant_id, unicode):
|
|
tenant_query = session.query(Tenant).\
|
|
filter(Tenant.id == tenant_id)
|
|
if tenant_query.count() == 0:
|
|
return 400, {"errors": ["No tenant matching ID found."]}
|
|
elif tenant_id is not None:
|
|
return 400, {"error": ["tenant must be a unicode string."]}
|
|
else:
|
|
return 400, {"missing parameter": {"tenant": "Tenant id."}}
|
|
return tenant_query[0]
|