Edm0nd/edmond/plugins/meteo_france.py

119 lines
3.6 KiB
Python
Raw Normal View History

import time
import meteofrance_api as mf
from edmond.plugin import Plugin
class MeteoFrancePlugin(Plugin):
REQUIRED_CONFIGS = [
"commands",
"result_message",
"nearest_message",
"temp_format",
"rain_chance_format",
"all_rain_format",
"today_message",
"minmax_temp_format",
]
def __init__(self, bot):
super().__init__(bot)
self.client = None
def on_welcome(self, event):
self.client = mf.MeteoFranceClient()
def on_pubmsg(self, event):
if not self.should_handle_command(event.arguments[0]):
return False
if self.command.ident == self.config["commands"][0]:
self.tell_weather(event)
return True
def tell_weather(self, event):
places = self.client.search_places(self.command.content)
if not places:
self.signal_failure(event.target)
return
place = places[0] # Assume first result is the good one.
try:
forecast = self.client.get_forecast_for_place(place)
except: # really unsure about that unofficial API
self.signal_failure(event.target)
return
nearest_f = forecast.nearest_forecast
today_f = forecast.today_forecast
try:
nearest_s = self.format_nearest_forecast(nearest_f, place)
today_s = self.format_today_forecast(today_f)
except KeyError:
self.signal_failure(event.target)
return
result = self.config["result_message"].format(
nearest_s=nearest_s, today_s=today_s
)
self.bot.say(event.target, result)
def format_nearest_forecast(self, nearest, place):
city = place.name
hour = format_ts_hour(nearest["dt"])
temp = self.format_temperature(nearest["T"])
weather = format_weather(nearest["weather"])
wind = self.format_wind(nearest["wind"])
rain = self.format_rain_perc(nearest["rain"])
return self.config["nearest_message"].format(
city=city,
hour=hour,
weather=weather,
temp=temp,
wind=wind,
rain=rain,
)
def format_temperature(self, temperature):
temp_c = temperature.get("value", "?")
temp_wc = temperature.get("windchill", "?")
return self.config["temp_format"].format(
temp_c=temp_c, temp_wc=temp_wc
)
def format_rain_perc(self, rain_percs):
perc_fmt = self.config["rain_chance_format"]
all_percs = ", ".join(
perc_fmt.format(p=p, h=h) for h, p in rain_percs.items()
)
return self.config["all_rain_format"].format(all_percs=all_percs)
def format_wind(self, wind):
speed = wind.get("speed")
direction = wind.get("icon")
if direction == "Variable":
direction = self.config["unknown_direction"]
return self.config["wind_format"].format(
speed=speed, direction=direction
)
def format_today_forecast(self, today_f):
temp = today_f["T"]
temp_minmax = self.config["minmax_temp_format"].format(
t_max=temp["max"], t_min=temp["min"]
)
rain = self.format_rain_perc(today_f["precipitation"])
sunset = format_ts_hour(today_f["sun"]["set"])
return self.config["today_message"].format(
temp_minmax=temp_minmax, rain=rain, sunset=sunset
)
def format_ts_hour(timestamp):
return time.strftime("%H:%M", time.localtime(timestamp))
def format_weather(weather):
return weather.get("desc", "?").lower()