diff options
author | elie <elie> | 2012-08-27 09:18:06 +0000 |
---|---|---|
committer | elie <elie> | 2012-08-27 09:18:06 +0000 |
commit | cec445554d3c962f6402742d18bccb7799f106e3 (patch) | |
tree | 9a3f29edfd1c90bac8e87b6fe0499eaad284aae7 | |
parent | cdd2190728784615183010144d81ab39d41e5a1b (diff) | |
download | pysnmp-cec445554d3c962f6402742d18bccb7799f106e3.tar.gz |
minor cleanups
-rw-r--r-- | pysnmp/entity/rfc3413/oneliner/target.py | 50 |
1 files changed, 18 insertions, 32 deletions
diff --git a/pysnmp/entity/rfc3413/oneliner/target.py b/pysnmp/entity/rfc3413/oneliner/target.py index 0b9279a..08e84e3 100644 --- a/pysnmp/entity/rfc3413/oneliner/target.py +++ b/pysnmp/entity/rfc3413/oneliner/target.py @@ -6,61 +6,47 @@ from pyasn1.compat.octets import null class _AbstractTransportTarget: transportDomain = protoTransport = None def __init__(self, transportAddr, timeout=1, retries=5, tagList=null): - self.transportAddr = transportAddr + self.transportAddr = self._resolveAddr(transportAddr) self.timeout = timeout self.retries = retries self.tagList = tagList - def __repr__(self): return '%s(%r, %r, %r, %r)' % ( - self.__class__.__name__, self.transportAddr, - self.timeout, self.retries, self.tagList + def __repr__(self): + return '%s(%r, timeout=%r, retries=%r, tagList=%r)' % ( + self.__class__.__name__, self.transportAddr, + self.timeout, self.retries, self.tagList ) - def __hash__(self): return hash(self.transportAddr) - - def __eq__(self, other): return self.transportAddr == other - def __ne__(self, other): return self.transportAddr != other - def __lt__(self, other): return self.transportAddr < other - def __le__(self, other): return self.transportAddr <= other - def __gt__(self, other): return self.transportAddr > other - def __ge__(self, other): return self.transportAddr >= other - def openClientMode(self): self.transport = self.protoTransport().openClientMode() return self.transport - + class UdpTransportTarget(_AbstractTransportTarget): transportDomain = udp.domainName protoTransport = udp.UdpSocketTransport - def __init__(self, transportAddr, timeout=1, retries=5, tagList=null): - _AbstractTransportTarget.__init__(self, transportAddr, timeout, - retries, tagList) + def _resolveAddr(self, transportAddr): try: - self.transportAddr = socket.getaddrinfo(transportAddr[0], - transportAddr[1], - socket.AF_INET, - socket.SOCK_DGRAM, - socket.IPPROTO_UDP)[0][4][:2] + return socket.getaddrinfo(transportAddr[0], + transportAddr[1], + socket.AF_INET, + socket.SOCK_DGRAM, + socket.IPPROTO_UDP)[0][4][:2] except socket.gaierror: raise error.PySnmpError('Bad IPv4/UDP transport address %s: %s' % ('@'.join([ str(x) for x in transportAddr ]), sys.exc_info()[1])) class Udp6TransportTarget(_AbstractTransportTarget): transportDomain = udp6.domainName protoTransport = udp6.Udp6SocketTransport - def __init__(self, transportAddr, timeout=1, retries=5, tagList=null): - _AbstractTransportTarget.__init__(self, transportAddr, timeout, - retries, tagList) + def _resolveAddr(self, transportAddr): try: - self.transportAddr = socket.getaddrinfo(transportAddr[0], - transportAddr[1], - socket.AF_INET6, - socket.SOCK_DGRAM, - socket.IPPROTO_UDP)[0][4][:2] + return socket.getaddrinfo(transportAddr[0], + transportAddr[1], + socket.AF_INET6, + socket.SOCK_DGRAM, + socket.IPPROTO_UDP)[0][4][:2] except socket.gaierror: raise error.PySnmpError('Bad IPv6/UDP transport address %s: %s' % ('@'.join([ str(x) for x in transportAddr ]), sys.exc_info()[1])) class UnixTransportTarget(_AbstractTransportTarget): transportDomain = unix.domainName protoTransport = unix.UnixSocketTransport - - |