能否在父类实例化后添加子类？SQLAlchemy表类动态继承问询

核心问题 #

能否先定义Foo类，再将Bar类设置为Foo的子类？换而言之，不通过class Bar(Foo):的方式，而是先定义Bar，再通过动态操作让Bar成为Foo的子类，效果和直接继承一致？

SQLAlchemy实际场景背景 #

常规使用SQLAlchemy时，我们会让表类直接继承declarative_base()生成的基类，代码如下：

import sqlalchemy as db
from sqlalchemy.orm import declarative_base

base = declarative_base()

class Table(base):
    __tablename__ = "table"
    id = db.Column(db.Integer(), primary_key=True, autoincrement=True)
    row = db.Column(db.String(50))
    
    def __init__(self, row_stuff):
        self.row = row_stuff

engine = db.create_engine("sqlite:///test.db")
base.metadata.create_all(bind=engine)

但实际需求是将Table作为模板单独存放在template.py中，后续再为其关联base类，期望的代码逻辑大致如下：

from template import Table
import sqlalchemy as db
from sqlalchemy.orm import declarative_base

base = declarative_base()
# 这里需要实现让Table继承base的操作，而非定义时就继承
Table.add_base = base  # 示例伪代码

engine = db.create_engine("sqlite:///test.db")
base.metadata.create_all(bind=engine)

解决方案 #

通用Python类动态继承实现 #

在Python中，类的继承关系可以动态修改，核心有两种方式：

1. 修改__bases__属性：直接替换类的基类元组

   # 假设已定义Foo和Bar
   class Foo:
       pass
   
   class Bar:
       pass
   
   # 让Bar成为Foo的子类
   Bar.__bases__ = (Foo,)
   

   注意：如果涉及自定义元类，直接修改__bases__可能引发元类冲突，此时更推荐用type()构造类。

2. 使用type()重新构造类：type()的参数依次为类名、基类元组、类属性字典

   # 重新构造Bar类，使其继承Foo
   Bar = type("Bar", (Foo,), dict(Bar.__dict__))
   

SQLAlchemy场景的具体实现 #

由于SQLAlchemy的declarative_base使用了自定义元类DeclarativeMeta，直接修改__bases__可能出现元类不兼容问题，推荐用type()重新构造表类，确保元类正确继承：

1. template.py中的模板类：

   # template.py
   import sqlalchemy as db
   
   class Table:
       __tablename__ = "table"
       id = db.Column(db.Integer(), primary_key=True, autoincrement=True)
       row = db.Column(db.String(50))
       
       def __init__(self, row_stuff):
           self.row = row_stuff
   

2. 主文件中关联基类：

   from template import Table
   import sqlalchemy as db
   from sqlalchemy.orm import declarative_base
   
   # 创建SQLAlchemy基类
   base = declarative_base()
   
   # 重新构造Table类，保留原属性方法的同时继承base的元类特性
   Table = type("Table", (base,), {k: v for k, v in Table.__dict__.items() if not k.startswith('__')})
   
   # 正常初始化引擎并创建表
   engine = db.create_engine("sqlite:///test.db")
   base.metadata.create_all(bind=engine)
   

若要直接修改__bases__，需手动触发SQLAlchemy的元类初始化逻辑：

Table.__bases__ = (base,)
# 手动将表注册到基类元数据中
base.metadata._add_table(Table.__tablename__, Table.__module__, Table)

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