Vladless
22 lines
692 B
Python
22 lines
692 B
Python
from sqlalchemy.orm import declarative_base
|
|
|
|
|
|
Base = declarative_base()
|
|
|
|
|
|
class DictLikeMixin:
|
|
"""Позволяет обращаться к ORM-объектам как к словарю.
|
|
|
|
Используется legacy-кодом, который мигрировал с dict-результатов asyncpg
|
|
на ORM и не хочет переписывать все `row["field"]` / `row.get("field")`.
|
|
"""
|
|
|
|
def __getitem__(self, key):
|
|
return getattr(self, key)
|
|
|
|
def get(self, key, default=None):
|
|
return getattr(self, key, default)
|
|
|
|
def to_dict(self):
|
|
return {column.name: getattr(self, column.name) for column in self.__table__.columns}
|