def main(): parser = argparse.ArgumentParser(description = "get imdb data in the command line") parser.add_argument("-V","--version", action = "version", version = "%(prog)s version 0.1.0") group = parser.add_mutually_exclusive_group() group.add_argument("-s","--search", action = "store_true", help= "display the imdb search results for the movie") group.add_argument("-r","--rating",action = "store_true", help = "display the rating of the movie") group.add_argument("-d","--description",action = "store_true",help = "display the plot of the movie") group.add_argument("-A", "--actors",action = "store_true",help = "display the actors in the movie") parser.add_argument("-i","--ID",action = "store_true",help = "pass an imdb id instead of a title") parser.add_argument("title" ) args = parser.parse_args() try: if args.search: search(args.title) else: imdbid,title = None,None if args.ID: imdbid = args.title else: title = args.title movie = Movie(title = title, ID = imdbid) if args.rating: print "Rating: ", movie.rating elif args.description: print "Plot: ",movie.plot elif args.actors: print "Actors:",movie.actors else: movie.print_data() except Exception, e: print e
def search(title): print "Search results for %s:" %title results = Search(title) results.search() print results.format_results() choice = input(">>> ") m = Movie(ID = results.results[choice]["imdbID"]) m.print_data()