mirror of https://github.com/captn3m0/Scripts.git
terminal IMDB
This commit is contained in:
parent
ce5ac42e1b
commit
c4360a81aa
|
@ -0,0 +1,59 @@
|
|||
#!/usr/bin/python
|
||||
|
||||
import urllib
|
||||
import argparse
|
||||
from xml.etree import ElementTree
|
||||
import sys
|
||||
|
||||
def retrieveMovie (title):
|
||||
title=urllib.quote(title.encode("utf8"))
|
||||
URL="http://www.omdbapi.com/?r=xml&plot=full&t=%s" % title
|
||||
xml = ElementTree.parse(urllib.urlopen(URL))
|
||||
#fall back to movie search if no movie is found
|
||||
for A in xml.iter('root'):
|
||||
if (A.get('response')=='False'):
|
||||
print "Movie not found!"
|
||||
sys.exit()
|
||||
xml=xml.getroot()
|
||||
printInfo(xml)
|
||||
return xml
|
||||
|
||||
#Search for movie and return plot for all the results
|
||||
def movieSearch (title):
|
||||
title=urllib.quote(title.encode("utf8"))
|
||||
URL="http://www.omdbapi.com/?r=xml&s=%s" % title
|
||||
xml = ElementTree.parse(urllib.urlopen(URL))
|
||||
xml=xml.getroot()
|
||||
for B in xml.findall('Movie'):
|
||||
apicall=retrieveMovie(B.get('Title'))
|
||||
printInfo(apicall)
|
||||
return xml
|
||||
|
||||
def printInfo(xml):
|
||||
for B in xml.findall('movie'):
|
||||
print "\n%s (%s) || %s || %s\n" %(B.get('title'),B.get('year'),
|
||||
B.get('runtime'),B.get('imdbRating'))
|
||||
print "Director: %s\nActors: %s\n" %(B.get('director'),B.get('actors'))
|
||||
print "%s\n" %(B.get('plot'))
|
||||
|
||||
if __name__=='__main__':
|
||||
parser=argparse.ArgumentParser(description='Command-Line Interface for the IMdB')
|
||||
|
||||
parser.add_argument("-t", help="Search by title. Return first result")
|
||||
parser.add_argument("-s", help="Search and return results")
|
||||
|
||||
args=parser.parse_args()
|
||||
choices=["None"]
|
||||
try:
|
||||
choices[0]=sys.argv[1]
|
||||
title=sys.argv[2]
|
||||
except:
|
||||
parser.print_usage()
|
||||
sys.exit()
|
||||
|
||||
if choices[0]=="-t":
|
||||
retrieveMovie(title)
|
||||
else:
|
||||
movieSearch(title)
|
||||
|
||||
|
Loading…
Reference in New Issue