soutade/iptogeo
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Raise an exception on empty packets (geoip Python class)

Browse Source
This commit is contained in:
Grégory Soutadé 2016-02-04 20:39:50 +01:00
parent 148a2a81c8
commit bc9c8c3dcc
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/iptogeo.py
View File

@ -75,11 +75,12 @@ class IPToGeo(object):
try: try:
self._socket.send(packet) self._socket.send(packet)
except IOError, e: except IOError, e:
# Give another chance (we may have been disconnected due to timeout # Give another chance (we may have been disconnected due to timeout)
self._create_socket() self._create_socket()
self._socket.send(packet) self._socket.send(packet)
packet = self._socket.recv(IPToGeo.PACKET_SIZE) packet = self._socket.recv(IPToGeo.PACKET_SIZE)
if not packet:
raise IPToGeoException('Error, empty packet')
(ip, country_code) = self._check_request(packet) (ip, country_code) = self._check_request(packet)
if country_code: if country_code:
# convert to string # convert to string