当使用python multiprocess.pool.map_async()时,无法pickle<type'thread.lock'>

2024-03-29 00:12:41 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试对类方法使用map_async,得到以下错误: PicklingError: Can't pickle <type 'thread.lock'>: attribute lookup thread.lock failed 我的代码:

def _pickle_method(method):
    func_name = method.im_func.__name__
    obj = method.im_self
    cls = method.im_class
    cls_name = ''
    if func_name.startswith('__') and not func_name.endswith('__'):
        cls_name = cls.__name__.lstrip('_')
    if cls_name:
        func_name = '_' + cls_name + func_name
    return _unpickle_method, (func_name, obj, cls)

def _unpickle_method(func_name, obj, cls):
    for cls in cls.mro():
        try:
            func = cls.__dict__[func_name]
        except KeyError:
            pass
        else:
            break
    return func.__get__(obj, cls)

copy_reg.pickle(types.MethodType, _pickle_method, _unpickle_method)

class MyClass(object):
    def Submit(self,cmd):
        subprocess.call(cmd, shell=True)

    def RunTest(self):
        cmds = []
        for i in range(50):
            cmd = CreateCmd(self)
            cmds.append(cmd)
        self.pool.map_async(self.Submit, cmds) 

    def Main(self):
        self.pool = mp.pool
        while True:
           RunTest(self)

if __name__ == "__main__":
    MyClass()

Submit在类之外时,它可以工作,但像这样我会得到错误。 另外,MyClass还有一些我没有编写的方法和属性,其中一个是记录器,这可能是问题所在吗?


Tags: nameselfcmdobjifdefmyclassmethod
1条回答
网友
1楼 · 发布于 2024-03-29 00:12:41

所以我用一些替代导入构建了您的代码,特别是dill,而不是pickle。我还使用了一个名为pathos.multiprocessingmultiprocessing叉,它使用dill。我可以腌制你的类方法和绑定方法。我忽略了你教copy_reg如何pickle模块的整个部分,因为dill已经可以做到了。

我不得不对你的代码做一些修改,因为它不起作用。我还得做一个CreateCmd函数,因为你没有给它。同样,这段代码本身也会启动多处理作业…但你永远不会得到结果,因为你没有要求它们。你到底想干什么?

不管怎么说,这里有一些代码和你的一样,但是很有用。它仍然没有给你任何有价值的结果,除了显示它会泡菜和代码运行。请发布可以运行的代码,并将抛出您正在报告的错误。

>>> import dill as pickle
>>> import subprocess 
>>> from pathos.multiprocessing import ProcessingPool as Pool
>>> 
>>> def CreateCmd(cmd):
...   return 'sleep {0}'.format(cmd)
>>>
>>> class MyClass(object):
...   def Submit(self, cmd):
...     subprocess.call(cmd, shell=True)
...   def RunTest(self):
...     cmds = []
...     for i in range(50):
...       cmd = CreateCmd(i)
...       cmds.append(cmd)
...     self.pool.amap(self.Submit, cmds) # equivalent to map_async
...   def Main(self): 
...     self.pool = Pool()
...     self.RunTest()
... 
>>> pickle.loads(pickle.dumps(MyClass))
<class '__main__.MyClass'>
>>> pickle.loads(pickle.dumps(MyClass.RunTest))
<unbound method MyClass.RunTest>
>>> x = MyClass()
>>> pickle.loads(pickle.dumps(x.RunTest))
<bound method MyClass.RunTest of <__main__.MyClass object at 0x10d015b10>>
>>> x.Main()
>>> x.Submit('sleep 1')
>>> # use get to get the result… so 'sleep' is felt by the script 
>>> res = x.pool.amap(x.Submit, (CreateCmd(i) for i in range(10)))
>>> res.get()
[None, None, None, None, None, None, None, None, None, None]

不管怎样,如果您想要dillpathos,您可以在https://github.com/uqfoundation找到它们

顺便说一句,如果你想撬开一个线程锁,你也可以这么做。

>>> import dill as pickle
>>> import threading
>>> lock = threading.Lock()
>>> 
>>> pickle.loads(pickle.dumps(lock))
<thread.lock object at 0x10c534650>

相关问题 更多 >