mirror of
https://github.com/mealie-recipes/mealie.git
synced 2026-02-26 01:33:12 -05:00
39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
from datetime import datetime
|
|
|
|
from sqlalchemy import Integer
|
|
from sqlalchemy.orm import DeclarativeBase, Mapped, declared_attr, mapped_column, synonym
|
|
from text_unidecode import unidecode
|
|
|
|
from ._model_utils.datetime import NaiveDateTime, get_utc_now
|
|
|
|
|
|
class SqlAlchemyBase(DeclarativeBase):
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
created_at: Mapped[datetime | None] = mapped_column(NaiveDateTime, default=get_utc_now, index=True)
|
|
update_at: Mapped[datetime | None] = mapped_column(NaiveDateTime, default=get_utc_now, onupdate=get_utc_now)
|
|
|
|
@declared_attr
|
|
def updated_at(cls) -> Mapped[datetime | None]:
|
|
return synonym("update_at")
|
|
|
|
@classmethod
|
|
def normalize(cls, val: str) -> str:
|
|
# We cap the length to 255 to prevent indexes from being too long; see:
|
|
# https://www.postgresql.org/docs/current/btree.html
|
|
return unidecode(val).lower().strip()[:255]
|
|
|
|
|
|
class BaseMixins:
|
|
"""
|
|
`self.update` method which directly passing arguments to the `__init__`
|
|
"""
|
|
|
|
def update(self, *args, **kwargs):
|
|
self.__init__(*args, **kwargs)
|
|
|
|
# sqlalchemy doesn't like this method to remove all instances of a 1:many relationship,
|
|
# so we explicitly check for that here
|
|
for k, v in kwargs.items():
|
|
if hasattr(self, k) and v == []:
|
|
setattr(self, k, v)
|