62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
import datetime
|
|
import functools
|
|
from typing import Any, Optional
|
|
|
|
from tortoise import fields, models, timezone
|
|
|
|
try:
|
|
from ciso8601 import parse_datetime
|
|
except ImportError: # pragma: nocoverage
|
|
from iso8601 import parse_date
|
|
|
|
parse_datetime = functools.partial(parse_date, default_timezone=None)
|
|
from tortoise.timezone import get_timezone, localtime
|
|
|
|
|
|
class BaseModel(models.Model):
|
|
id = fields.BigIntField(pk=True, index=True)
|
|
|
|
async def to_dict(self, m2m=False):
|
|
d = {}
|
|
for field in self._meta.db_fields:
|
|
d[field] = getattr(self, field)
|
|
if m2m:
|
|
for field in self._meta.m2m_fields:
|
|
values = await getattr(self, field).all().values()
|
|
d[field] = values
|
|
return d
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
class UUIDModel:
|
|
uuid = fields.UUIDField(unique=True, pk=False)
|
|
|
|
|
|
class CustomDatetimeField(fields.DatetimeField):
|
|
def to_python_value(self, value: Any) -> Optional[datetime.datetime]:
|
|
if value is None:
|
|
value = None
|
|
else:
|
|
if isinstance(value, datetime.datetime):
|
|
value = value.strftime("%Y-%m-%d %H:%M:%S")
|
|
elif isinstance(value, int):
|
|
value = datetime.datetime.fromtimestamp(value)
|
|
value = value.strftime("%Y-%m-%d %H:%M:%S")
|
|
else:
|
|
value = parse_datetime(value)
|
|
if timezone.is_naive(value):
|
|
value = timezone.make_aware(value, get_timezone())
|
|
value = value.strftime("%Y-%m-%d %H:%M:%S")
|
|
else:
|
|
value = localtime(value)
|
|
value = value.strftime("%Y-%m-%d %H:%M:%S")
|
|
self.validate(value)
|
|
return value
|
|
|
|
|
|
class TimestampMixin:
|
|
created_at = CustomDatetimeField(auto_now_add=True)
|
|
updated_at = CustomDatetimeField(auto_now=True)
|