未使用多处理获取多设备连接的输出
我尝试了以下代码:
import multiprocessing as mp
from netmiko import ConnectHandler
import netmiko
import re
from datetime import datetime
def get_ip (input):
return(re.findall(r'(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)', input))
def make_connection (ip, username, password):
device_connect = ConnectHandler(device_type='cisco_nxos', ip=ip, username=username, password=password)
output = device_connect.send_command("sh hostname")
print(output)
device_connect.disconnect()
def get_ips_nexus (file_name):
for line in open(file_name, 'r').readlines():
line = get_ip(line)
for ip in line:
ips_nexus.append(ip)
ips_nexus = []
get_ips_nexus("ips-Copy.txt")
print(ips_nexus)
username = 'x.x.x.x'
password = 'xxxxxxxx'
startTime = datetime.now()
processes = []
for ip in ips_nexus:
print('***********')
print(ip)
print('***********')
p = mp.Process(target=make_connection, args=(ip, username, password))
processes.append(p)
p.start()
print('!!!!!!!!!!!!!!!!!!!!!!!!!')
for p in processes:
print(p)
p.join()
print(datetime.now() - startTime我只得到了print ip,而不是sh hostname的netmiko连接输出。
我只想使用多进程实现多设备连接!
请帮帮忙。
发布于 2019-08-11 05:17:03
您遗漏了if __name__ == "__main__"子句!
将它添加到进程的for循环之上:
if __name__ == "__main__": # add this
for ip in ips_nexus:
print('***********')
print(ip)
print('***********')
p = mp.Process(target=make_connection, args=(ip, username, password))
processes.append(p)
p.start()
print('!!!!!!!!!!!!!!!!!!!!!!!!!')为什么我需要它?
首先:如果您运行自己的代码,python解释器会将其作为"main“文件导入。您可以通过添加print(__name__)来尝试。无论如何,如果您使用multiprocessing启动一个新进程,那么python解释器将再次导入您的代码,并将其作为额外进程的和作为新的skript运行!好吧,如果它再次到达(在你的例子中)这一部分:
p = mp.Process(target=make_connection, args=(ip, username, password)) 它将再次启动一个新的进程,再次导入它,依此类推。最后,您将拥有一个无限循环,它每次都会将您的代码作为一个新进程启动!这就是为什么你必须添加(important) if __name__ == "__main__"子句的原因!通过添加这一点,在第一次调用新进程后,__name__将不再是__main__!
https://stackoverflow.com/questions/57385664
复制相似问题