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

聆雨淋夜

1. #!/usr/bin/env python
   
2. # -*- coding:utf-8 -*-
   
3. from multiprocessing import Process,Pool
   
4. import paramiko
   
5. import time
   
6. 
   
7. IPList=['1.2.3.4','5.6.7.8']
   
8. cmd='ls'
   
9. result_dct={}
   
10. def remote_exec(hostname,command):
    
11.     try:
    
12.         private_key = paramiko.RSAKey.from_private_key_file('/home/test/.ssh/id_rsa')
    
13.         transport = paramiko.Transport((hostname,22))
    
14.         transport.connect(username='test', pkey=private_key)
    
15.         ssh = paramiko.SSHClient()
    
16.         ssh._transport = transport
    
17.         stdin, stdout, stderr = ssh.exec_command(command)
    
18.         result_dct[hostname]=stdout.readlines() if stdout else stderr.readlines()
    
19.         transport.close()
    
20.     except Exception as e:
    
21.         print 'ssh failed'
    
22. 
    
23. if __name__ == '__main__':
    
24.     pool = Pool(5)
    
25.     for ip in IPList:
    
26.         pool.apply_async(func=remote_exec, args=(ip,cmd),)
    
27.     pool.close()
    
28.     pool.join()
    
29.     print result_dct

复制代码

如上是一个多进程执行命令的脚本，定义了全局字典result_dct，在函数中给字典赋值，“result_dct[hostname]=stdout.readlines() if stdout else stderr.readlines()”在脚本结束输出result_dct，发现还是{}  
在函数中加上 “global result_dct”也不行
大神帮忙看看，感谢。

q1208c

回复 1# 聆雨淋夜

进程间是 fork 呀.

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

聆雨淋夜

回复 2# q1208c

改成多线程也是不行

urey_pp

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

聆雨淋夜

回复 4# urey_pp

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

urey_pp

回复 5# 聆雨淋夜

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

urey_pp

回复 5# 聆雨淋夜

比如dbm、pickle、shelve，看你喜欢哪一个了。可以参考我的代码。

1. https://github.com/DingGuodong/LinuxBashShellScriptForOps/blob/f15e45e65dcd919e36f2039b6162caff45291443/functions/net/tcp/ssh/pySshConnectParallel.py

复制代码

urey_pp

我的代码如下：https://github.com/DingGuodong/L ... hConnectParallel.py

q1208c

回复 3# 聆雨淋夜

线程肯定是可以的.

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

除非你的实现有问题.