将__slots__与SQLAlchemy模型一起使用

Question

我在我的新项目中使用SQLAlchemy，并且希望将__slots__用于模型（在没有炼金术的测试版中，__slots__是必要的，因为创建了大量对象）。但我无法将它们与SQLAlchemy声明组合在一起，并为代码获取以下错误：

from sqlalchemy.ext.declarative import declarative_base
Base = declarative_base()

class NotWorking(Base):
     __tablename__ = 'table1'
     pk = Column(Integer, primary_key=True)
     name = Text(length=64, convert_unicode=True)

     __slots__ = ['name', 'pk']

错误

ValueError: 'name' in __slots__ conflicts with class variable

期望以__slots__修改类来为其中定义的字段创建描述符，因此一种解决方法是创建隐藏字段（_name）并使模型字段充当属性，如下所示：

class Working(Base):
    __tablename__ = 'table2'
    pk = Column(Integer, primary_key=True)
    name = Text(length=64, convert_unicode=True)

    __slots__ = ['_name', '_pk']

     # Workaround
    @property
    def name(self):
        return self._name

    @name.setter
    def name(self, name):
        self._name = name

但是这段代码编写起来有点乏味，你需要为每个新字段添加相应的属性。我想知道，有没有更好的方法没有属性，同时仍然使用声明基（不使用经典映射）。

Answer 1

您可以在类主体上使用简单的Python语句，根据__slots__属性本身自动创建简单属性：

class Working(Base):
    ...

    __slots__ = ['_name', '_pk']

    for _tmp_name in __slots__:
        locals()[_tmp_name[1:]] = property(lambda self, name=_tmp_name: getattr(self, name))
    del _tmp_name

请注意，虽然在函数或方法中返回y locals的字典有些特殊，但对它的修改不会影响实际变量，它在类体本身中作为普通字典工作。

如果您有多个模型，或者发现这个太模糊，可以将它放在元类或__init_subclass__方法中（Python 3.6 +）：

class MyBase(Base):
    def __init_subclass__(cls, *args, **kwargs):
        for name in cls.__slots__:
            setattr(cls, name[1:], property(lambda self, name=name: getattr(self, name)))

class Working(MyBase):
    ...

对于元类案例（python＆lt; 3.6），只需在元类__init__中放入相同的三行：

class MetaBase(type):
    def __init__(cls, name, bases, namespace):
         super().__init__(name, bases, namespace)
         for name in cls.__slots__:
             setattr(cls, name[1:], property(lambda self, name=name: getattr(self, name)))


class Working(Base, metaclass=MetaBase):
    ...