加入marshmallow数据验证

applications/models/__init__.py

@@ -1,5 +1,47 @@
 from flask_sqlalchemy import SQLAlchemy
 from flask_marshmallow import Marshmallow
+from marshmallow import fields
+from marshmallow.validate import (
+    URL, Email, Range, Length, Equal, Regexp,
+    Predicate, NoneOf, OneOf, ContainsOnly
+)
 
+URL.default_message = '无效的链接'
+Email.default_message = '无效的邮箱地址'
+Range.message_min = '不能小于{min}'
+Range.message_max = '不能小于{max}'
+Range.message_all = '不能超过{min}和{max}这个范围'
+Length.message_min = '长度不得小于{min}位'
+Length.message_max = '长度不得大于{max}位'
+Length.message_all = '长度不能超过{min}和{max}这个范围'
+Length.message_equal = '长度必须等于{equal}位'
+Equal.default_message = '必须等于{other}'
+Regexp.default_message = '非法输入'
+Predicate.default_message = '非法输入'
+NoneOf.default_message = '非法输入'
+OneOf.default_message = '无效的选择'
+ContainsOnly.default_message = '一个或多个无效的选择'
+
+fields.Field.default_error_messages = {
+    "required": "缺少必要数据",
+    "null": "数据不能为空",
+    "validator_failed": "非法数据",
+}
+
+fields.Str.default_error_messages = {
+    'invalid': "不是合法文本"
+}
+
+fields.Int.default_error_messages = {
+    "invalid": "不是合法整数"
+}
+
+fields.Number.default_error_messages = {
+    "invalid": "不是合法数字"
+}
+
+fields.Boolean.default_error_messages = {
+    "invalid": "不是合法布尔值"
+}
 db = SQLAlchemy()
-ma = Marshmallow()
+ma = Marshmallow()

applications/models/admin_dept.py

@@ -1,6 +1,6 @@
 import datetime
 from applications.models import db, ma
-from marshmallow import fields
+from marshmallow import fields,validate
 
 
 class Dept(db.Model):
@@ -25,7 +25,7 @@ class DeptSchema(ma.Schema):  # 序列化类
     deptName = fields.Str(attribute="dept_name")
     leader = fields.Str()
     phone = fields.Str()
-    email = fields.Str()
+    email = fields.Str(validate=validate.Email())
     address = fields.Str()
-    status = fields.Str()
-    sort = fields.Str()
+    status = fields.Str(validate=validate.OneOf(["0", "1"]))
+    sort = fields.Integer()