加入marshmallow数据验证

2021-06-07 22:57:21 +08:00
parent 375d3a10b2
commit f93a36c094
6 changed files with 72 additions and 9 deletions
@@ -1,6 +1,20 @@
 # xss过滤
+from flask import abort, make_response, jsonify
+
+
 def xss_escape(s: str):
    if s is None:
        return None
    else:
-        return s.replace("&", "&amp;").replace(">", "&gt;").replace("<", "&lt;").replace("'", "&#39;").replace('"', "&#34;")
+        return s.replace("&", "&amp;").replace(">", "&gt;").replace("<", "&lt;").replace("'", "&#39;").replace('"',
+                                                                                                               "&#34;")
+
+
+def check_data(schema, data):
+    errors = schema.validate(data)
+    for k, v in errors.items():
+        for i in v:
+            # print("{}{}".format(k, i))
+            msg = "{}{}".format(k, i)
+    if errors:
+        abort(make_response(jsonify(result=False, msg=msg), 200))