user.py 2.22 KB
Newer Older
wanli's avatar
wanli committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
from . import BaseSchema
from marshmallow import fields, validate, RAISE, INCLUDE, EXCLUDE

'''
默认情况下,如果在架构中遇到不匹配的键,load将引发a 。ValidationErrorField
可以使用unknown选项修改此行为,该选项接受以下之一:

    RAISE(默认):ValidationError 如果存在未知字段,则引发
    EXCLUDE:排除未知字段
    INCLUDE:接受并包含未知字段

您可以unknown在类的Meta中指定Schema
'''

class AddSchema(BaseSchema):
wanli's avatar
wanli committed
16 17 18
    account = fields.String(required=True)
    username = fields.String(required=False)
    password = fields.String(required=True)
wanli's avatar
wanli committed
19 20 21 22
    email = fields.Email(required=False, default="")
    phone = fields.String(required=False, default="")
    gender = fields.Int(required=False, default=1)
    birthday = fields.Date(required=False, default=None)
wanli's avatar
wanli committed
23
    role = fields.String(required=False)
wanli's avatar
wanli committed
24 25 26 27 28 29 30 31 32 33 34
    remarks = fields.String(required=False)

    class Meta:
        unknown = EXCLUDE

class DeleteSchema(BaseSchema):
    class Meta:
        unknown = EXCLUDE

class UpdateSchema(BaseSchema):
    uuid = fields.UUID(required=False)
wanli's avatar
wanli committed
35 36
    account = fields.String(required=False)
    username = fields.String(required=False)
wanli's avatar
wanli committed
37
    email = fields.Email(required=False)
wanli's avatar
wanli committed
38
    phone = fields.String(required=False)
wanli's avatar
wanli committed
39
    gender = fields.Int(required=False)
wanli's avatar
wanli committed
40 41
    birthday = fields.Date(required=False)
    role = fields.String(required=False)
wanli's avatar
wanli committed
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
    remarks = fields.String(required=False)

    class Meta:
        unknown = EXCLUDE

class QuerySchema(BaseSchema):
    uuid = fields.UUID(required=False)
    scope_type = fields.String(required=False)
    pagenum = fields.Int(required=False)
    pagesize = fields.Int(required=False, max=50) # 防止用户传特别大的数,导致数据库查询阻塞

    class Meta:
        unknown = EXCLUDE

class ResponseSchema(BaseSchema):
    uuid = fields.UUID(required=True)
wanli's avatar
wanli committed
58 59 60
    account = fields.String(required=True)
    username = fields.String(required=True)
    password = fields.String(required=True)
wanli's avatar
wanli committed
61
    email = fields.Email(required=False)
wanli's avatar
wanli committed
62
    phone = fields.String(required=True)
wanli's avatar
wanli committed
63 64 65 66 67 68
    gender = fields.Int(required=True)
    birthday = fields.Date(required=False)
    remarks = fields.String(required=True)

    class Meta:
        unknown = INCLUDE