Как в Sqlalchemy сделать разные insert в зависимости от типа юзера?

Question

[pcdesign]

Переписываю один проект.
И там такая ситуация.
Юзеры вставляют в таблицу данные вот так:

insert into hange (date,person,from_cur,from_sum,to_cur,to_sum,comment) values (now(),?,?,?,?,?,?)

А администраторы вот так:

insert into hange (date,person,from_cur,from_sum,to_cur,to_sum,course,fee_cur,fee,fee_percent,comment) values (?,?,?,?,?,?,?,?,?,?,?)

То есть админы вставляют больше данных, чем обычные юзеры.

Вопрос как быть с __init__ в описании модели?

class Hange(db.Model):
    # Таблицы и поля. 
    
    def __init__ # Вот каким он должен быть?

Answer 1

[zsts]

1. Как вариант - отнаследовать от Hange два класса, для юзеров и для админов.

2. Если по вашему, то в __init__ должен быть анализ: от какого юзера создается обьект (это уж вы должны решить, как определять: через параметр или через app_context или через request). А также парсинг args/kwargs.
Если юзер - админ, из args/kwargs достаются course,fee_cur (и т.е.) и обьект инициализируется с этими значениями. Если их нет в args/kwargs - поднимаем ошибку.

Comments

[pcdesign]

Спасибо за ответ. А как вам вариант сделать все как будто для админа. А когда юзеры будут добавлять, то лишние поля передавать через None?

[zsts]

pcdesign: Это было бы проще всего :)

Если наследовать классы моделей declarative_base, то можно сделать, например, так:

from sqlalchemy.ext.declarative import declarative_base
Base = declarative_base()

class Hange(db.Model):
    date = Column(DateTime, default=datetime.now)
    person = ...
    ...
    course = Column(String) # default=None cтоит по-умолчанию
    fee_cur = Column(String)
    fee = Column(String)
    fee_percent = Column(String)

    def __init__():

[zsts]

zsts: При создании обьекта класса Hange поля course, fee_cur, fee, fee_percent получат дефолтные значения (в данном случае, None).