Files
nCovTrack-Faker/utils/model.py
2022-01-26 17:11:24 +08:00

18 lines
462 B
Python

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