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