创建SQLAlchemy的ORM类的基类(二)


要为SQLAlchemy的ORM类的基类还有一种方法,就是利用继承的方法来配置。

rom sqlalchemy.ext.declarative import declared_attr
from sqlalchemy.ext.declarative import AbstractConcreteBase

engine = create_engine(...)
Base = declarative_base()

class ModelBase(AbstractConcreteBase,Base):
    id=db.Column(db.String(32), default=generate_uuid, primary_key=True)
    @declared_attr
    def __mapper_args__(cls):
        return {'polymorphic_identity': cls.__name__.lower(),
                'concrete': True} if cls.__name__ != "ModelBase" else {}
    
    

上述实现上就是利用 Concrete Table Inheritance的继承机制来自动为继承类增加__mapper_args__参数.

如果你是在Flask-SQLAlchemy中使用,只要将上述的Base替换成db.Model即可。

相比起另外一种方法来说,该方法更加清晰明了。





你可能感兴趣的:(创建SQLAlchemy的ORM类的基类(二))