class imdbPopular(object):
def __init__(self):
"""Gets a list of most popular TV series from imdb"""
self.session = helpers.make_session()
self.imdb = Imdb(session=self.session)
def fetch_popular_shows(self):
"""Get popular show information from IMDB"""
return self.imdb.get_popular_shows()
@staticmethod
def change_size(image_url, factor=3):
match = re.search(
r"^(.*)V1_(.{2})(.*?)_(.{2})(.*?),(.*?),(.*?),(.\d?)_(.*?)_.jpg$",
image_url)
if match:
matches = match.groups()
os.path.basename(image_url)
matches = list(matches)
matches[2] = int(matches[2]) * factor
matches[4] = int(matches[4]) * factor
matches[5] = int(matches[5]) * factor
matches[6] = int(matches[6]) * factor
matches[7] = int(matches[7]) * factor
return "{0}V1._{1}{2}_{3}{4},{5},{6},{7}_.jpg".format(
matches[0], matches[1], matches[2], matches[3], matches[4],
matches[5], matches[6], matches[7])
else:
return image_url
def cache_image(self, image_url):
"""
Store cache of image in cache dir
:param image_url: Source URL
"""
path = os.path.abspath(
os.path.join(settings.CACHE_DIR, 'images', 'imdb_popular'))
if not os.path.exists(path):
os.makedirs(path)
full_path = os.path.join(path, os.path.basename(image_url))
if not os.path.isfile(full_path):
helpers.download_file(image_url, full_path, session=self.session)
def query_movie(self, query, data):
imdb = Imdb()
res = ""
if query == "popularshows":
ans = imdb.get_popular_shows()
for i in range(10):
res += ans['ranks'][i]['title'] + "\n"
elif query == "popularmovies":
ans = imdb.get_popular_movies()
for i in range(10):
res += ans['ranks'][i]['title'] + "\n"
elif query == "search":
ans = imdb.search_for_title(data)
for i in range(5):
res += ans[i]['title'] + "\n"
return res
def query_movie(self, message, bot_handler: Any):
imdb = Imdb()
res = ""
query = message['content'].split()[1]
if query == "popularshows":
ans = imdb.get_popular_shows()
for i in range(10):
res += ans['ranks'][i]['title'] + "\n"
elif query == "popularmovies":
ans = imdb.get_popular_movies()
for i in range(10):
res += ans['ranks'][i]['title'] + "\n"
elif query == "search":
data = message['content'].split()[2]
ans = imdb.search_for_title(data)
for i in range(5):
res += ans[i]['title'] + "\n"
else:
res += "No movie reponse found. \n Use `@Savior help` for commands"
return res
