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()
