diff options
Diffstat (limited to 'imdb-lookup/imdbinfo.py')
| -rwxr-xr-x | imdb-lookup/imdbinfo.py | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/imdb-lookup/imdbinfo.py b/imdb-lookup/imdbinfo.py index fe4bd60..15942f4 100755 --- a/imdb-lookup/imdbinfo.py +++ b/imdb-lookup/imdbinfo.py @@ -8,6 +8,7 @@ import re import json import time import argparse +import locale import math import shutil import logging @@ -102,6 +103,11 @@ class TMDBCache(object): raise Exception("Failed to query movie-id {id}: {reason}".format( id=movie_id, reason=str(e))) + def tmdb_search(self, query): + results = tmdb.Search().movie(query=query)["results"] + results = map(lambda i: self.infos(str(i['id'])), results) + return list(results) + def alternative_title(self, movie_id, locale): """Returns the title in selected locale or original title otherwise""" try: @@ -418,6 +424,65 @@ class HelpAction(argparse._HelpAction): print(formatter.format_help()) parser.exit(0) +def do_rename(args): + def filter_filename(filename): + """remove possibly everything except the movie name""" + stopwords = [ + "dvd", "ac3", "r5", "unrated", "ts", "720p", "md", + "ts", "ld", "bdrip", "tvrip", "dvdrip", "dvdscr", "uncut", + "german", "english", "telesync", "20[0-1][0-9]|19[0-9][0-9]", + "x264", "hdtv", "ws" + ] + def findword(word): + return re.findall(u"(.*?)\.?" + word + u"[\.\- ]", filename, re.IGNORECASE) + matches = [i for i in map(findword, stopwords) if i!=[]] + matches.sort() + if len(matches) > 0: + name = matches[0][0] + else: + name = filename + return name.replace(u".", u" ") + + def substitute_filename(filename): + """remove potentially harmful characters from filename""" + return re.sub(u"[^a-zA-Z0-9()\- _#]", u"_", filename) + + from pprint import pprint + countrycode = locale.getdefaultlocale()[0][3:5] + db = Protector(TMDBCache()) + search_term = filter_filename(args.file[0]) + print("Search for: {}".format(search_term)) + results = db.tmdb_search(search_term) + results.sort(key=lambda i:i["popularity"],reverse=True) + results = filter(lambda i: i["imdb_id"] and i["release_date"], results) + results = list(results) + + for (i,result) in zip(range(len(results)), results): + result["i"] = i+1 + result["year"] = result["release_date"].split("-")[0] + print("{i} {title} ({year}) #{imdb_id}".format(**result)) + result["alternative_title"] = db.alternative_title(result["imdb_id"], locale=countrycode) + if result["alternative_title"] != result["title"]: + print(" a.k.a.: {}".format(result["alternative_title"])) + + select = None + while not select: + try: + select = int(input("Choose: ")) + except ValueError: + pass + + filename = "{alternative_title} ({year}) #{imdb_id}".format(**results[select-1]) + filename = substitute_filename(filename) + + if os.path.exists(args.file[0]): + choice = input("Move '{}' -> '{}' ? [y/N]: ".format(args.file[0], filename)).lower() + if choice == "y": + os.mkdir(filename) + shutil.move(args.file[0], filename) + else: + print("mkdir '{}'".format(filename)) + os.mkdir(filename) def do_test(args): import doctest @@ -472,6 +537,12 @@ if __name__ == "__main__": parser_index.add_argument("files", action="append", nargs="+", help="Files containing distinct movie-ids") + parser_rename = subparsers.add_parser("search", add_help=False, + help="Search supported rename") + parser_rename.set_defaults(func=do_rename) + parser_rename.add_argument("file", metavar="PATH", nargs=1, + help="File or folder to rename into tagged-imdb format") + parser_test = subparsers.add_parser("test", add_help=False, help="Run testsuite") parser_test.set_defaults(func=do_test) |