switch tfw module to use new metar parser
mutantmonkey
parent
83518a8dbc
commit
3e0738191f
|
|
@ -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)
|
||||
|
|
|
|||
172
modules/tfw.py
172
modules/tfw.py
|
|
@ -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
|
||||
|
||||
remark_sel = lxml.cssselect.CSSSelector('p.remark')
|
||||
remark = remark_sel(doc)[0].text_content()
|
||||
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!"]
|
||||
|
||||
flavor_sel = lxml.cssselect.CSSSelector('p.flavor')
|
||||
flavor = flavor_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"
|
||||
|
||||
response = "%s %s - %s - %s" % (temp, remark, flavor, location)
|
||||
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."]
|
||||
|
||||
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'(.*)')
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue