You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.0 KiB
36 lines
1.0 KiB
11 years ago
|
#!/usr/bin/env python
|
||
|
|
# -*- encoding: utf-8 -*-
|
||
|
|
|
||
|
|
"""Helper functions."""
|
||
|
|
|
||
|
|
from functools import wraps
|
||
|
|
|
||
|
|
from flask import request
|
||
|
|
from flask import jsonify
|
||
|
|
|
||
|
|
class ForceJSON(object):
|
||
|
|
def __init__(self, required=None):
|
||
|
|
self.required = required or []
|
||
|
|
|
||
|
|
def __call__(self, func):
|
||
|
|
@wraps(func)
|
||
|
|
def check_json(*args, **kwargs):
|
||
|
|
json = request.get_json(force=True, silent=True)
|
||
|
|
if not json:
|
||
|
|
return jsonify(status='ERROR',
|
||
|
|
error='Request MUST be in JSON format'), 400
|
||
|
|
|
||
|
|
# now we have the JSON, let's check if all the fields are here.
|
||
|
|
missing = []
|
||
|
|
for field in required or []:
|
||
|
|
if not field in json:
|
||
|
|
missing.append(field)
|
||
|
|
|
||
|
|
if missing:
|
||
|
|
return jsonify(status='ERROR',
|
||
|
|
error='Missing fields: {fields}'.format(
|
||
|
|
fields=', '.join(missing)))
|
||
|
|
|
||
|
|
return func(*args, **kwargs)
|
||
|
|
return check_json
|