feat: faker data: identification

This commit is contained in:
Zhaolong
2022-01-26 17:11:24 +08:00
parent da9965e2ae
commit 1e3d584870
10 changed files with 206 additions and 28 deletions

17
utils/model.py Normal file
View File

@@ -0,0 +1,17 @@
from base import db
def model2dict(row: db.Model):
return {c.name: str(getattr(row, c.name)) for c in row.__table__.columns}
def models2dicts(rows: list[db.Model]):
return [model2dict(row) for row in rows]
def convert2dict(data: object) -> object:
if isinstance(data, list) and len(data) > 0 and isinstance(data[0], db.Model):
data = models2dicts(data)
if isinstance(data, db.Model):
data = model2dict(data)
return data

View File

@@ -1,4 +1,4 @@
from __future__ import annotations
from utils.model import convert2dict
def response(resp: dict, status: int, headers=None):
@@ -6,6 +6,7 @@ def response(resp: dict, status: int, headers=None):
def success(code: int, msg: str, data: object):
data = convert2dict(data)
resp = {
"code": code,
"msg": msg,