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.
37 lines
1.1 KiB
37 lines
1.1 KiB
#!/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 self.required or []: |
|
if not field in json: |
|
missing.append(field) |
|
|
|
if missing: |
|
fields = ', '.join(missing) |
|
error = 'Missing fields: {fields}'.format(fields=fields) |
|
return jsonify(status='ERROR', |
|
error=error) |
|
|
|
return func(*args, **kwargs) |
|
return check_json
|
|
|