我有一个班级(下):
class InstrumentChange(object):
'''This class acts as the DTO object to send instrument change information from the
client to the server. See InstrumentChangeTransport below
'''
def __init__(self, **kwargs):
self.kwargs = kwargs
self._changed = None
def _method_name(self, text):
return text.replace(' ','_').lower()
def _what_changed(self):
''' Denotes the column that changed on the instrument returning the column_name of what changed.'''
if not self._changed:
self._changed = self._method_name(self.kwargs.pop('What Changed'))
return self._changed
def __getattr__(self, attr):
for key in self.kwargs.iterkeys():
if self._method_name(key) == attr:
return self.kwargs[key]
def __str__(self):
return "Instrument:%s" % self.kwargs
__repr__ = __str__
what_changed = property(_what_changed)
当我运行以下测试时:
def test_that_instrumentchangetransport_is_picklable(self):
test_dict = {'Updated': 'PAllum', 'Description': 'BR/EUR/BRAZIL/11%/26/06/2017/BD',
'Ask Q': 500, 'Bbg': 'On', 'C Bid': 72.0, 'Benchmark': 'NL/USD/KKB/7.000%/03/11/2009/BD',
'ISIN': 'XS0077157575', 'Bid YTM': 0.0, 'Bid Q': 100, 'C Ask': 72.25, 'Ask YTM': 0.0, 'Bid ASW': 0.0,
'Position': 1280000, 'What Changed': 'C Bid', 'Ask ASW': 0.0}
ins_change = InstrumentChangeTransport(**test_dict)
assert isinstance(ins_change, InstrumentChangeTransport)
# Create a mock filesystem object
file = open('testpickle.dat', 'w')
file = Mock()
pickle.dump(ins_change, file)
我明白了:
Traceback (most recent call last):
File "c:\python23\lib\site-packages\nose-0.11.0-py2.3.egg\nose\case.py", line 183, in runTest
self.test(*self.arg)
File "C:\Code\branches\demo\tests\test_framework.py", line 142, in test_that_instrumentchangetransport_is_picklable
pickle.dump(ins_change, file)
File "C:\Python23\Lib\copy_reg.py", line 83, in _reduce_ex
dict = getstate()
TypeError: 'NoneType' object is not callable
我看过pickle文档,但我不太明白。
有什么想法吗?
本
答案 0 :(得分:33)
你的代码有几个小的“侧面”问题:在测试中使用的类名中突然出现'Transport'(它不是你定义的类名),对内置标识符的可疑践踏file
作为一个局部变量(不要这样做 - 它不会在这里受到伤害,但是践踏内置标识符的习惯有一天会导致神秘的错误),Mock
的误用已经注意到,默认使用最慢,最笨重的酸洗协议和文本而不是pickle文件的二进制文件。
然而,正如@coonj所说,其核心是缺乏国家控制。一个“普通”类不需要它(因为self.__dict__
默认情况下在缺少状态控制的类中没有其他特性而被pickle和unpickled) - 但是因为你覆盖了__getattr__
而没有适用于你的班级。您只需要两个非常简单的方法:
def __getstate__(self): return self.__dict__
def __setstate__(self, d): self.__dict__.update(d)
基本上告诉pickle
将您的类视为正常类,将self.__dict__
表示为整个实例状态,尽管存在__getattr__
。
答案 1 :(得分:7)
它失败了,因为找不到对象的__getstate__()
。 Pickle需要这些来确定如何挑选/取消对象。您只需要__getstate__()
和__setstate__()
方法。
请参阅文档中的TextReader示例:http://docs.python.org/library/pickle.html
更新:我刚看了sourceforge page for the Mock module,,我认为你也错误地使用了它。
你正在嘲笑一个文件对象,但是当pickle试图从中读取时,它将不会得到任何回复,这就是getattr()
没有返回的原因。
答案 2 :(得分:2)
file = open('testpickle.dat', 'w')
file = Mock()
你在这里失去对已打开文件的引用。可能有问题吗?