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

SQLAlchemy功能非常强大,一般来说我们可以采用下面方法建立一个ORM类.

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

engine = create_engine('sqlite:///:memory:', echo=True)
Session = sessionmaker(bind=engine)
Base = declarative_base()
class User(Base):
      __tablename__ = 'users'
     id = Column(Integer, primary_key=True)
engine .create_all()


在上面例中,我们要建立的ORM类均要基于Base,如果我们要创建一个自己的ORM基类,应该这样:

class BaseModelMixin(object):
    """ 为所有Model提供公用方法 """

    @declared_attr
    def id(self):
        return  Column(String(32),primary_key=True,default=uuid.uuid4().hex)#唯一性的UUID

    def test(self):
        pass

ModelBase=DeclarativeBase(cls=BaseModelMixin)

上述例子中,我们为所有ORM类提供了一个基类,所有基于ModelBase的ORM model均具有一个uuid id字段和一个test方法。

上述例子要点是:

 1、为DeclarativeBase提供一个cls参数,该参数的成员会被ModelBase继承或mixin.

2、为基类增加Column不能直接定义,只能通过@declared_attr修饰符定义。






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