feat: faker data: identification
This commit is contained in:
17
utils/model.py
Normal file
17
utils/model.py
Normal 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
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user