def main():
parser = argparse.ArgumentParser()
parser.add_argument('-t',
'--title',
nargs='+',
type=str,
required=True,
help='Title of the movie')
parser.add_argument('--title_type',
type=str,
nargs='+',
choices=[
'feature', 'tv_movie', 'tv_series', 'tv_episode',
'tv_special', 'tv_miniseries', 'documentary',
'video_game', 'short', 'video', 'tv_short'
],
help='The Title type of the movie')
args = parser.parse_args()
imdb = IMDB()
if args.title_type:
IMDB.title_type = args.title_type
url = imdb.build_query_string(args.title, args.title_type)
search_result = imdb.execute_query(url)
imdb.extract_data(search_result)
imdb.build_table()
parser.add_argument("-t", "--title", type=str)
parser.add_argument("-y", "--year", type=int)
parser.add_argument("-g", "--genre", type=str,
choices=[\
'Film-Noir', 'History', 'Biography', 'Fantasy',\
'Thriller', 'Comedy', 'Horror', 'Musical',\
'Drama', 'Mystery', 'Western', 'Music',\
'Animation', 'Sport', 'Crime', 'War', 'Family',\
'Sci-Fi', 'Action', 'Adventure', 'Romance'])
args = parser.parse_args()
search_terms = args.director
if args.title:
search_terms.insert(0, args.title)
search_terms += args.actor
if args.year is not None:
search_terms.append(str(args.year))
if args.genre is not None:
search_terms.append(args.genre)
if len(search_terms) == 0:
print(
"Please provide at least one search term (movie title, director, actor, genre, or year)"
)
print("Use --help for more info")
sys.exit()
data = IMDB()
data.search(search_terms)
