python:如何使用多个init args pickleable创建自定义异常类

How to make a custom exception class with multiple init args pickleable

为什么我下面的自定义异常类不能使用pickle模块正确地序列化/反序列化?

1
2
3
4
5
6
7
8
9
10
11
12
13

import pickle

class MyException(Exception):
def __init__(self, arg1, arg2):
self.arg1 = arg1
self.arg2 = arg2

super(MyException, self).__init__(arg1)

e = MyException("foo","bar")

str = pickle.dumps(e)
obj = pickle.loads(str)

此代码抛出以下错误:

1
2
3
4
5
6
7
8
9
10

Traceback (most recent call last):
File"test.py", line 13, in <module>
obj = pickle.loads(str)
File"/usr/lib/python2.7/pickle.py", line 1382, in loads
return Unpickler(file).load()
File"/usr/lib/python2.7/pickle.py", line 858, in load
dispatch[key](self)
File"/usr/lib/python2.7/pickle.py", line 1133, in load_reduce
value = func(*args)
TypeError: __init__() takes exactly 3 arguments (2 given)

我确信这个问题的根源在于我对如何使课堂变得对腌菜友好缺乏知识。有趣的是，当我的类没有扩展Exception时，这个问题不会发生。

谢谢你的帮助。凯尔

编辑:修正我的呼叫超级每shx2编辑:清理标题/内容

如果您同时使用两个参数来构造传递给父异常类的错误消息，则当前答案将被分解。我认为最好的方法是在异常中简单地覆盖__reduce__方法。__reduce__方法应该返回一个两项元组。元组中的第一项是类。第二项是一个元组，包含传递给类的__init__方法的参数。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

import pickle

class MyException(Exception):
def __init__(self, arg1, arg2):
self.arg1 = arg1
self.arg2 = arg2

super(MyException, self).__init__('arg1: {}, arg2: {}'.format(arg1, arg2))

def __reduce__(self):
return (MyException, (self.arg1, self.arg2))

original = MyException('foo', 'bar')
print repr(original)
print original.arg1
print original.arg2

reconstituted = pickle.loads(pickle.dumps(original))
print repr(reconstituted)
print reconstituted.arg1
print reconstituted.arg2

更多关于__reduce__的信息请点击这里。