bitbot-3.11-fork/modules/weather.py
2020-03-04 14:27:48 +00:00

90 lines
3.5 KiB
Python

#--depends-on commands
#--depends-on location
#--require-config openweathermap-api-key
from src import ModuleManager, utils
URL_WEATHER = "http://api.openweathermap.org/data/2.5/weather"
class Module(ModuleManager.BaseModule):
def _user_location(self, user):
user_location = user.get_setting("location", None)
if not user_location == None:
name = user_location.get("name", None)
return [user_location["lat"], user_location["lon"], name]
@utils.hook("received.command.w", alias_of="weather")
@utils.hook("received.command.weather")
def weather(self, event):
"""
:help: Get current weather for you or someone else
:usage: [nickname]
:require_setting: location
:require_setting_unless: 1
"""
api_key = self.bot.config["openweathermap-api-key"]
location = None
query = None
nickname = None
if event["args"]:
query = event["args"]
if len(event["args_split"]) == 1 and event["server"].has_user_id(
event["args_split"][0]):
target_user = event["server"].get_user(event["args_split"][0])
location = self._user_location(target_user)
if not location == None:
nickname = target_user.nickname
else:
location = self._user_location(event["user"])
nickname = event["user"].nickname
if location == None:
raise utils.EventError("You don't have a location set")
args = {"units": "metric", "APPID": api_key}
if location == None and query:
location_info = self.exports.get_one("get-location")(query)
if not location_info == None:
location = [location_info["lat"], location_info["lon"],
location_info.get("name", None)]
if location == None:
raise utils.EventError("Unknown location")
lat, lon, location_name = location
args["lat"] = lat
args["lon"] = lon
page = utils.http.request(URL_WEATHER, get_params=args).json()
if page:
if "weather" in page:
if location_name:
location_str = location_name
else:
location_parts = [page["name"]]
if "country" in page["sys"]:
location_parts.append(page["sys"]["country"])
location_str = ", ".join(location_parts)
celsius = "%dC" % page["main"]["temp"]
fahrenheit = "%dF" % ((page["main"]["temp"]*(9/5))+32)
description = page["weather"][0]["description"].title()
humidity = "%s%%" % page["main"]["humidity"]
# wind speed is in metres per second - 3.6* for KMh
wind_speed = 3.6*page["wind"]["speed"]
wind_speed_k = "%skm/h" % round(wind_speed, 1)
wind_speed_m = "%smi/h" % round(0.6214*wind_speed, 1)
if not nickname == None:
location_str = "(%s) %s" % (nickname, location_str)
event["stdout"].write(
"%s | %s/%s | %s | Humidity: %s | Wind: %s/%s" % (
location_str, celsius, fahrenheit, description, humidity,
wind_speed_k, wind_speed_m))
else:
event["stderr"].write("No weather information for this location")
else:
raise utils.EventResultsError()