使用SqlAlchemy无法pickle对象错误

时间:2016-07-13 08:00:08

标签: python sqlalchemy pickle

我正在尝试使用SqlAlchemy将可变对象存储在postgresql数据库中。对象的类大致是:

class Data(Mutable, object):
    @classmethod
    def coerce(cls, key, value):
        return value

    def __init__(self, foo, bar):
        self._foo = foo     # foo is an array of short strings
        self._bar = bar     # bar is a single short string

    def append_to_foo(self, baz):
        self._foo.append(baz)
        self.changed()

    # Various other methods for mutating/accessing foo and bar, which call self.changed()
    # when they finish

列定义是:

data = Column(Data.as_mutable(PickleType))

每当我尝试向包含此列的表中添加行时,都会出现以下错误:

sqlalchemy.exc.StatementError: (builtins.AttributeError) Can't pickle local object 'WeakKeyDictionary.__init__.<locals>.remove' [SQL: "INSERT INTO table (id, user_id, data) VALUES (nextval('data_id_seq'), %(user_id)s, %(data)s) RETURNING data.id"] [parameters: [{'data': <mypkg.foo.Data object at 0x7f79b3b52c88>, 'user_id': 36}]]

我确保Data类本身可以使用Python提示进行pickleable;我可以pickle.dumpspickle.loads个实例没有任何问题。谷歌没有产生任何结果,我找不到任何相关的错误报告。

我正在使用SqlAlchemy 1.0.13和Python 3.5。

1 个答案:

答案 0 :(得分:3)

从阅读"Supporting Pickling"开始,您似乎必须至少为自定义可变类型提供__getstate__方法:

  

此处的开发人员责任只是提供__getstate__方法,该方法从pickle流中排除_parents()集合:

这是因为可变扩展名将weakref.WeakKeyDictionary放在值对象上,而不能进行pickle。文档中提供了最小__getstate__实现:

class Data(Mutable, object):

    ...

    def __getstate__(self):
        d = self.__dict__.copy()
        d.pop('_parents', None)
        return d

根据您的类型实施方式,您可能还必须提供__setstate__