def __init__(self, key, cachedur, zippath):
if not key:
self.key = ""
else:
#Sanitize, because I can
self.key = "".join([c for c in key.split()[0] if c.isalnum()])
self.zipcache = CacheDict(duration=cachedur)
self.hostcache = CacheDict(duration=cachedur)
self.savecache = {}
self.zipfile = None
if zippath:
self.load_zips(zippath)
class Weather(object):
NAME = "weather"
PARAMS = [("str", "key"),
("int", "cachedur"),
("str", "zippath")]
HELP = "!w [-s <save_zipcode>] [query_zipcode] - Provides weather " \
"information about the location specified by query_zipcode. If " \
"unspecified, will attempt to geolocate user. Can also save a " \
"zipcode using the -s switch."
def __init__(self, key, cachedur, zippath):
if not key:
self.key = ""
else:
#Sanitize, because I can
self.key = "".join([c for c in key.split()[0] if c.isalnum()])
self.zipcache = CacheDict(duration=cachedur)
self.hostcache = CacheDict(duration=cachedur)
self.savecache = {}
self.zipfile = None
if zippath:
self.load_zips(zippath)
def load_zips(self, path):
try:
self.zipfile = open(path, "r+", 1) # r+w line buffered
for line in self.zipfile:
nick, zipcode = tuple(line.split())
if self.valid_zip(zipcode):
self.savecache[nick] = zipcode
self.zipfile.seek(0)
self.zipfile.truncate()
for nick, zipcode in six.iteritems(self.savecache):
self.zipfile.write("{0} {1}\n".format(nick, zipcode))
except:
self.zipfile = None
raise
def on_pubmsg(self, source, channel, text):
if not self.key:
return None
try:
args = self.parse_args(text)
if not args:
return None
except ArgumentError:
return "Syntax: !w [-s <save_zipcode>] [query_zipcode]"
save_zipcode = args.save_zipcode
if save_zipcode:
return self.save_zip(source.nick, save_zipcode)
zipcode = args.zipcode
if not zipcode:
if source.nick in self.savecache:
zipcode = self.savecache[source.nick]
elif source.host:
zipcode = self.get_zip(source.host)
if not zipcode:
return "Please specify zip"
return self.get_weather(zipcode)
def parse_args(self, text):
args = text.split()
if not args or args[0] not in ["!w", "!weather"]:
return None
sargs = shlex.split(" ".join(args[1:]))
parser = UndeadArgumentParser(add_help=False)
parser.add_argument("zipcode", nargs="?", default="")
parser.add_argument("-s", "--save", default="", dest="save_zipcode")
return parser.parse_args(sargs)
def save_zip(self, nick, zipcode):
if not self.valid_zip(zipcode):
return "Zip code is not in 5-digit format!"
if nick in self.savecache and self.savecache[nick] == zipcode:
return
self.savecache[nick] = zipcode
if self.zipfile:
self.zipfile.write("{0} {1}\n".format(nick, zipcode))
def valid_zip(self, zipcode):
return len(zipcode) == 5 and zipcode.isdigit()
def get_zip(self, host):
if host in self.hostcache:
return self.hostcache[host]
url = "https://freegeoip.net/json/{0}".format(host)
res = ""
try:
page = requests.get(url).text
data = json.loads(page)
res = data["zip_code"]
except:
pass
self.hostcache[host] = res
return res
def get_weather(self, zipcode):
if not self.valid_zip(zipcode):
return "Zip code is not in 5-digit format!"
if zipcode in self.zipcache:
return self.zipcache[zipcode]
url = "http://api.worldweatheronline.com/free/v1/weather.ashx?q={0}&format=json&fx=no&includelocation=yes&key={1}".format(zipcode, self.key)
res = ""
try:
page = requests.get(url).text
wdata = json.loads(page)["data"]
if "error" in wdata:
res = "No data found for {0}!".format(zipcode)
else:
area = wdata['nearest_area'][0]
weather = wdata['current_condition'][0]
city = area['areaName'][0]['value']
state = area['region'][0]['value']
tempF = weather['temp_F']
tempC = weather['temp_C']
desc = weather['weatherDesc'][0]['value']
humidity = weather['humidity']
res = "{0}, {1}: {2}. {3}°F ({4}°C). Humidity: {5}%.".format(city, state, desc, tempF, tempC, humidity)
except:
# Don't cache connection failures
return "Unable to connect to weather API!"
self.zipcache[zipcode] = res
return res
def on_poll(self, channel):
self.zipcache.prune()
self.hostcache.prune()
