224 lines
9.3 KiB
Python
224 lines
9.3 KiB
Python
# coding=utf-8
|
|
|
|
"""The weather module. Powered by Dark Sky <https://darksky.net/poweredby/>"""
|
|
|
|
import datetime
|
|
import re
|
|
|
|
from nemubot import context
|
|
from nemubot.exception import IMException
|
|
from nemubot.hooks import hook
|
|
from nemubot.tools import web
|
|
from nemubot.tools.xmlparser.node import ModuleState
|
|
|
|
from nemubot.module import mapquest
|
|
|
|
nemubotversion = 4.0
|
|
|
|
from nemubot.module.more import Response
|
|
|
|
URL_DSAPI = "https://api.darksky.net/forecast/%s/%%s,%%s?lang=%%s&units=%%s"
|
|
|
|
def load(context):
|
|
if not context.config or "darkskyapikey" not in context.config:
|
|
raise ImportError("You need a Dark-Sky API key in order to use this "
|
|
"module. Add it to the module configuration file:\n"
|
|
"<module name=\"weather\" darkskyapikey=\"XXX\" />\n"
|
|
"Register at http://developer.forecast.io/")
|
|
context.data.setIndex("name", "city")
|
|
global URL_DSAPI
|
|
URL_DSAPI = URL_DSAPI % context.config["darkskyapikey"]
|
|
|
|
|
|
def format_wth(wth):
|
|
return ("{temperature} °C {summary}; precipitation ({precipProbability:.0%} chance) intensity: {precipIntensity} mm/h; relative humidity: {humidity:.0%}; wind speed: {windSpeed} km/s {windBearing}°; cloud coverage: {cloudCover:.0%}; pressure: {pressure} hPa; ozone: {ozone} DU"
|
|
.format(**wth)
|
|
)
|
|
|
|
|
|
def format_forecast_daily(wth):
|
|
return ("{summary}; between {temperatureMin}-{temperatureMax} °C; precipitation ({precipProbability:.0%} chance) intensity: maximum {precipIntensity} mm/h; relative humidity: {humidity:.0%}; wind speed: {windSpeed} km/h {windBearing}°; cloud coverage: {cloudCover:.0%}; pressure: {pressure} hPa; ozone: {ozone} DU".format(**wth))
|
|
|
|
|
|
def format_timestamp(timestamp, tzname, tzoffset, format="%c"):
|
|
tz = datetime.timezone(datetime.timedelta(hours=tzoffset), tzname)
|
|
time = datetime.datetime.fromtimestamp(timestamp, tz=tz)
|
|
return time.strftime(format)
|
|
|
|
|
|
def treat_coord(msg):
|
|
if len(msg.args) > 0:
|
|
|
|
# catch dans X[jh]$
|
|
if len(msg.args) > 2 and (msg.args[-2] == "dans" or msg.args[-2] == "in" or msg.args[-2] == "next"):
|
|
specific = msg.args[-1]
|
|
city = " ".join(msg.args[:-2]).lower()
|
|
else:
|
|
specific = None
|
|
city = " ".join(msg.args).lower()
|
|
|
|
if len(msg.args) == 2:
|
|
coords = msg.args
|
|
else:
|
|
coords = msg.args[0].split(",")
|
|
|
|
try:
|
|
if len(coords) == 2 and str(float(coords[0])) == coords[0] and str(float(coords[1])) == coords[1]:
|
|
return coords, specific
|
|
except ValueError:
|
|
pass
|
|
|
|
if city in context.data.index:
|
|
coords = list()
|
|
coords.append(context.data.index[city]["lat"])
|
|
coords.append(context.data.index[city]["long"])
|
|
return city, coords, specific
|
|
|
|
else:
|
|
geocode = [x for x in mapquest.geocode(city)]
|
|
if len(geocode):
|
|
coords = list()
|
|
coords.append(geocode[0]["latLng"]["lat"])
|
|
coords.append(geocode[0]["latLng"]["lng"])
|
|
return mapquest.where(geocode[0]), coords, specific
|
|
|
|
raise IMException("Je ne sais pas où se trouve %s." % city)
|
|
|
|
else:
|
|
raise IMException("indique-moi un nom de ville ou des coordonnées.")
|
|
|
|
|
|
def get_json_weather(coords, lang="en", units="auto"):
|
|
wth = web.getJSON(URL_DSAPI % (float(coords[0]), float(coords[1]), lang, units))
|
|
|
|
# First read flags
|
|
if wth is None or "darksky-unavailable" in wth["flags"]:
|
|
raise IMException("The given location is supported but a temporary error (such as a radar station being down for maintenace) made data unavailable.")
|
|
|
|
return wth
|
|
|
|
|
|
@hook.command("coordinates")
|
|
def cmd_coordinates(msg):
|
|
if len(msg.args) < 1:
|
|
raise IMException("indique-moi un nom de ville.")
|
|
|
|
j = msg.args[0].lower()
|
|
if j not in context.data.index:
|
|
raise IMException("%s n'est pas une ville connue" % msg.args[0])
|
|
|
|
coords = context.data.index[j]
|
|
return Response("Les coordonnées de %s sont %s,%s" % (msg.args[0], coords["lat"], coords["long"]), channel=msg.channel)
|
|
|
|
|
|
@hook.command("alert",
|
|
keywords={
|
|
"lang=LANG": "change the output language of weather sumarry; default: en",
|
|
"units=UNITS": "return weather conditions in the requested units; default: auto",
|
|
})
|
|
def cmd_alert(msg):
|
|
loc, coords, specific = treat_coord(msg)
|
|
wth = get_json_weather(coords,
|
|
lang=msg.kwargs["lang"] if "lang" in msg.kwargs else "en",
|
|
units=msg.kwargs["units"] if "units" in msg.kwargs else "auto")
|
|
|
|
res = Response(channel=msg.channel, nomore="No more weather alert", count=" (%d more alerts)")
|
|
|
|
if "alerts" in wth:
|
|
for alert in wth["alerts"]:
|
|
if "expires" in alert:
|
|
res.append_message("\x03\x02%s\x03\x02 (see %s expire on %s): %s" % (alert["title"], alert["uri"], format_timestamp(int(alert["expires"]), wth["timezone"], wth["offset"]), alert["description"].replace("\n", " ")))
|
|
else:
|
|
res.append_message("\x03\x02%s\x03\x02 (see %s): %s" % (alert["title"], alert["uri"], alert["description"].replace("\n", " ")))
|
|
|
|
return res
|
|
|
|
|
|
@hook.command("météo",
|
|
help="Display current weather and previsions",
|
|
help_usage={
|
|
"CITY": "Display the current weather and previsions in CITY",
|
|
},
|
|
keywords={
|
|
"lang=LANG": "change the output language of weather sumarry; default: en",
|
|
"units=UNITS": "return weather conditions in the requested units; default: auto",
|
|
})
|
|
def cmd_weather(msg):
|
|
loc, coords, specific = treat_coord(msg)
|
|
wth = get_json_weather(coords,
|
|
lang=msg.kwargs["lang"] if "lang" in msg.kwargs else "en",
|
|
units=msg.kwargs["units"] if "units" in msg.kwargs else "auto")
|
|
|
|
res = Response(channel=msg.channel, nomore="No more weather information")
|
|
|
|
if "alerts" in wth:
|
|
alert_msgs = list()
|
|
for alert in wth["alerts"]:
|
|
if "expires" in alert:
|
|
alert_msgs.append("\x03\x02%s\x03\x02 expire on %s" % (alert["title"], format_timestamp(int(alert["expires"]), wth["timezone"], wth["offset"])))
|
|
else:
|
|
alert_msgs.append("\x03\x02%s\x03\x02" % (alert["title"]))
|
|
res.append_message("\x03\x16\x03\x02/!\\\x03\x02 Alert%s:\x03\x16 " % ("s" if len(alert_msgs) > 1 else "") + ", ".join(alert_msgs))
|
|
|
|
if specific is not None:
|
|
gr = re.match(r"^([0-9]*)\s*([a-zA-Z])", specific)
|
|
if gr is None or gr.group(1) == "":
|
|
gr1 = 1
|
|
else:
|
|
gr1 = int(gr.group(1))
|
|
|
|
if gr.group(2).lower() == "h" and gr1 < len(wth["hourly"]["data"]):
|
|
hour = wth["hourly"]["data"][gr1]
|
|
res.append_message("\x03\x02At %sh:\x03\x02 %s" % (format_timestamp(int(hour["time"]), wth["timezone"], wth["offset"], '%H'), format_wth(hour)))
|
|
|
|
elif gr.group(2).lower() == "d" and gr1 < len(wth["daily"]["data"]):
|
|
day = wth["daily"]["data"][gr1]
|
|
res.append_message("\x03\x02On %s:\x03\x02 %s" % (format_timestamp(int(day["time"]), wth["timezone"], wth["offset"], '%A'), format_forecast_daily(day)))
|
|
|
|
else:
|
|
res.append_message("I don't understand %s or information is not available" % specific)
|
|
|
|
else:
|
|
res.append_message("\x03\x02Currently:\x03\x02 " + format_wth(wth["currently"]))
|
|
|
|
nextres = "\x03\x02Today:\x03\x02 %s " % wth["daily"]["data"][0]["summary"]
|
|
if "minutely" in wth:
|
|
nextres += "\x03\x02Next hour:\x03\x02 %s " % wth["minutely"]["summary"]
|
|
nextres += "\x03\x02Next 24 hours:\x03\x02 %s \x03\x02Next week:\x03\x02 %s" % (wth["hourly"]["summary"], wth["daily"]["summary"])
|
|
res.append_message(nextres)
|
|
|
|
for hour in wth["hourly"]["data"][1:4]:
|
|
res.append_message("\x03\x02At %sh:\x03\x02 %s" % (format_timestamp(int(hour["time"]), wth["timezone"], wth["offset"], '%H'),
|
|
format_wth(hour)))
|
|
|
|
for day in wth["daily"]["data"][1:]:
|
|
res.append_message("\x03\x02On %s:\x03\x02 %s" % (format_timestamp(int(day["time"]), wth["timezone"], wth["offset"], '%A'),
|
|
format_forecast_daily(day)))
|
|
|
|
return res
|
|
|
|
|
|
gps_ask = re.compile(r"^\s*(?P<city>.*\w)\s*(?:(?:se|est)\s+(?:trouve|situ[ée]*)\s+[aà])\s*(?P<lat>-?[0-9]+(?:[,.][0-9]+))[^0-9.](?P<long>-?[0-9]+(?:[,.][0-9]+))\s*$", re.IGNORECASE)
|
|
|
|
|
|
@hook.ask()
|
|
def parseask(msg):
|
|
res = gps_ask.match(msg.message)
|
|
if res is not None:
|
|
city_name = res.group("city").lower()
|
|
gps_lat = res.group("lat").replace(",", ".")
|
|
gps_long = res.group("long").replace(",", ".")
|
|
|
|
if city_name in context.data.index:
|
|
context.data.index[city_name]["lat"] = gps_lat
|
|
context.data.index[city_name]["long"] = gps_long
|
|
else:
|
|
ms = ModuleState("city")
|
|
ms.setAttribute("name", city_name)
|
|
ms.setAttribute("lat", gps_lat)
|
|
ms.setAttribute("long", gps_long)
|
|
context.data.addChild(ms)
|
|
context.save()
|
|
return Response("ok, j'ai bien noté les coordonnées de %s" % res.group("city"),
|
|
msg.channel, msg.frm)
|