我在我的python代码中使用多处理
我的python代码导入了Pysnmp和多处理
我代码运行顺利的一半时间
但不幸的是,有一半时间我的代码不起作用并显示异常“pyasn1.error.pyasn1 error:type tagset”。
我的代码首先创建了一个带有多个线程的“multiprocessing.dummy.Pool(numOfThreads)”。
然后它调用“p.map(sendSNMPGet,[ipRange +'。'+ str(x)for x in range(1,256)])”,它将函数“sendSNMPGet”作为线程函数,并调用函数255次值“1-255”。
这是我的代码:
from pysnmp.entity.rfc3413.oneliner import cmdgen
from pysnmp.proto.rfc1902 import Integer, IpAddress, OctetString
import multiprocessing.dummy
import multiprocessing
def SNMPGet(ip, community, oid, version = 1):
generator = cmdgen.CommandGenerator()
comm_data = cmdgen.CommunityData('server', community, version) # 1 means version SNMP v2c
transport = cmdgen.UdpTransportTarget((ip, 161),timeout=0.5,retries=2)
real_fun = getattr(generator, 'getCmd')
res = (errorIndication, errorStatus, errorIndex, varBinds)\
= real_fun(comm_data, transport, oid)
if (errorStatus == 2): #Is there an EndOfMib() Error
return
elif (errorIndication is None or errorStatus is True):
with open("Maprinter.txt", "a") as myfile: myfile.write("Date is: " + datetime.datetime.now().strftime("%d-%m-%Y %H:%M:%S") + " IP is: " + ip + ", Response: %s\n" % varBinds)
print("IP is: " + ip + ", Response: %s\n" % varBinds)
return
def sendSNMPGet(ip):
return SNMPGet(ip, 'public', '1.3.6.1.2.1.43.5.1.1.1.1', 1)
def snmp_range(ipRange, start, end):
num_threads = 4 * multiprocessing.cpu_count()
p = multiprocessing.dummy.Pool(num_threads)
p.map(sendSNMPGet, [ipRange + '.' + str(x) for x in range(start,end)])
def GetIpsRange(ips):
print('starting scan')
if __name__ == "__main__":
if(ips != ''): #if there are ips in registry
for ip in range(len(ips)):
snmp_range(ips[ip], 1, 256)
print('ending scan')
GetIpsRange('10.0.0')
尽管使用了Pysnmp,我认为我的异常是由于缺乏适当的多处理编码而发生的。 我怎么能改变我的代码来处理这个异常甚至更好,防止它发生?