Расширение декларативной базы Flask-SqlAlchemy

Я пробовал следующее, и это, кажется, работает:

class BaseModel(db.Model):
    __abstract__ = True

    row_ver = db.Column(db.Integer, nullable=False)

    @declared_attr
    def __mapper_args__(cls):
        return {'version_id_col': cls.row_ver}

    def to_dict(self):
        res = dict()
        for c in self.__table__.columns:
            value = getattr(self, c.name)
            if isinstance(value, date):
                res[c.name] = value.isoformat()
            elif isinstance(value, uuid.UUID):
                res[c.name] = str(value)
            else:
                res[c.name] = value
        return res

class Account(BaseModel):
    __tablename__ = 'account'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100), nullable=False)

Я не уверен, что это правильный способ расширения декларативного базового класса, то есть класса db.Model. Что-то не так с приведенным выше кодом?

Также связано: возможно ли создать пользовательскую декларативную базу, наследуя от db.Model (которая сама по себе является декларативной базой), что-то вроде ниже:

class Base(db.Model):
    #some code here

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

class Account(BaseModel)
    #...

python flask sqlalchemy flask-sqlalchemy
person synergetic    schedule 27.03.2014    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Чтобы настроить db.Model, объявите надкласс flask_sqlalchemy.Model и передайте его model_class, как показано ниже.

from flask_sqlalchemy import Model, SQLAlchemy


class AppModel(Model):
    row_ver = db.Column(db.Integer, nullable=False)


db = SQLAlchemy(model_class=AppModel)


class Account(db.Model):
    pass


class Thing(db.Model):
    pass

Обратите внимание, что Account и Thing наследуются от db.Model, а не AppModel.

person Panic    schedule 27.01.2021