Методы или функции для работы с БД через SQLAlchemy?

Question

[domanskiy]

Делаю API на фласке.
Возник вопрос: как лучше работать с БД через ORM SQLAlchemy?
1. Написать функции в роутах
db.query...
db.add(данные)
db.commit

Можно, как вариант, функции вынести в отдельный файл services

2. Написать методы в классах моделей таблиц алхимии.

Оба варианта рабочие.
Но чем методы в классах лучше?

Answer 1

[Ivan Yakushenko]

Общие запросы, которые могут повторяться множество раз в коде пишешь в классе модели, что бы потом не дублировать код, всё остальное по месту, то есть в самих роутах. Выносить в какой то там отдельный файл по 1-2 строчки кода, которые используются 1 раз в 1 месте - такая себе идея.

Comments

[domanskiy]

Если отдельными функциями, то тотже гет можно сделать универсальным... и импортировать в роуты
А вот методы нужно у каждого класса повторять.

Если смотреть документацию, то все в роутах.
Но на просторах ютуба есть варианты решения методами в классах.

Вот ту же авторизацию по jwt, с получением токена, реализуют чаще обычно методами в классах.

Это всё кому как удобнее? Или есть разница?

[domanskiy]

На ютубе Олег Молчанов на фласке приложение реализует. У него функции к роутам.

вариант с реализацией в классах
https://youtube.com/playlist?list=PLWQhUNXl0LnjBIa...

[Ivan Yakushenko]

domanskiy, в общем смысле разницы никакой нет, есть стандарты и паттерны, к которым по хорошему следует прислушиваться. Ни авторы flask ни sqlalchemy вроде нигде не упоминают своей позиции по подобным вопросам, зато, например, авторы pyramid много где используют общие запросы именно как методы класса, тоже самое замечал и в документации django, fastapi и много где в целом, так это логично и вполне себе в стиле python.

[domanskiy]

Ivan Yakushenko, сейчас как раз параллельно пробую переписать проест с фласка на fastapi
Просмотрел документацию. Вроде бы внимательно. Максимум, вынос функций в отдельный файл
https://fastapi.tiangolo.com/tutorial/sql-databases/
В примере это crud.py

Я же имел ввиду, когда добавляют методы в классы sqlalchemy

Как пример:

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

SQLALCHEMY_DATABASE_URL = "postgresql://......"  # для доступа к БД из контейнера

engine = create_engine(SQLALCHEMY_DATABASE_URL)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
Base = declarative_base()

class User(Base):
    __tablename__ = 'user'
    id  = Column(Integer, primary_key=True, unique=True, nullable=False, autoincrement=True)
    descript = Column(String(100), nullable=False)

    @classmethod
    def get_list(cls):
        try:
            users = cls.query.all()
            session.commit()
        except Exception:
            session.rollback()
            raise
        return users

    def save(self):
        try:
            session.add(self)
            session.commit()
        except Exception:
            session.rollback()
            raise


    def __init__(self, *args, **kwargs):
        super(User, self).__init__(*args, **kwargs)
        self.upper_name()

    def upper_name(self):
        if self.name:
            self.name = self.name.upper()

Ну а дальше, при обращении к модели users = models.User.get_list()

[domanskiy]

Разобрался... Но это не точно.
Если перенести функции из роутов в методы моделей, можно обрабатывать исключения каждой сессии. Т.е. можно установить session.rollback() на исключение, что бы ошибка не блокировала транзакции в сессию.
Кажется я правильно понял.

Думаю в FastAPI это стоит реализовать в промежуточном middleware который запускается на каждую сессию