ip_to_geo : reset connection after 50 requests
This commit is contained in:
parent
b2f37d83f4
commit
9f9bfebe7a
|
@ -46,11 +46,14 @@ class IPToGeo(object):
|
||||||
5 : 'Unsupported IP version',
|
5 : 'Unsupported IP version',
|
||||||
6 : 'IP not found'}
|
6 : 'IP not found'}
|
||||||
|
|
||||||
|
MAX_REQUESTS = 50
|
||||||
|
|
||||||
def __init__(self, remote_addr='127.0.0.1', remote_port=53333, timeout=None, family=socket.AF_INET):
|
def __init__(self, remote_addr='127.0.0.1', remote_port=53333, timeout=None, family=socket.AF_INET):
|
||||||
self._remote_addr = remote_addr
|
self._remote_addr = remote_addr
|
||||||
self._remote_port = remote_port
|
self._remote_port = remote_port
|
||||||
self._timeout = timeout
|
self._timeout = timeout
|
||||||
self._family = family
|
self._family = family
|
||||||
|
self._nb_requests_sent = 0
|
||||||
|
|
||||||
self._create_socket()
|
self._create_socket()
|
||||||
|
|
||||||
|
@ -109,6 +112,19 @@ class IPToGeo(object):
|
||||||
|
|
||||||
return (ip_res, '%c%c%c%c' % (cc0, cc1, cc2, cc3))
|
return (ip_res, '%c%c%c%c' % (cc0, cc1, cc2, cc3))
|
||||||
|
|
||||||
|
def _send_request(self, packet):
|
||||||
|
self._nb_requests_sent += 1
|
||||||
|
if self._nb_requests_sent == self.MAX_REQUESTS:
|
||||||
|
self.close()
|
||||||
|
self._create_socket()
|
||||||
|
self._nb_requests_sent = 0
|
||||||
|
try:
|
||||||
|
self._socket.send(packet)
|
||||||
|
except IOError, e:
|
||||||
|
# Give another chance (we may have been disconnected due to timeout)
|
||||||
|
self._create_socket()
|
||||||
|
self._socket.send(packet)
|
||||||
|
|
||||||
def ip_to_geo(self, ip):
|
def ip_to_geo(self, ip):
|
||||||
ip_type = IPToGeo.IPV4
|
ip_type = IPToGeo.IPV4
|
||||||
if ip.find('.') >= 0:
|
if ip.find('.') >= 0:
|
||||||
|
|
Loading…
Reference in New Issue
Block a user