summaryrefslogtreecommitdiff
path: root/modules
diff options
context:
space:
mode:
authorJon Bergli Heier <snakebite@jvnv.net>2010-06-09 01:10:24 +0200
committerJon Bergli Heier <snakebite@jvnv.net>2010-06-09 01:10:24 +0200
commit1b890679774d64ff83b12fe5e3bf44401d3b2467 (patch)
treedf4eab6100cb57e3796a2b5dfa15e0bbf1c0a86e /modules
parent7cec7c641da41c3c456fcaab07046b84d49566ef (diff)
modules: Added a TVRage.com parser along with a rfc3339 parser and formatter.
Diffstat (limited to 'modules')
-rw-r--r--modules/tvrage.py79
1 files changed, 79 insertions, 0 deletions
diff --git a/modules/tvrage.py b/modules/tvrage.py
new file mode 100644
index 0000000..cd8ee8f
--- /dev/null
+++ b/modules/tvrage.py
@@ -0,0 +1,79 @@
+info = {
+ 'author': 'Jon Bergli Heier',
+ 'title': 'TVRage',
+ 'description': 'TVRage.com feed parser',
+}
+
+import urllib, urllib2, datetime, rfc3339, pytz
+from lxml import etree
+
+class Module:
+ def __init__(self, bot):
+ self.irc = bot
+ if self.irc:
+ self.irc.register_keyword('!tv', self)
+
+ def find_show(self, search):
+ try:
+ u = urllib2.urlopen('http://services.tvrage.com/tools/quickinfo.php?%s' % urllib.urlencode({'show': search}))
+ except:
+ return 'Could not fetch show data from TVRage.'
+
+ rawdata = u.read()
+ if rawdata.startswith('No Show Results'):
+ return rawdata.strip()
+
+ # Why is there a pre-tag here in the first place?
+ if rawdata.startswith('<pre>'):
+ rawdata = rawdata[5:]
+
+ data = {}
+ for line in rawdata.strip().split('\n'):
+ key, values = line.split('@')
+ values = values.split('^')
+ data[key] = values
+
+ status = data['Status'][0]
+ if 'Ended' in status or 'Canceled' in status:
+ return '\002%s\002 does not currently air.' % data['Show Name'][0]
+
+ # TODO: Fetch this from somewhere user-configurable
+ local_tz = pytz.timezone('Europe/Oslo')
+
+ airdate = data['RFC3339'][0]
+ if len(airdate) == 24: # Assume missing 0 in timezone
+ airdate = airdate[:20] + '0' + airdate[20:]
+ airdate = rfc3339.parse_datetime(airdate)
+
+ # Convert airdate to our local timezone
+ airdate = airdate.astimezone(local_tz)
+
+ # Localize utcnow() as UTC
+ now = pytz.utc.localize(datetime.datetime.utcnow())
+
+ eta = airdate - now
+ # Get rid of microseconds
+ eta = datetime.timedelta(eta.days, eta.seconds)
+
+
+ aired = eta.days < 0 or eta.seconds < 0
+
+ if aired:
+ return '\002%s\002 aired on \002%s\002' % (data['Show Name'][0], airdate.strftime('%d.%m.%Y %H:%M %Z'))
+ else:
+ return '\002%s\002 %s airs on \002%s\002 (eta: %s)' % (data['Show Name'][0],
+ data['Next Episode'][1],
+ airdate.strftime('%d.%m.%Y %H:%M %Z'),
+ eta)
+
+ def keyword(self, nick, channel, kw, msg):
+ target = channel if not channel == self.irc.nickname else nick.split('!')[0]
+ args = msg.split()
+ if len(args) == 0:
+ self.irc.msg(target, 'Usage: !tv search')
+ return
+
+if __name__ == '__main__':
+ import sys
+ m = Module(None)
+ print m.find_show(' '.join(sys.argv[1:]))