聆雨淋夜 发表于 2017-08-08 12:15

在函数中修改全局字典失败

本帖最后由 聆雨淋夜 于 2017-08-08 12:20 编辑

#!/usr/bin/env python
# -*- coding:utf-8 -*-
from multiprocessing import Process,Pool
import paramiko
import time

IPList=['1.2.3.4','5.6.7.8']
cmd='ls'
result_dct={}
def remote_exec(hostname,command):
    try:
      private_key = paramiko.RSAKey.from_private_key_file('/home/test/.ssh/id_rsa')
      transport = paramiko.Transport((hostname,22))
      transport.connect(username='test', pkey=private_key)
      ssh = paramiko.SSHClient()
      ssh._transport = transport
      stdin, stdout, stderr = ssh.exec_command(command)
      result_dct=stdout.readlines() if stdout else stderr.readlines()
      transport.close()
    except Exception as e:
      print 'ssh failed'

if __name__ == '__main__':
    pool = Pool(5)
    for ip in IPList:
      pool.apply_async(func=remote_exec, args=(ip,cmd),)
    pool.close()
    pool.join()
    print result_dct
如上是一个多进程执行命令的脚本,定义了全局字典result_dct,在函数中给字典赋值,“result_dct=stdout.readlines() if stdout else stderr.readlines()”在脚本结束输出result_dct,发现还是{}
在函数中加上 “global result_dct”也不行
大神帮忙看看,感谢。

q1208c 发表于 2017-08-08 13:19

回复 1# 聆雨淋夜

进程间是 fork 呀.

所以, 根本不存在共用的东西.
你改成线程试试.

聆雨淋夜 发表于 2017-08-08 14:23

回复 2# q1208c

改成多线程也是不行

urey_pp 发表于 2017-08-08 15:15

大抵原因是字典为可变变量,试试持久化方法,把子进程或子线程的数据写入持久化的对象中,再在主进程中读出来使用。

聆雨淋夜 发表于 2017-08-08 15:34

回复 4# urey_pp

再请教一下,这里持久化对象指的什么呢?比如说?

urey_pp 发表于 2017-08-08 15:38

回复 5# 聆雨淋夜

比如dbm、pickle、shelve,看你喜欢哪一个了。

urey_pp 发表于 2017-08-08 15:39

回复 5# 聆雨淋夜

比如dbm、pickle、shelve,看你喜欢哪一个了。可以参考我的代码。
https://github.com/DingGuodong/LinuxBashShellScriptForOps/blob/f15e45e65dcd919e36f2039b6162caff45291443/functions/net/tcp/ssh/pySshConnectParallel.py

urey_pp 发表于 2017-08-08 15:40

我的代码如下:https://github.com/DingGuodong/LinuxBashShellScriptForOps/blob/f15e45e65dcd919e36f2039b6162caff45291443/functions/net/tcp/ssh/pySshConnectParallel.py

q1208c 发表于 2017-08-10 17:11

回复 3# 聆雨淋夜

线程肯定是可以的.

我就是这个办法来控制数据库连接的.

除非你的实现有问题.
页: [1]
查看完整版本: 在函数中修改全局字典失败