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修饰符定义。