21 lines
559 B
Python
21 lines
559 B
Python
from applications.extensions import ma
|
|
from marshmallow import fields
|
|
from applications.models import Dept
|
|
|
|
|
|
# 用户models的序列化类
|
|
class UserSchema(ma.Schema):
|
|
id = fields.Integer()
|
|
username = fields.Str()
|
|
realname = fields.Str()
|
|
enable = fields.Integer()
|
|
create_at = fields.DateTime()
|
|
update_at = fields.DateTime()
|
|
dept = fields.Method("get_dept")
|
|
|
|
def get_dept(self, obj):
|
|
if obj.dept_id != None:
|
|
return Dept.query.filter_by(id=obj.dept_id).first().dept_name
|
|
else:
|
|
return None
|