PsychoticNinja/JLU5-SupyPlugins
1
0
Fork 0
mirror of https://github.com/jlu5/SupyPlugins.git synced 2025-05-02 08:21:10 -05:00
Code Issues Projects Releases Wiki Activity

NuWeather: implement geocoding from OpenStreetMap/Nominatim

Browse Source 
This will be used by future weather fetchers.
This commit is contained in:
James Lu 2018-12-30 21:38:03 -08:00
parent 925b3b3b0b
commit 80dbff83a0
1 changed files with 44 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
NuWeather/plugin.py
View File

@ -28,6 +28,7 @@
###
import json
import os
from supybot import utils, plugins, ircutils, callbacks, world, conf
from supybot.commands import *
@ -52,12 +53,28 @@ class NuWeather(callbacks.Plugin):
def __init__(self, irc):
super().__init__(irc)
# We use 2 DBs: one to store preferred locations and another for location latlon (geocoding)
self.db = accountsdb.AccountsDB("NuWeather", 'NuWeather.db')
geocode_db_filename = conf.supybot.directories.data.dirize("NuWeather-geocode.json")
if os.path.exists(geocode_db_filename):
with open(geocode_db_filename) as f:
self.geocode_db = json.load(f)
else:
self.log.info("NuWeather: Creating new geocode DB")
self.geocode_db = {}
world.flushers.append(self.db.flush)
world.flushers.append(self._flush_geocode_db)
def _flush_geocode_db(self):
geocode_db_filename = conf.supybot.directories.data.dirize("NuWeather-geocode.json")
with open(geocode_db_filename, 'w') as f:
json.dump(self.geocode_db, f)
def die(self):
world.flushers.remove(self.db.flush)
world.flushers.remove(self._flush_geocode_db)
self.db.flush()
self._flush_geocode_db()
super().die()
def _format_temp(self, f, c=None, msg=None):
@ -116,6 +133,33 @@ class NuWeather(callbacks.Plugin):
string = '%d (%s)' % (uv, risk)
return ircutils.mircColor(string, color)
def _nominatim_geocode(self, location):
location = location.lower()
if location in self.geocode_db:
self.log.debug('NuWeather: using cached latlon %s for location %s', self.geocode_db[location], location)
return self.geocode_db[location]
url = 'https://nominatim.openstreetmap.org/search/%s?format=jsonv2' % location
self.log.debug('NuWeather: using url %s (geocoding)', url)
# Custom User agent & caching are required for Nominatim per https://operations.osmfoundation.org/policies/nominatim/
f = utils.web.getUrl(url, headers=HEADERS).decode('utf-8')
data = json.loads(f)
if not data:
raise callbacks.Error("Unknown location %s." % location)
data = data[0]
display_name = data['display_name']
lat = data['lat']
lon = data['lon']
osm_id = data['osm_id']
self.log.debug('NuWeather: saving %s,%s (osm_id %s, %s) for location %s', lat, lon, osm_id, display_name, location)
self.geocode_db[location] = (lat, lon)
return (lat, lon)
_geocode = _nominatim_geocode # Maybe we'll add more backends for this in the future?
_geocode.backend = "OSM/Nominatim"
def _apixu_fetcher(self, location):
"""Grabs weather data from Apixu."""
apikey = self.registryValue('apikeys.apixu')