base.py 540 B

123456789101112131415161718
  1. from flask import request
  2. from wtforms import Form
  3. from app.libs.error_code import ParameterException
  4. class BaseForm(Form):
  5. def __init__(self, *args, **kwargs):
  6. data = request.get_json(silent=True)
  7. args = request.args.to_dict()
  8. kwargs['csrf_enabled'] = False
  9. super(BaseForm, self).__init__(data=data, **args, **kwargs)
  10. def validate_for_api(self):
  11. valid = super(BaseForm, self).validate()
  12. if not valid:
  13. raise ParameterException(message=self.errors)
  14. return self