You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.8 KiB
62 lines
1.8 KiB
import asyncio
|
|
from datetime import datetime
|
|
|
|
from tortoise import fields, models
|
|
|
|
from app.settings import settings
|
|
|
|
|
|
class BaseModel(models.Model):
|
|
id = fields.BigIntField(pk=True, index=True)
|
|
|
|
async def to_dict(self, m2m: bool = False, exclude_fields: list[str] | None = None):
|
|
if exclude_fields is None:
|
|
exclude_fields = []
|
|
|
|
d = {}
|
|
for field in self._meta.db_fields:
|
|
if field not in exclude_fields:
|
|
value = getattr(self, field)
|
|
if isinstance(value, datetime):
|
|
value = value.strftime(settings.DATETIME_FORMAT)
|
|
d[field] = value
|
|
|
|
if m2m:
|
|
tasks = [
|
|
self.__fetch_m2m_field(field, exclude_fields)
|
|
for field in self._meta.m2m_fields
|
|
if field not in exclude_fields
|
|
]
|
|
results = await asyncio.gather(*tasks)
|
|
for field, values in results:
|
|
d[field] = values
|
|
|
|
return d
|
|
|
|
async def __fetch_m2m_field(self, field, exclude_fields):
|
|
values = await getattr(self, field).all().values()
|
|
formatted_values = []
|
|
|
|
for value in values:
|
|
formatted_value = {}
|
|
for k, v in value.items():
|
|
if k not in exclude_fields:
|
|
if isinstance(v, datetime):
|
|
formatted_value[k] = v.strftime(settings.DATETIME_FORMAT)
|
|
else:
|
|
formatted_value[k] = v
|
|
formatted_values.append(formatted_value)
|
|
|
|
return field, formatted_values
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
class UUIDModel:
|
|
uuid = fields.UUIDField(unique=True, pk=False, index=True)
|
|
|
|
|
|
class TimestampMixin:
|
|
created_at = fields.DatetimeField(auto_now_add=True, index=True)
|
|
updated_at = fields.DatetimeField(auto_now=True, index=True)
|
|
|