dr.botzo/modules/Weather.py

72 lines
3.2 KiB
Python

# coding: utf-8
"""
Weather - query various weather services for info
Copyright (C) 2010 Brian S. Stephan
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
import re
from extlib import irclib
from extlib import pywapi
from urllib import quote
from urllib2 import URLError
from Module import Module
class Weather(Module):
"""Provide weather lookup services to the bot."""
def do(self, connection, event, nick, userhost, replypath, what, admin_unlocked):
"""Query Google Weather for a location's weather."""
whats = what.split(' ')
if whats[0] == "weather" and len(whats) >= 2:
try:
google_weather = pywapi.get_weather_from_google(quote(' '.join(whats[1:])))
city = google_weather['forecast_information']['city'].encode('utf-8')
condition = google_weather['current_conditions']['condition'].encode('utf-8')
temp_f = google_weather['current_conditions']['temp_f'].encode('utf-8')
temp_c = google_weather['current_conditions']['temp_c'].encode('utf-8')
wind = google_weather['current_conditions']['wind_condition'].encode('utf-8')
humidity = google_weather['current_conditions']['humidity'].encode('utf-8')
wind_speed_regex = re.compile("Wind: [NSEW]+ at ([0-9]+) mph")
matches = wind_speed_regex.search(wind)
if matches is not None and float(temp_f) < 50.0:
wind_speed = matches.group(1)
windchill = 35.74 + (0.6215 * float(temp_f)) - (35.75 * float(wind_speed)**0.16) + (0.4275 * float(temp_f) * float(wind_speed)**0.16)
else:
windchill = ""
weatherstr = "Current weather for " + city + ": " + condition + ". " + temp_f + "°F (" + temp_c + "°C), " + wind + ", " + humidity
if windchill is not "":
weatherstr += ", Wind chill: " + str(round(windchill, 2)) + "°F"
weatherstr += "."
for city in google_weather['forecasts']:
weatherstr += " " + city['day_of_week'].encode('utf-8') + ": " + city['condition'].encode('utf-8') + ". High " + city['high'].encode('utf-8') + "°F, Low " + city['low'].encode('utf-8') + "°F."
return self.reply(connection, replypath, weatherstr)
except URLError as e:
return self.reply(connection, replypath, "error connecting to google weather:" + str(e))
except IndexError as e:
return self.reply(connection, replypath, "error in pywapi: " + str(e))
# vi:tabstop=4:expandtab:autoindent
# kate: indent-mode python;indent-width 4;replace-tabs on;