116 lines
4.7 KiB
Python
116 lines
4.7 KiB
Python
"""Show many information about a movie or serie"""
|
|
|
|
# PYTHON STUFFS #######################################################
|
|
|
|
import re
|
|
import urllib.parse
|
|
|
|
from bs4 import BeautifulSoup
|
|
|
|
from nemubot.exception import IMException
|
|
from nemubot.hooks import hook
|
|
from nemubot.tools import web
|
|
|
|
from nemubot.module.more import Response
|
|
|
|
|
|
# MODULE CORE #########################################################
|
|
|
|
def get_movie_by_id(imdbid):
|
|
"""Returns the information about the matching movie"""
|
|
|
|
url = "http://www.imdb.com/title/" + urllib.parse.quote(imdbid)
|
|
soup = BeautifulSoup(web.getURLContent(url))
|
|
|
|
return {
|
|
"imdbID": imdbid,
|
|
"Title": soup.body.find('h1').contents[0].strip(),
|
|
"Year": soup.body.find(id="titleYear").find("a").text.strip() if soup.body.find(id="titleYear") else ", ".join([y.text.strip() for y in soup.body.find(attrs={"class": "seasons-and-year-nav"}).find_all("a")[1:]]),
|
|
"Duration": soup.body.find(attrs={"class": "title_wrapper"}).find("time").text.strip() if soup.body.find(attrs={"class": "title_wrapper"}).find("time") else None,
|
|
"imdbRating": soup.body.find(attrs={"class": "ratingValue"}).find("strong").text.strip() if soup.body.find(attrs={"class": "ratingValue"}) else None,
|
|
"imdbVotes": soup.body.find(attrs={"class": "imdbRating"}).find("a").text.strip() if soup.body.find(attrs={"class": "imdbRating"}) else None,
|
|
"Plot": re.sub(r"\s+", " ", soup.body.find(attrs={"class": "summary_text"}).text).strip(),
|
|
|
|
"Type": "TV Series" if soup.find(id="title-episode-widget") else "Movie",
|
|
"Genre": ", ".join([x.text.strip() for x in soup.body.find(id="titleStoryLine").find_all("a") if x.get("href") is not None and x.get("href")[:21] == "/search/title?genres="]),
|
|
"Country": ", ".join([x.text.strip() for x in soup.body.find(id="titleDetails").find_all("a") if x.get("href") is not None and x.get("href")[:32] == "/search/title?country_of_origin="]),
|
|
"Credits": " ; ".join([x.find("h4").text.strip() + " " + (", ".join([y.text.strip() for y in x.find_all("a") if y.get("href") is not None and y.get("href")[:6] == "/name/"])) for x in soup.body.find_all(attrs={"class": "credit_summary_item"})]),
|
|
}
|
|
|
|
|
|
def find_movies(title, year=None):
|
|
"""Find existing movies matching a approximate title"""
|
|
|
|
title = title.lower()
|
|
|
|
# Built URL
|
|
url = "https://v2.sg.media-imdb.com/suggests/%s/%s.json" % (urllib.parse.quote(title[0]), urllib.parse.quote(title.replace(" ", "_")))
|
|
|
|
# Make the request
|
|
data = web.getJSON(url, remove_callback=True)
|
|
|
|
if "d" not in data:
|
|
return None
|
|
elif year is None:
|
|
return data["d"]
|
|
else:
|
|
return [d for d in data["d"] if "y" in d and str(d["y"]) == year]
|
|
|
|
|
|
# MODULE INTERFACE ####################################################
|
|
|
|
@hook.command("imdb",
|
|
help="View movie/serie details, using OMDB",
|
|
help_usage={
|
|
"TITLE": "Look for a movie titled TITLE",
|
|
"IMDB_ID": "Look for the movie with the given IMDB_ID",
|
|
})
|
|
def cmd_imdb(msg):
|
|
if not len(msg.args):
|
|
raise IMException("precise a movie/serie title!")
|
|
|
|
title = ' '.join(msg.args)
|
|
|
|
if re.match("^tt[0-9]{7}$", title) is not None:
|
|
data = get_movie_by_id(imdbid=title)
|
|
else:
|
|
rm = re.match(r"^(.+)\s\(([0-9]{4})\)$", title)
|
|
if rm is not None:
|
|
data = find_movies(rm.group(1), year=rm.group(2))
|
|
else:
|
|
data = find_movies(title)
|
|
|
|
if not data:
|
|
raise IMException("Movie/series not found")
|
|
|
|
data = get_movie_by_id(data[0]["id"])
|
|
|
|
res = Response(channel=msg.channel,
|
|
title="%s (%s)" % (data['Title'], data['Year']),
|
|
nomore="No more information, more at http://www.imdb.com/title/%s" % data['imdbID'])
|
|
|
|
res.append_message("%s \x02genre:\x0F %s; \x02rating\x0F: %s (%s votes); \x02plot\x0F: %s" %
|
|
(data['Type'], data['Genre'], data['imdbRating'], data['imdbVotes'], data['Plot']))
|
|
res.append_message("%s \x02from\x0F %s; %s"
|
|
% (data['Type'], data['Country'], data['Credits']))
|
|
|
|
return res
|
|
|
|
|
|
@hook.command("imdbs",
|
|
help="Search a movie/serie by title",
|
|
help_usage={
|
|
"TITLE": "Search a movie/serie by TITLE",
|
|
})
|
|
def cmd_search(msg):
|
|
if not len(msg.args):
|
|
raise IMException("precise a movie/serie title!")
|
|
|
|
data = find_movies(' '.join(msg.args))
|
|
|
|
movies = list()
|
|
for m in data:
|
|
movies.append("\x02%s\x0F%s with %s" % (m['l'], (" (" + str(m['y']) + ")") if "y" in m else "", m['s']))
|
|
|
|
return Response(movies, title="Titles found", channel=msg.channel)
|