pysnmp twisted

from twisted.internet import reactor
from twisted.internet import defer, task
from pysnmp.entity import engine, config
from pysnmp.carrier.twisted import dispatch
from pysnmp.carrier.twisted.dgram import udp
from pysnmp.entity.rfc3413.twisted import cmdgen


import __webimport__
import tools.config
from tools.database import makedsn
import psycopg2


def cbFun(cbCtx):
    (errorIndication, errorStatus, errorIndex, varBinds) = cbCtx
    for oid, val in varBinds:
        print('%s = %s' % (oid.prettyPrint(), val.prettyPrint()))


def parallel(iterable, count, callable, *args, **named):
    coop = task.Cooperator()
    work = (callable(elem, *args, **named) for elem in iterable)
    return defer.DeferredList([coop.coiterate(work) for i in xrange(count)])


def fetch(host, snmpEngine, getCmdGen):
    id, ip, community, hc = host
    config.addV1System(snmpEngine, 'test-agent', community)
    config.addTargetAddr(
                     snmpEngine, 'myRouter', config.snmpUDPDomain,
                     (ip, 161), 'myParams', timeout=1
    )
    df = getCmdGen.sendReq(snmpEngine, 'myRouter',
                           (('1.3.6.1.2.1.1.1.0', None), ))
    df.addCallback(cbFun)
    return df

dsn = makedsn(tools.config.main_db)
connection = psycopg2.connect(dsn)
cursor = connection.cursor()
cursor.execute("""SELECT e.id, e.ip, e.snmpcomm, e.hccnt
           FROM snmp_ports sp, equipment e
           WHERE e.snmp = 'Y' and sp.equipment = e.id
           GROUP BY e.id,e.ip,e.snmpcomm,e.hccnt
           ORDER BY e.id limit 4000""")
hosts = cursor.fetchall()

snmpEngine = engine.SnmpEngine()
config.addTargetParams(snmpEngine, 'myParams', 'test-agent', 'noAuthNoPriv', 1)
# Transport
config.addSocketTransport(snmpEngine, udp.domainName,
                          udp.UdpTwistedTransport().openClientMode())

finished = parallel(hosts, 100, fetch, snmpEngine, cmdgen.GetCommandGenerator())
finished.addCallback(lambda ign: reactor.stop())
reactor.run()