如何在SQLAlchemy中为所有表设置公共前缀

时间:2016-07-07 12:11:17

标签: python sqlalchemy

我知道有一种方法可以让SQLAlchemy将常用prefix添加到所有,是否可以为所有表名称添加公共prefix 派生自单declarative_base

1 个答案:

答案 0 :(得分:6)

使用declared_attr和自定义Base的行:

from sqlalchemy.ext.declarative import declared_attr

class PrefixerBase(Base):

    __abstract__ = True

    _the_prefix = 'someprefix_'

    @declared_attr
    def __tablename__(cls):
        return cls._the_prefix + cls.__incomplete_tablename__


class SomeModel(PrefixerBase):

    __incomplete_tablename__ = 'sometable'
    ...

标有__abstract__的类不会创建表或映射器,并且可以充当扩展声明基础。

您还可以使用自定义元类(originally described here)隐藏它:

from sqlalchemy.ext.declarative.api import DeclarativeMeta


class PrefixerMeta(DeclarativeMeta):

    def __init__(cls, name, bases, dict_):
        if '__tablename__' in dict_:
            cls.__tablename__ = dict_['__tablename__'] = \
                'someprefix_' + dict_['__tablename__']

        return super().__init__(name, bases, dict_)

Base = declarative_base(metaclass=PrefixerMeta)


class SomeModel(Base):

    __tablename__ = 'sometable'
    ...