如何创建SQLAlchemy抽象基类供所有数据表继承复用属性方法

需求概述 #

在SQLAlchemy项目中，希望所有数据表类都继承自一个包含通用属性与方法的抽象类，同时满足两个核心要求：

• 能通过Base.metadata.create_all(engine)正常创建所有数据表
• 能直接调用数据表类上的通用方法（如Users.some_func()）

期望的伪代码示例 #

Base = declarative_base()
# 伪代码
class Table(ABC, Base):
    # 类似@abstractattribute
    some_attribute = list()
    @staticmethod
    def some_func(self):
        pass

class Users(Table):
    __tablename__ = "users"
    user_id = Column(Integer, primary_key=True)
    username = Column(String, nullable=False)
    some_attribute = list()
    @staticmethod
    def some_func():
        do_something()

现存问题 #

上述伪代码无法直接运行，核心问题包括：

• 同时继承ABC与SQLAlchemy的Base存在元类兼容问题
• 缺少抽象属性的强制约束（子类未实现时无报错）
• 抽象基类不能直接包含__tablename__和主键列，否则会被当作真实数据表处理

尝试的装饰器思路框架 #

class Table(ABC):
    some_attribute=None
    @staticmethod
    def some_func(self):
        pass

# 创建装饰器
def sql_table():
    def decorator(abstract_class):
        class SQLTable(Base): # 如何正确命名该类？
            __tablename__ = abstract_class.__dict__["__tablename__"]
            some_attribute = abstract_class.__dict__["some_attribute"]
            for name, obj in abstract_class.__dict__.items():
                if isinstance(obj, Column):
                    locals()[name] = obj
            # 如何获取some_func方法？
@sql_table
class Users(Table):
    __tablename__ = "users"
    user_id = Column(Integer, primary_key=True)
    username = Column(String, nullable=False)
    some_attribute = "some_val"
    @staticmethod
    def some_func():
        do_something()

解决方案 #

方案1：基于ABC与SQLAlchemy基类的合法多重继承 #

Python 3.6+支持元类合并，可直接将ABC与SQLAlchemy的Base结合作为抽象基类，通过抽象属性/方法强制子类实现必要内容：

from abc import ABC, abstractmethod, abstractattribute
from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class AbstractTable(ABC, Base):
    # 强制子类实现表名
    @abstractattribute
    def __tablename__(self):
        pass
    
    # 强制子类实现通用属性
    @abstractattribute
    def some_attribute(self):
        pass
    
    # 强制子类实现通用方法
    @staticmethod
    @abstractmethod
    def some_func():
        pass

class Users(AbstractTable):
    __tablename__ = "users"
    user_id = Column(Integer, primary_key=True)
    username = Column(String, nullable=False)
    
    # 实现抽象属性
    some_attribute = ["user_specific_data"]
    
    @staticmethod
    def some_func():
        print("执行用户专属逻辑")

# 正常创建表
# Base.metadata.create_all(engine)

# 调用类方法
Users.some_func()

方案2：改进装饰器实现 #

通过动态创建SQLAlchemy表类的方式，复用抽象类的属性与方法，避免多重继承的元类问题：

from abc import ABC
from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class AbstractTable(ABC):
    some_attribute = None
    
    @staticmethod
    def some_func():
        pass

def sql_table(cls):
    # 动态创建继承自Base的新类，保留原类名称
    new_table_cls = type(cls.__name__, (Base,), {})
    # 复制原类的所有非特殊属性与方法
    for attr_name, attr_value in cls.__dict__.items():
        if not attr_name.startswith('__'):
            setattr(new_table_cls, attr_name, attr_value)
    return new_table_cls

@sql_table
class Users(AbstractTable):
    __tablename__ = "users"
    user_id = Column(Integer, primary_key=True)
    username = Column(String, nullable=False)
    
    some_attribute = "user_custom_value"
    
    @staticmethod
    def some_func():
        print("执行用户类方法")

# 正常创建表
# Base.metadata.create_all(engine)

# 调用类方法
Users.some_func()

方案对比 #

• 方案1：符合面向对象抽象设计，强制子类实现约束，适合需要严格规范的项目
• 方案2：避免多重继承的潜在问题，实现更灵活，适合快速复用通用逻辑的场景

内容的提问来源于stack exchange，提问作者CodingTil