69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
"""Transform name location to GPS coordinates"""
|
|
|
|
# PYTHON STUFFS #######################################################
|
|
|
|
import re
|
|
from urllib.parse import quote
|
|
|
|
from nemubot.exception import IMException
|
|
from nemubot.hooks import hook
|
|
from nemubot.tools import web
|
|
|
|
from nemubot.module.more import Response
|
|
|
|
# GLOBALS #############################################################
|
|
|
|
URL_API = "https://open.mapquestapi.com/geocoding/v1/address?key=%s&location=%%s"
|
|
|
|
|
|
# LOADING #############################################################
|
|
|
|
def load(context):
|
|
if not context.config or "apikey" not in context.config:
|
|
raise ImportError("You need a MapQuest API key in order to use this "
|
|
"module. Add it to the module configuration file:\n"
|
|
"<module name=\"mapquest\" key=\"XXXXXXXXXXXXXXXX\" "
|
|
"/>\nRegister at https://developer.mapquest.com/")
|
|
global URL_API
|
|
URL_API = URL_API % context.config["apikey"].replace("%", "%%")
|
|
|
|
|
|
# MODULE CORE #########################################################
|
|
|
|
def geocode(location):
|
|
obj = web.getJSON(URL_API % quote(location))
|
|
|
|
if "results" in obj and "locations" in obj["results"][0]:
|
|
for loc in obj["results"][0]["locations"]:
|
|
yield loc
|
|
|
|
|
|
def where(loc):
|
|
return re.sub(" +", " ",
|
|
"{street} {adminArea5} {adminArea4} {adminArea3} "
|
|
"{adminArea1}".format(**loc)).strip()
|
|
|
|
|
|
# MODULE INTERFACE ####################################################
|
|
|
|
@hook.command("geocode",
|
|
help="Get GPS coordinates of a place",
|
|
help_usage={
|
|
"PLACE": "Get GPS coordinates of PLACE"
|
|
})
|
|
def cmd_geocode(msg):
|
|
if not len(msg.args):
|
|
raise IMException("indicate a name")
|
|
|
|
res = Response(channel=msg.channel, nick=msg.frm,
|
|
nomore="No more geocode", count=" (%s more geocode)")
|
|
|
|
for loc in geocode(' '.join(msg.args)):
|
|
res.append_message("%s is at %s,%s (%s precision)" %
|
|
(where(loc),
|
|
loc["latLng"]["lat"],
|
|
loc["latLng"]["lng"],
|
|
loc["geocodeQuality"].lower()))
|
|
|
|
return res
|