Mirrors/wttr.in
2
0
Fork 0
mirror of https://github.com/chubin/wttr.in synced 2024-11-15 00:27:09 +00:00
Code Issues Projects Releases Packages Wiki Activity

Do cache read and write in get_location instead of duplicated in each IPLOCATION_ORDER method

Browse source
This commit is contained in:
Gregory Danielson 2020-11-01 16:31:15 -06:00
parent 7af4aaa8ec
commit e4ac3f266f
No known key found for this signature in database
GPG key ID: 88D4EF22F6C14CA7
1 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
lib/location.py
View file

@ -187,6 +187,10 @@ def get_location(ip_addr):
"""
Return location triple (CITY, REGION, COUNTRY) for `ip_addr`
"""
location = ipcache(ip_addr)
if location:
return location
for method in IPLOCATION_ORDER:
if method == 'geoip':
city, region, country = geoip(ip_addr)
@ -195,10 +199,13 @@ def get_location(ip_addr):
elif method == 'ipinfo':
city, region, country = ipinfo(ip_addr)
else:
print("ERROR: invalid iplocation method speficied: %s" % method)
if city is not None:
city, region, country = workaround(city, region, country)
return city, region, country
print("ERROR: invalid iplocation method specified: %s" % method)
if all((city, region, country)):
ipcachewrite(ip_addr, (city, region, country))
# cache write used to happen before workaround, preserve that
city, region, country = workaround(city, region, country)
return city, region, country
#
# temporary disabled it because of geoip services capcacity
#