nemunaire/nemubot
1
0
Fork 0
Code Releases Activity

IMDB module: detect IMDBid

Browse Source
This commit is contained in:
nemunaire 2014-08-26 16:38:30 +02:00
parent 0e5562fa01
commit 84d3ee262c
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
modules/imdb.py
View File

@ -1,7 +1,8 @@
# coding=utf-8 # coding=utf-8
import urllib.request
import json import json
import re
import urllib.request
from hooks import hook from hooks import hook
@ -18,20 +19,22 @@ def cmd_imdb(msg):
if len(msg.cmds) < 2: if len(msg.cmds) < 2:
raise IRCException("precise a movie/serie title!") raise IRCException("precise a movie/serie title!")
url = "http://www.omdbapi.com/?t=%s" % urllib.parse.quote(' '.join(msg.cmds[1:])) title = ' '.join(msg.cmds[1:])
if re.match("^tt[0-9]{7}$", title) is not None:
url = "http://www.omdbapi.com/?i=%s" % urllib.parse.quote(title)
else:
url = "http://www.omdbapi.com/?t=%s" % urllib.parse.quote(title)
print_debug(url) print_debug(url)
response = urllib.request.urlopen(url) response = urllib.request.urlopen(url)
data = json.loads(response.read().decode()) data = json.loads(response.read().decode())
if "Error" in data:
url = "http://www.omdbapi.com/?i=%s" % urllib.parse.quote(' '.join(msg.cmds[1:]))
response = urllib.request.urlopen(url)
data = json.loads(response.read().decode())
if "Error" in data: if "Error" in data:
raise IRCException(data["Error"]) raise IRCException(data["Error"])
if "Response" in data and data["Response"] == "True": elif "Response" in data and data["Response"] == "True":
res = Response(msg.sender, channel=msg.channel, res = Response(msg.sender, channel=msg.channel,
title="%s (%s)" % (data['Title'], data['Year']), title="%s (%s)" % (data['Title'], data['Year']),
nomore="No more information, more at http://www.imdb.com/title/%s" % data['imdbID']) nomore="No more information, more at http://www.imdb.com/title/%s" % data['imdbID'])