I would like to use class decorators (not function decorators!), eg
def class_decorator(cls):
class new_cls(cls):
def run(self, *args, **kwargs):
print 'In decorator'
super(new_cls,self).run(*args, **kwargs)
return new_cls
@class_decorator
class cls(object):
'''
a class
'''
def run(self):
print 'called'
and to be able to pickle the objects:
import pickle
a = cls()
a.run()
s = pickle.dumps(a)
However pickle returns an error:
PicklingError: Can't pickle <class '__main__.new_cls'>: it's not found as __main__.new_cls
Any help will be appreciated!
When you pickle a class, the name of the class -- not its value -- is pickled . If the class_decorator
returns a new class whose name is not defined at the top level of the module, then you get the error:
PicklingError: Can't pickle <class '__main__.new_cls'>: it's not found as __main__.new_cls
You can avoid the error by naming the new decorated class the same as the undecorated class:
new_cls.__name__ = cls.__name__
Then the code runs without error:
import pickle
def class_decorator(cls):
class new_cls(cls):
def run(self, *args, **kwargs):
print 'In decorator'
super(new_cls,self).run(*args, **kwargs)
new_cls.__name__ = cls.__name__
return new_cls
@class_decorator
class cls(object):
def run(self):
print 'called'
a = cls()
print(a)
# <__main__.cls object at 0x7f57d3743650>
a.run()
# In decorator
# called
s = pickle.dumps(a)
# Note "cls" in the `repr(s)` below refers to the name of the class. This is
# what `pickle.loads` is using to unpickle the string
print(repr(s))
# 'ccopy_reg\n_reconstructor\np0\n(c__main__\ncls\np1\nc__builtin__\nobject\np2\nNtp3\nRp4\n.'
b = pickle.loads(s)
print(b)
# <__main__.cls object at 0x7f57d3743690>
b.run()
# In decorator
# called
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.