如何在SQLAlchemy 2.0中重写抽象类的自增ID列属性？

方法一：用@declared_attr.directive定义抽象类的ID字段 #

将抽象类中固定的__id字段定义改为动态类属性方法，子类可直接重写该方法修改列参数：

from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, declared_attr

class Base(DeclarativeBase):
    pass

class ManagedEntity(Base):
    __abstract__ = True

    @declared_attr.directive
    def __id(cls) -> Mapped[int]:
        return mapped_column("id", primary_key=True, autoincrement=True)

    @declared_attr.directive
    def __tablename__(cls) -> str:
        name: str = cls.__name__[0].lower()
        for char in cls.__name__[1:]:
            if char.isupper():
                name += "_"
            name += char.lower()
        return name

    @property
    def id(self) -> int:
        return self.__id

# 子类重写ID字段，关闭自增
class NonAutoIncrementEntity(ManagedEntity):
    @declared_attr.directive
    def __id(cls) -> Mapped[int]:
        return mapped_column("id", primary_key=True, autoincrement=False)

核心是利用declared_attr的动态特性，子类重写后会完全替换父类的ID列定义，不会触发重复列错误。

方法二：在子类中排除父类的私有ID字段 #

通过__mapper_args__的exclude_properties参数排除父类私有ID字段后，重新定义自己的ID字段：

class NonAutoIncrementEntity(ManagedEntity):
    # 排除父类被名称修饰后的私有ID字段
    __mapper_args__ = {"exclude_properties": ["_ManagedEntity__id"]}
    __id: Mapped[int] = mapped_column("id", primary_key=True, autoincrement=False)

注意要使用Python名称修饰后的字段名（_ManagedEntity__id），因为父类用了双下划线开头的私有字段__id。

方法三：通过钩子方法动态调整ID列参数 #

在抽象类中定义可被子类重写的钩子方法，返回ID列的额外参数，避免完全重写列定义：

class ManagedEntity(Base):
    __abstract__ = True

    @declared_attr.directive
    def __id(cls) -> Mapped[int]:
        id_kwargs = cls.get_id_column_kwargs()
        return mapped_column("id", primary_key=True, **id_kwargs)

    @classmethod
    def get_id_column_kwargs(cls):
        # 默认开启自增
        return {"autoincrement": True}

    # __tablename__和id属性保持不变

# 子类重写钩子方法，关闭自增
class NonAutoIncrementEntity(ManagedEntity):
    @classmethod
    def get_id_column_kwargs(cls):
        return {"autoincrement": False}

这种方式更灵活，适合仅调整列参数（如自增、类型）的场景，无需重复编写完整列定义代码。

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