Browse Source

update geoip related stuff

pull/3/head
Jeff Becker 8 years ago
parent
commit
e548512f1e
  1. 8
      baddie-detector/netdb/netdb.py

8
baddie-detector/netdb/netdb.py

@ -7,9 +7,12 @@
import os,sys,struct,time,hashlib,fnmatch,io import os,sys,struct,time,hashlib,fnmatch,io
import base64 import base64
import logging import logging
import pygeoip
geo = pygeoip.GeoIP('/usr/share/GeoIP/GeoIPCity.dat') try:
import pygeoip
geo = pygeoip.GeoIP('/usr/share/GeoIP/GeoIPCity.dat')
except:
geo = None
b64encode = lambda x : base64.b64encode(x, b'~-').decode('ascii') b64encode = lambda x : base64.b64encode(x, b'~-').decode('ascii')
@ -156,6 +159,7 @@ class Entry:
@staticmethod @staticmethod
def geolookup(entry): def geolookup(entry):
if geo:
return geo.record_by_addr(entry) return geo.record_by_addr(entry)
@staticmethod @staticmethod

Loading…
Cancel
Save