定制 Django ORM / SQLAlchemy ORM 的 BaseModel

原由

在具体项目中，通常 ORM 框架提供给咱们的 Model 父类没法知足咱们的要求，这时有很大动力来定制属于项目的 Model 父类。好比但愿每一个 Entity 都有 create_time/update_time 这几个属性，或定制其余行为，而不是在每一个 Entity 里面手写对应的重复的代码。

How to do：Django ORM

Django ORM 提供了丰富的功能，所以达到这个目的很简单。只须要在 BaseModel 中提供 abstract 关键字便可。

代码以下：

from django.db import models


class BaseModel(models.Model):
    create_time = models.IntegerField(db_index=True)
    update_time = models.IntegerField(db_index=True)

    class Meta:
        abstract = True


class SomeEntity(BaseModel):
    name = models.CharField(max_length=1000)
    address = models.CharField(max_length=1000, default='')
    info = models.CharField(max_length=2000, default='')
复制代码

How to do: SQLAlchemy ORM

SQLAlchemy 是另外一款使用普遍的 ORM 框架，达到这个目的也十分简单（只不过查文档比较费力 ）。

代码以下：

import sqlalchemy as sa
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

_Base = declarative_base(bind=sa.create_engine())


class BaseModel(_Base):
    __abstract__ = True  # 主要是这一句起做用

    # 定制基本属性
    id = sa.Column(sa.Integer, nullable=False, primary_key=True, unique=True)
    create_time = sa.Column(sa.Integer, nullable=False, index=True)
    update_time = sa.Column(sa.Integer, nullable=False, index=True)
复制代码