New module: weather; close #7

This commit is contained in:
nemunaire 2014-04-30 22:41:47 +02:00
parent 26502abe35
commit b925cee08a

207
modules/weather.py Normal file
View File

@ -0,0 +1,207 @@
# coding=utf-8
import datetime
import json
import re
from urllib.parse import quote
from urllib.request import urlopen
nemubotversion = 3.3
def load(context):
global DATAS
DATAS.setIndex("name", "city")
if not CONF.hasNode("darkskyapi") or not CONF.getNode("darkskyapi").hasAttribute("key"):
print ("You need a Dark-Sky API key in order to use this "
"module. Add it to the module configuration file:\n<darkskyapi"
" key=\"XXXXXXXXXXXXXXXX\" />\nRegister at "
"http://developer.forecast.io/")
return None
from hooks import Hook
add_hook("cmd_hook", Hook(cmd_weather, "météo"))
add_hook("cmd_hook", Hook(cmd_alert, "alert"))
def help_tiny ():
"""Line inserted in the response to the command !help"""
return "The weather module"
def help_full ():
return "!weather /city/: Display the current weather in /city/."
def fahrenheit2celsius(temp):
return int((temp - 32) * 50/9)/10
def mph2kmph(speed):
return int(speed * 160.9344)/100
def inh2mmh(size):
return int(size * 254)/10
def format_wth(wth):
return ("%s °C %s; precipitation (%s %% chance) intensity: %s mm/h; relative humidity: %s %%; wind speed: %s km/h %s°; cloud coverage: %s %%; pressure: %s hPa; ozone: %s DU" %
(
fahrenheit2celsius(wth["temperature"]),
wth["summary"],
int(wth["precipProbability"] * 100),
inh2mmh(wth["precipIntensity"]),
int(wth["humidity"] * 100),
mph2kmph(wth["windSpeed"]),
wth["windBearing"],
int(wth["cloudCover"] * 100),
int(wth["pressure"]),
int(wth["ozone"])
))
def format_forecast_daily(wth):
return ("%s; between %s-%s °C; precipitation (%s %% chance) intensity: maximum %s mm/h; relative humidity: %s %%; wind speed: %s km/h %s°; cloud coverage: %s %%; pressure: %s hPa; ozone: %s DU" %
(
wth["summary"],
fahrenheit2celsius(wth["temperatureMin"]), fahrenheit2celsius(wth["temperatureMax"]),
int(wth["precipProbability"] * 100),
inh2mmh(wth["precipIntensityMax"]),
int(wth["humidity"] * 100),
mph2kmph(wth["windSpeed"]),
wth["windBearing"],
int(wth["cloudCover"] * 100),
int(wth["pressure"]),
int(wth["ozone"])
))
def treat_coord(msg):
if len(msg.cmds) > 1:
# catch dans X[jh]$
if len(msg.cmds) > 3 and msg.cmds[len(msg.cmds) - 2] == "dans" or msg.cmds[len(msg.cmds) - 2] == "in" or msg.cmds[len(msg.cmds) - 2] == "next":
specific = msg.cmds[len(msg.cmds) - 1]
msg.cmds = msg.cmds[:len(msg.cmds) - 2]
else:
specific = None
j = " ".join(msg.cmds[1:]).lower()
if len(msg.cmds) == 3:
coords = msg.cmds[1:3]
else:
coords = msg.cmds[1].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 j in DATAS.index:
coords = list()
coords.append(DATAS.index[j]["lat"])
coords.append(DATAS.index[j]["long"])
return coords, specific
raise IRCException("Je ne sais pas où se trouve %s." % " ".join(msg.cmds[1:]))
else:
raise IRCException("indique-moi un nom de ville ou des coordonnées.")
def get_json_weather(coords):
raw = urlopen("https://api.forecast.io/forecast/%s/%s,%s" % (CONF.getNode("darkskyapi")["key"], float(coords[0]), float(coords[1])), timeout=10)
wth = json.loads(raw.read().decode())
# First read flags
if "darksky-unavailable" in wth["flags"]:
raise IRCException("The given location is supported but a temporary error (such as a radar station being down for maintenace) made data unavailable.")
return wth
def cmd_alert(msg):
coords, specific = treat_coord(msg)
wth = get_json_weather(coords)
res = Response(msg.sender, channel=msg.channel, nomore="No more weather alert", count=" (%d more alerts)")
if "alerts" in wth:
for alert in wth["alerts"]:
res.append_message("\x03\x02%s\x03\x02 (see %s expire on %s): %s" % (alert["title"], alert["uri"], datetime.datetime.fromtimestamp(int(alert["expires"])).strftime("%c"), alert["description"].replace("\n", " ")))
return res
def cmd_weather(msg):
coords, specific = treat_coord(msg)
wth = get_json_weather(coords)
res = Response(msg.sender, channel=msg.channel, nomore="No more weather information")
if "alerts" in wth:
alert_msgs = list()
for alert in wth["alerts"]:
alert_msgs.append("\x03\x02%s\x03\x02 expire on %s" % (alert["title"], datetime.datetime.fromtimestamp(int(alert["expires"])).strftime("%c")))
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]
time = datetime.datetime.fromtimestamp(int(hour["time"]))
res.append_message("\x03\x02At %sh:\x03\x02 %s" % (time.strftime('%H'), format_wth(hour)))
elif gr.group(2).lower() == "d" and gr1 < len(wth["daily"]["data"]):
day = wth["daily"]["data"][gr1]
time = datetime.datetime.fromtimestamp(int(day["time"]))
res.append_message("\x03\x02On %s:\x03\x02 %s" % (time.strftime('%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]:
time = datetime.datetime.fromtimestamp(int(hour["time"]))
res.append_message("\x03\x02At %sh:\x03\x02 %s" % (time.strftime('%H'), format_wth(hour)))
for day in wth["daily"]["data"][1:]:
time = datetime.datetime.fromtimestamp(int(day["time"]))
res.append_message("\x03\x02On %s:\x03\x02 %s" % (time.strftime('%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)
def parseask(msg):
res = gps_ask.match(msg.content)
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 DATAS.index:
DATAS.index[city_name]["lat"] = gps_lat
DATAS.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)
DATAS.addChild(ms)
save()
return Response(msg.sender,
"ok, j'ai bien noté les coordonnées de %s" % res.group("city"),
msg.channel, msg.nick)