定製 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)
複製代碼