get city not country list

This commit is contained in:
j 2020-05-30 03:13:57 +02:00
parent 6217fb774f
commit 5cde8977ef
2 changed files with 13 additions and 7 deletions

View File

@ -394,8 +394,7 @@ def update_static():
def update_geoip(force=False):
path = os.path.join(settings.GEOIP_PATH, 'GeoLite2-City.mmdb')
if not os.path.exists(path) or force:
url = 'http://geolite.maxmind.com/download/geoip/database/GeoLite2-City.mmdb.gz'
index = ox.net.read_url('https://db-ip.com/db/download/ip-to-country-lite').decode()
index = ox.net.read_url('https://db-ip.com/db/download/ip-to-city-lite').decode()
match = re.compile('href=[\'"](http.*.mmdb.gz)').findall(index)
if match:
url = match[0]
@ -405,7 +404,7 @@ def update_geoip(force=False):
os.unlink(path)
os.system('gunzip "%s.gz"' % path)
else:
print('failed to download dbip-country-lite-2020-03.mmdb.gz')
print('failed to download GeoLite2-City.mmdb')
def init():
if not settings.RELOADER_RUNNING:

View File

@ -17,15 +17,22 @@ def get_location(ip):
country = city = None
try:
g = GeoIP2()
location = g.city(ip)
except:
country = city = None
else:
try:
location = g.city(ip)
except django.contrib.gis.geoip2.GeoIP2Exception:
try:
location = g.country(s.ip)
except:
location = None
if location:
country = ox.get_country_name(location['country_code'])
if location['city']:
if location.get('city'):
city = location['city']
if isinstance(city, bytes):
city = city.decode('latin-1')
except:
country = city = None
return city, country