#!/usr/bin/env python
"""
get_company.py
Usage: get_company "companyID"
Show some info about the company with the given imdbID (e.g. '0071509'
for "Columbia Pictures [us]".
"""
import sys
# Import the IMDbPY package.
try:
import imdb
except ImportError:
print 'You bad boy! You need to install the IMDbPY package!'
sys.exit(1)
if len(sys.argv) != 2:
print 'Only one argument is required:'
print ' %s "imdbID"' % sys.argv[0]
sys.exit(2)
imdbID = sys.argv[1]
i = imdb.IMDb()
out_encoding = sys.stdout.encoding or sys.getdefaultencoding()
try:
# Get a company object with the data about the company identified by
# the given imdbID.
company = i.get_company(imdbID)
except imdb.IMDbError, e:
print "Probably you're not connected to Internet. Complete error report:"
print e
sys.exit(3)
if not company:
print 'It seems that there\'s no company with imdbID "%s"' % imdbID
sys.exit(4)
# XXX: this is the easier way to print the main info about a company;
# calling the summary() method of a company object will returns a string
# with the main information about the company.
# Obviously it's not really meaningful if you want to know how
# to access the data stored in a company object, so look below; the
# commented lines show some ways to retrieve information from a
# company object.
print company.summary().encode(out_encoding, 'replace')
|