Limnoria/plugins/Weather.py

363 lines
15 KiB
Python
Raw Normal View History

#!/usr/bin/env python
2004-01-07 16:17:53 +01:00
###
# Copyright (c) 2002, Jeremiah Fincher
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions, and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions, and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# * Neither the name of the author of this software nor the name of
# contributors to this software may be used to endorse or promote products
# derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
###
"""
2004-02-23 10:05:12 +01:00
This plugin does weather-related stuff. It can't change the weather, though,
so don't get your hopes up. We just report it.
2004-01-07 16:17:53 +01:00
"""
__revision__ = "$Id$"
2004-03-19 17:58:54 +01:00
2004-01-07 16:17:53 +01:00
import plugins
import re
import sets
2004-02-23 10:05:12 +01:00
import urllib
2004-01-07 16:17:53 +01:00
import conf
2004-01-07 16:17:53 +01:00
import utils
import webutils
2004-02-23 10:49:47 +01:00
import ircutils
2004-01-07 16:17:53 +01:00
import privmsgs
import registry
2004-01-07 16:17:53 +01:00
import callbacks
2004-02-23 12:08:22 +01:00
unitAbbrevs = utils.abbrev(['Fahrenheit', 'Celsius', 'Centigrade', 'Kelvin'])
unitAbbrevs['C'] = 'Celsius'
unitAbbrevs['Ce'] = 'Celsius'
class WeatherUnit(registry.String):
def setValue(self, s):
2004-02-23 12:08:22 +01:00
#print '***', repr(s)
s = s.capitalize()
if s not in unitAbbrevs:
raise registry.InvalidRegistryValue,\
2004-03-30 10:32:17 +02:00
'Unit must be one of Fahrenheit, Celsius, or Kelvin.'
2004-02-23 12:08:22 +01:00
s = unitAbbrevs[s]
registry.String.setValue(self, s)
class WeatherCommand(registry.String):
2004-02-23 10:05:12 +01:00
def setValue(self, s):
m = Weather.weatherCommands
if s not in m:
raise registry.InvalidRegistryValue,\
2004-03-30 10:32:17 +02:00
'Command must be one of %s' % utils.commaAndify(m)
2004-02-23 10:05:12 +01:00
else:
method = getattr(Weather, s)
2004-02-23 10:49:47 +01:00
Weather.weather.im_func.__doc__ = method.__doc__
2004-02-23 10:05:12 +01:00
registry.String.setValue(self, s)
2004-03-30 10:32:17 +02:00
# Registry variables moved to the bottom to use Weather.weatherCommands.
2004-01-07 16:17:53 +01:00
class Weather(callbacks.Privmsg):
2004-02-23 10:05:12 +01:00
weatherCommands = ['ham', 'cnn']
2004-01-07 16:17:53 +01:00
threaded = True
def callCommand(self, method, irc, msg, *L):
try:
callbacks.Privmsg.callCommand(self, method, irc, msg, *L)
except webutils.WebError, e:
irc.error(str(e))
2004-01-07 16:17:53 +01:00
2004-02-23 10:05:12 +01:00
def weather(self, irc, msg, args):
# This specifically does not have a docstring.
channel = None
if ircutils.isChannel(msg.args[0]):
channel = msg.args[0]
if not args:
s = self.userValue('lastLocation', msg.prefix)
if s:
args = [s]
else:
location = privmsgs.getArgs(args)
self.setUserValue(msg.prefix, 'lastLocation',
location, ignoreNoUser=True)
2004-02-25 16:46:22 +01:00
realCommandName = self.registryValue('command', channel)
2004-02-23 10:05:12 +01:00
realCommand = getattr(self, realCommandName)
realCommand(irc, msg, args)
2004-02-23 12:08:22 +01:00
def _toCelsius(self, temp, unit):
if unit == 'K':
return temp - 273.15
2004-02-23 12:35:06 +01:00
elif unit == 'F':
2004-02-23 12:08:22 +01:00
return (temp - 32) * 5 /9
2004-02-23 12:35:06 +01:00
else:
return temp
2004-02-23 12:08:22 +01:00
def _getTemp(self, temp, deg, unit, chan):
2004-02-23 12:08:22 +01:00
assert unit == unit.upper()
assert temp == int(temp)
default = self.registryValue('temperatureUnit', chan)
if unitAbbrevs[unit] == default:
2004-02-23 10:49:47 +01:00
# Short circuit if we're the same unit as the default.
2004-02-23 12:08:22 +01:00
return '%s%s%s' % (temp, deg, unit)
temp = self._toCelsius(temp, unit)
unit = 'C'
if default == 'Kelvin':
temp = temp + 273.15
unit = 'K'
deg = ' '
elif default == 'Fahrenheit':
temp = temp * 9 / 5 + 32
unit = 'F'
return '%s%s%s' % (temp, deg, unit)
2004-01-07 16:17:53 +01:00
_cityregex = re.compile(
r'<td><font size="4" face="arial"><b>'
r'(.*?), (.*?),(.*?)</b></font></td>', re.I)
2004-01-07 16:17:53 +01:00
_interregex = re.compile(
r'<td><font size="4" face="arial"><b>'
r'([^,]+), ([^<]+)</b></font></td>', re.I)
2004-01-07 16:17:53 +01:00
_condregex = re.compile(
r'<td width="100%" colspan="2" align="center"><strong>'
r'<font face="arial">([^<]+)</font></strong></td>', re.I)
2004-01-07 16:17:53 +01:00
_tempregex = re.compile(
r'<td valign="top" align="right"><strong><font face="arial">'
2004-02-23 12:08:22 +01:00
r'(-?\d+)(.*?)(F|C)</font></strong></td>', re.I)
_temp = re.compile(r'(-?\d+)(.*?)(F|C)')
2004-01-07 16:17:53 +01:00
_chillregex = re.compile(
2004-02-23 12:08:22 +01:00
r'Wind Chill</font></strong>:</small></td>\s+<td align="right">'
r'<small><font face="arial">([^N][^<]+)</font></small></td>',
re.I | re.S)
2004-01-07 16:17:53 +01:00
_heatregex = re.compile(
2004-02-23 12:08:22 +01:00
r'Heat Index</font></strong>:</small></td>\s+<td align="right">'
r'<small><font face="arial">([^N][^<]+)</font></small></td>',
re.I | re.S)
2004-01-07 16:17:53 +01:00
# States
_realStates = sets.Set(['ak', 'al', 'ar', 'az', 'ca', 'co', 'ct',
2004-01-11 15:47:44 +01:00
'dc', 'de', 'fl', 'ga', 'hi', 'ia', 'id',
'il', 'in', 'ks', 'ky', 'la', 'ma', 'md',
'me', 'mi', 'mn', 'mo', 'ms', 'mt', 'nc',
'nd', 'ne', 'nh', 'nj', 'nm', 'nv', 'ny',
'oh', 'ok', 'or', 'pa', 'ri', 'sc', 'sd',
'tn', 'tx', 'ut', 'va', 'vt', 'wa', 'wi',
'wv', 'wy'])
2004-01-07 16:17:53 +01:00
# Provinces. (Province being a metric state measurement mind you. :D)
_fakeStates = sets.Set(['ab', 'bc', 'mb', 'nb', 'nf', 'ns', 'nt',
'nu', 'on', 'pe', 'qc', 'sk', 'yk'])
# Certain countries are expected to use a standard abbreviation
# The weather we pull uses weird codes. Map obvious ones here.
_hamCountryMap = {'uk': 'gb'}
def ham(self, irc, msg, args):
2004-02-17 07:29:19 +01:00
"""<US zip code | US/Canada city, state | Foreign city, country>
2004-01-07 16:17:53 +01:00
Returns the approximate weather conditions for a given city.
"""
#If we received more than one argument, then we have received
#a city and state argument that we need to process.
if len(args) > 1:
#If we received more than 1 argument, then we got a city with a
#multi-word name. ie ['Garden', 'City', 'KS'] instead of
#['Liberal', 'KS']. We join it together with a + to pass
#to our query
state = args.pop()
state = state.lower()
city = '+'.join(args)
city = city.rstrip(',')
city = city.lower()
#We must break the States up into two sections. The US and
#Canada are the only countries that require a State argument.
if state in self._realStates:
country = 'us'
elif state in self._fakeStates:
country = 'ca'
else:
country = state
state = ''
if country in self._hamCountryMap.keys():
country = self._hamCountryMap[country]
url = 'http://www.hamweather.net/cgi-bin/hw3/hw3.cgi?' \
'pass=&dpp=&forecast=zandh&config=&' \
'place=%s&state=%s&country=%s' % (city, state, country)
2004-01-11 15:47:44 +01:00
html = webutils.getUrl(url)
if 'was not found' in html:
url = 'http://www.hamweather.net/cgi-bin/hw3/hw3.cgi?' \
'pass=&dpp=&forecast=zandh&config=&' \
'place=%s&state=&country=%s' % (city, state)
html = webutils.getUrl(url)
if 'was not found' in html: # Still.
irc.error('No such location could be found.')
return
2004-01-07 16:17:53 +01:00
#We received a single argument. Zipcode or station id.
else:
zip = privmsgs.getArgs(args)
zip = zip.replace(',', '')
zip = zip.lower()
url = 'http://www.hamweather.net/cgi-bin/hw3/hw3.cgi?' \
'config=&forecast=zandh&pands=%s&Submit=GO' % zip
2004-01-07 16:17:53 +01:00
html = webutils.getUrl(url)
2004-01-11 15:47:44 +01:00
if 'was not found' in html:
irc.error('No such location could be found.')
return
2004-01-07 16:17:53 +01:00
headData = self._cityregex.search(html)
2004-02-08 08:16:58 +01:00
if headData is not None:
2004-01-07 16:17:53 +01:00
(city, state, country) = headData.groups()
else:
headData = self._interregex.search(html)
2004-01-11 15:47:44 +01:00
if headData:
(city, state) = headData.groups()
else:
irc.error('No such location could be found.')
return
2004-01-07 16:17:53 +01:00
city = city.strip()
state = state.strip()
temp = self._tempregex.search(html)
convert = self.registryValue('convert', msg.args[0])
2004-02-08 08:16:58 +01:00
if temp is not None:
2004-02-23 12:08:22 +01:00
(temp, deg, unit) = temp.groups()
if convert:
temp = self._getTemp(int(temp), deg, unit, msg.args[0])
else:
temp = deg.join((temp, unit))
2004-01-07 16:17:53 +01:00
conds = self._condregex.search(html)
2004-02-08 08:16:58 +01:00
if conds is not None:
2004-01-07 16:17:53 +01:00
conds = conds.group(1)
index = ''
chill = self._chillregex.search(html)
2004-02-08 08:16:58 +01:00
if chill is not None:
2004-01-07 16:17:53 +01:00
chill = chill.group(1)
2004-02-23 12:08:22 +01:00
chill = utils.htmlToText(chill)
if convert:
tempsplit = self._temp.search(chill)
if tempsplit:
(chill, deg, unit) = tempsplit.groups()
chill = self._getTemp(int(chill), deg, unit,msg.args[0])
if float(chill[:-2]) < float(temp[:-2]):
index = ' (Wind Chill: %s)' % chill
2004-01-07 16:17:53 +01:00
heat = self._heatregex.search(html)
2004-02-08 07:24:00 +01:00
if heat is not None:
2004-01-07 16:17:53 +01:00
heat = heat.group(1)
2004-02-23 12:08:22 +01:00
heat = utils.htmlToText(heat)
if convert:
tempsplit = self._temp.search(heat)
if tempsplit:
(heat, deg, unit) = tempsplit.groups()
if convert:
heat = self._getTemp(int(heat), deg, unit,msg.args[0])
if float(heat[:-2]) > float(temp[:-2]):
index = ' (Heat Index: %s)' % heat
2004-01-07 16:17:53 +01:00
if temp and conds and city and state:
2004-02-08 08:16:58 +01:00
conds = conds.replace('Tsra', 'Thunderstorms')
s = 'The current temperature in %s, %s is %s%s. Conditions: %s.'% \
2004-01-07 16:17:53 +01:00
(city, state, temp, index, conds)
irc.reply(s)
2004-01-07 16:17:53 +01:00
else:
irc.error('The format of the page was odd.')
2004-01-07 16:17:53 +01:00
_cnnUrl = 'http://weather.cnn.com/weather/search?wsearch='
_fTemp = re.compile(r'(-?\d+)(&deg;)(F)</span>', re.I | re.S)
_conds = re.compile(r'align="center"><b>([^<]+)</b></div></td>', re.I|re.S)
_humidity = re.compile(r'Rel. Humidity: <b>(\d+%)</b>', re.I | re.S)
_wind = re.compile(r'Wind: <b>([^<]+)</b>', re.I | re.S)
_location = re.compile(r'<title>([^<]+)</title>', re.I | re.S)
# Certain countries are expected to use a standard abbreviation
# The weather we pull uses weird codes. Map obvious ones here.
_cnnCountryMap = {'uk': 'en', 'de': 'ge'}
def cnn(self, irc, msg, args):
"""<US zip code | US/Canada city, state | Foreign city, country>
Returns the approximate weather conditions for a given city.
"""
if len(args) > 1:
#If we received more than 1 argument, then we got a city with a
#multi-word name. ie ['Garden', 'City', 'KS'] instead of
#['Liberal', 'KS']. We join it together with a + to pass
#to our query
state = args.pop().lower()
city = ' '.join(args)
city = city.rstrip(',').lower()
if state in self._cnnCountryMap:
state = self._cnnCountryMap[state]
loc = ' '.join([city, state])
else:
#We received a single argument. Zipcode or station id.
zip = privmsgs.getArgs(args)
loc = zip.replace(',', '').lower()
url = '%s%s' % (self._cnnUrl, urllib.quote(loc))
try:
text = webutils.getUrl(url)
except webutils.WebError, e:
irc.error(str(e))
return
if "No search results" in text or "does not match a zip code" in text:
irc.error('No such location could be found.')
return
location = self._location.search(text)
temp = self._fTemp.search(text)
conds = self._conds.search(text)
humidity = self._humidity.search(text)
wind = self._wind.search(text)
convert = self.registryValue('convert', msg.args[0])
if location and temp:
location = location.group(1)
location = location.split('-')[-1].strip()
(temp, deg, unit) = temp.groups()
if convert:
temp = self._getTemp(int(temp), deg, unit, msg.args[0])
else:
temp = deg.join((temp, unit))
resp = 'The current temperature in %s is %s.' % (location, temp)
resp = [resp]
if conds is not None:
resp.append('Conditions: %s.' % conds.group(1))
if humidity is not None:
resp.append('Humidity: %s.' % humidity.group(1))
if wind is not None:
resp.append('Wind: %s.' % wind.group(1))
resp = map(utils.htmlToText, resp)
irc.reply(' '.join(resp))
else:
irc.error('Could not find weather information.')
2004-01-07 16:17:53 +01:00
2004-02-23 10:49:47 +01:00
conf.registerPlugin('Weather')
2004-02-23 12:08:22 +01:00
conf.registerChannelValue(conf.supybot.plugins.Weather, 'temperatureUnit',
2004-02-23 10:49:47 +01:00
WeatherUnit('Fahrenheit', """Sets the default temperature unit to use when
reporting the weather."""))
2004-02-25 16:46:22 +01:00
conf.registerChannelValue(conf.supybot.plugins.Weather, 'command',
2004-02-23 10:49:47 +01:00
WeatherCommand('cnn', """Sets the default command to use when retrieving
2004-03-30 10:32:17 +02:00
the weather. Command must be one of %s.""" %
utils.commaAndify(Weather.weatherCommands, And='or')))
conf.registerChannelValue(conf.supybot.plugins.Weather, 'convert',
registry.Boolean(True, """Determines whether the weather commands will
automatically convert weather units to the unit specified in
supybot.plugins.Weather.temperatureUnit."""))
2004-02-23 10:49:47 +01:00
conf.registerUserValue(conf.users.plugins.Weather, 'lastLocation',
registry.String('', ''))
2004-01-07 16:17:53 +01:00
Class = Weather
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: