Question

我需要在我的类中使用tzinfo类作为datetime对象。我需要挑选我的课。但是pickle.loads（obj）失败了。 mytz课程有什么问题？如果我不使用mytz课程，一切正常。为什么呢？

from datetime import datetime, tzinfo, timedelta

# Timezone class
class mytz (tzinfo):
    def __init__(self, offset, dst):
        self._tzname = offset + ' ' + dst
        self._offset = timedelta(hours=int(offset[0:3]), minutes=int(offset[0:1]+offset[3:]))
        self._dst = timedelta(hours=int(dst[0:3]), minutes=int(dst[0:1]+dst[3:]))
    def utcoffset(self, dt):
        return self._offset + self.dst(dt)
    def tzname(self, dt):
        return self._tzname
    def dst(self, dt):
        return self._dst

# Root class
class A:
    def __init__(self, val):
        self.val = val
        self.val2 = mytz(val, val)
    def __str__(self):
        return 'response'+str(self.val)
# Sample code
a = A('+0100')

import pickle
apickled = pickle.dumps(a)
print (a)
print (a.val2)
b = pickle.loads(apickled)
print (b)
print(b.val2)

这里是输出：

response+0100
<__main__.mytz object at 0x7fe2873b5748>
Traceback (most recent call last):
  File "/home/maksim/Projects/amodule/ttt.py", line 32, in <module>
    b = pickle.loads(apickled)
TypeError: __init__() missing 2 required positional arguments: 'offset' and 'dst'

Answer 1

我们应该将__getinitargs__()方法添加到我们的自定义类中。并在某处转储我们的论点：

self.args = offset, dst

tzinfo是具有__reduce__方法的抽象类。 __reduce__尝试调用__getinitars__()方法。但总是坚持None值。

class mytz (tzinfo):
    def __init__(self, offset, dst):
        self.args = offset, dst
        self._tzname = offset + ' ' + dst
        self._offset = timedelta(
            hours=int(offset[0:3]), minutes=int(offset[0:1] + offset[3:]))
        self._dst = timedelta(
            hours=int(dst[0:3]), minutes=int(dst[0:1] + dst[3:]))

    def utcoffset(self, dt):
        return self._offset + self.dst(dt)

    def tzname(self, dt):
        return self._tzname

    def dst(self, dt):
        return self._dst

    def __getinitargs__(self):
        return self.args

Python：类实例

1 个答案: