switch tfw module to use new metar parser

master
mutantmonkey 2013-01-10 22:08:52 -05:00
parent 83518a8dbc
commit 3e0738191f
2 changed files with 144 additions and 40 deletions

View File

@ -18,14 +18,16 @@ class TestTfw(unittest.TestCase):
input = Mock(group=lambda x: 'tu3jgoajgoahghqog')
tfw(self.phenny, input)
self.phenny.say.assert_called_once_with("UNKNOWN FUCKING LOCATION. Try another?")
self.phenny.say.assert_called_once_with(
"WHERE THE FUCK IS THAT? Try another location.")
def test_celsius(self):
input = Mock(group=lambda x: '24060')
tfw(self.phenny, input, celsius=True)
out = self.phenny.say.call_args[0][0]
m = re.match('^\d+°C‽ .* \- .*$', out, flags=re.UNICODE)
m = re.match('^\d+°C‽ .* \- .* \- [A-Z]{4} \d{2}:\d{2}Z$', out,
flags=re.UNICODE)
self.assertTrue(m)
def test_fahrenheit(self):
@ -33,7 +35,8 @@ class TestTfw(unittest.TestCase):
tfw(self.phenny, input, fahrenheit=True)
out = self.phenny.say.call_args[0][0]
m = re.match('^\d+°F‽ .* \- .*$', out, flags=re.UNICODE)
m = re.match('^\d+°F‽ .* \- .* \- [A-Z]{4} \d{2}:\d{2}Z$', out,
flags=re.UNICODE)
self.assertTrue(m)
def test_mev(self):
@ -41,5 +44,6 @@ class TestTfw(unittest.TestCase):
tfw(self.phenny, input)
out = self.phenny.say.call_args[0][0]
m = re.match('^[\d\.]+ meV‽ .* \- .*$', out, flags=re.UNICODE)
m = re.match('^[\d\.]+ meV‽ .* \- .* \- [A-Z]{4} \d{2}:\d{2}Z$', out,
flags=re.UNICODE)
self.assertTrue(m)

View File

@ -5,12 +5,12 @@ tfw.py - the fucking weather module
author: mutantmonkey <mutantmonkey@mutantmonkey.in>
"""
from urllib.parse import quote as urlquote
from urllib.error import HTTPError
from tools import GrumbleError
from modules import weather
import urllib.error
import random
import metar
import web
import lxml.html
import lxml.cssselect
def tfw(phenny, input, fahrenheit=False, celsius=False):
""".tfw <city/zip> - Show the fucking weather at the specified location."""
@ -18,49 +18,149 @@ def tfw(phenny, input, fahrenheit=False, celsius=False):
where = input.group(2)
if not where:
# default to Blacksburg, VA
where = "24060"
icao_code = "KBCB"
else:
icao_code = weather.code(phenny, where)
url = "http://thefuckingweather.com/?where=" + urlquote(where)
if not fahrenheit:
url += "&CELSIUS=yes"
try:
req = web.get(url)
except (HTTPError, IOError):
# the fucking weather is fucking unstable, try again
try:
req = web.get(url)
except (HTTPError, IOError):
raise GrumbleError("THE INTERNET IS FUCKING BROKEN. Please try again later.")
doc = lxml.html.fromstring(req)
try:
location = doc.get_element_by_id('locationDisplaySpan').text_content()
temp_sel = lxml.cssselect.CSSSelector('span.temperature')
temp = temp_sel(doc)[0].text_content()
temp = int(temp)
except (IndexError, KeyError):
phenny.say("UNKNOWN FUCKING LOCATION. Try another?")
if not icao_code:
phenny.say("WHERE THE FUCK IS THAT? Try another location.")
return
uri = 'http://weather.noaa.gov/pub/data/observations/metar/stations/%s.TXT'
try:
bytes = web.get(uri % icao_code)
except AttributeError:
raise GrumbleError("THE INTERNET IS FUCKING BROKEN. Please try again later.")
except urllib.error.HTTPError:
phenny.say("WHERE THE FUCK IS THAT? Try another location.")
return
if 'Not Found' in bytes:
phenny.say("WHERE THE FUCK IS THAT? Try another location.")
return
w = metar.parse(bytes)
tempf = w.temperature * 9 / 5 + 32
# add units and convert if necessary
if fahrenheit:
temp = "{0:d}°F‽".format(temp)
temp = "{0:d}°F‽".format(int(tempf))
elif celsius:
temp = "{0:d}°C‽".format(temp)
temp = "{0:d}°C‽".format(w.temperature)
else:
tempev = (temp + 273.15) * 8.617343e-5 * 1000
tempev = (w.temperature + 273.15) * 8.617343e-5 * 1000
temp = "%f meV‽" % tempev
if w.temperature < 6:
remark = "IT'S FUCKING COLD"
flavors = ["Where's the cat? Oh shit. Fluffy's frozen.",
"Nothing a few shots couldn't fix",
"Should have gone south",
"You think this is cold? Have you been to upstate New York?",
"Why do I live here?", "wang icicles.",
"Freezing my balls off out here", "Fuck this place.",
"GREAT! If you're a penguin.", "Fresh off the tap.",
"Fantastic do-nothing weather.",
"Put on some fucking socks.", "Blue balls x 2",
"Good news, food won't spoil nearly as fast outside. Bad news, who cares?",
"Really?", "Wear a fucking jacket.",
"I hear Siberia is the same this time of year.",
"NOT FUCKING JOGGING WEATHER", "Shrinkage's best friend.",
"Warmer than Hoth.", "Good baby making weather.",
"Where's a Tauntaun when you need one?",
"My nipples could cut glass", "Global Warming? Bullshit.",
"Call your local travel agency and ask them if they're serious.",
"Freezing my balls off IN here",
"I'm not sure how you can stand it", "I'm sorry.",
"Even penguins are wearing jackets.",
"Keep track of your local old people.",
"WHAT THE FUCK DO YOU MEAN IT'S NICER IN ALASKA?",
"Sock warmers are go. Everywhere.",
"Why does my car feel like a pair of ice skates?",
"Actually, a sharp-stick in the eye might not all be that bad right now.",
"THO Season.", "It's a tit-bit nipplie.",
"Anything wooden will make a good fireplace. Thank us later.",
"MOVE THE FUCK ON GOLDILOCKS",
"I'm defrosting inside of my freezer.",
"It's time for a vacation.",
"It's bone chilling cold out. Sorry ladies."]
elif w.temperature < 20:
remark = "IT'S FUCKING...ALRIGHT"
flavors = ["Might as well rain, I'm not going out in that.",
"Better than a sharp stick in the eye.",
"Everything's nice butter weather!",
"At least you aren't living in a small town in Alaska",
"It could be worse.", "FUCKING NOTHING TO SEE HERE",
"Listen, weather. We need to have a talk.",
"OH NO. THE WEATHER MACHINE IS BROKEN.",
"An Eskimo would beat your ass to be here",
"Where life is mediocre",
"Can't complain about today, but I want to!",
"Maybe inviting the inlaws over will improve today.",
"Let's go to the beach! In three months when it's nice again...",
"From inside it looks nice out.", "WHAT THE FUCK EVER",
"I love keeping the heat on for this long.",
"Inside or outside? Either way it's still today.",
"It's either only going to get better or worse from here!",
"If it's raining cats and dogs, hope you're not a pet person.",
"Today makes warm showers way nicer.",
"Here's to making your blankets feel useful.",
"I've seen better days",
"Compared to how awful it's been this is great!",
"If we go running maybe we won't notice.",
"Is that the sun outside? Why isn't it doing anything?",
"Well, at least we're not in prison.",
"Slap me around and call me Sally. It'd be an improvement.",
"Today is the perfect size, really honey.",
"Maybe Jersey Shore is on tonight."]
else:
remark = "IT'S FUCKING NICE"
flavors = ["I made today breakfast in bed.", "FUCKING SWEET",
"Quit your bitching", "Enjoy.", "IT'S ABOUT FUCKING TIME",
"READ A FUCKIN' BOOK", "LETS HAVE A FUCKING PICNIC",
"It is safe to take your ball-mittens off.", "More please.",
"uh, can we trade?", "WOO, Spring Break!",
"I can't believe it's not porn!", "I approve of this message!",
"Operation beach volleyball is go.", "Plucky ducky kinda day.",
"Today called just to say \"Hi.\"",
"STOP AND SMELL THE FUCKING ROSES",
"FUCKING NOTHING WRONG WITH TODAY", "LETS HAVE A FUCKING SOIREE",
"What would you do for a holyshititsniceout bar?",
"There are no rules today, blow shit up!",
"Celebrate Today's Day and buy your Today a present so it knows you care.",
"I feel bad about playing on my computer all day.",
"Party in the woods.", "It is now safe to leave your home.",
"PUT A FUCKING CAPE ON TODAY, BECAUSE IT'S SUPER",
"Today is like \"ice\" if it started with an \"n\". Fuck you, we don't mean nce.",
"Water park! Water drive! Just get wet!",
"The geese are on their way back! Unless you live where they migrate to for the winter.",
"FUCKING AFFABLE AS SHIT", "Give the sun a raise!",
"Today is better than an original holographic charizard. Loser!"]
remark_sel = lxml.cssselect.CSSSelector('p.remark')
remark = remark_sel(doc)[0].text_content()
if w.descriptor == "thunderstorm":
remark += " AND THUNDERING"
elif w.precipitation in ("snow", "snow grains"):
remark += " AND SNOWING"
elif w.precipitation in ("drizzle", "rain", "unknown precipitation"):
remark += " AND WET"
elif w.precipitation in ("ice crystals", "ice pellets"):
remark += " AND ICY"
elif w.precipitation in ("hail", "small hail"):
remark += " AND HAILING"
flavor_sel = lxml.cssselect.CSSSelector('p.flavor')
flavor = flavor_sel(doc)[0].text_content()
if tempf == 69:
remark = "IT'S FUCKING SEXY TIME"
flavors = ["Why is 77 better than 69? You get eight more.",
"What comes after 69? Mouthwash.",
"If you are given two contradictory orders, obey them both.",
"a good fuckin' time! ;)",
"What's the square root of 69? Eight something."]
response = "%s %s - %s - %s" % (temp, remark, flavor, location)
flavor = random.choice(flavors)
response = "{temp} {remark} - {flavor} - {location} {time}Z".format(
temp=temp, remark=remark, flavor=flavor, location=w.station,
time=w.time.strftime("%H:%M"))
phenny.say(response)
tfw.rule = (['tfw'], r'(.*)')