Ruud
14 years ago
2 changed files with 0 additions and 206 deletions
@ -1,206 +0,0 @@ |
|||||
import json |
|
||||
import os |
|
||||
import urllib2 |
|
||||
|
|
||||
__author__ = 'Therms' |
|
||||
__tmdb_apikey__ = '6d96a9efb4752ed0d126d94e12e52036' |
|
||||
|
|
||||
class XmgException(Exception): |
|
||||
pass |
|
||||
|
|
||||
class ApiError(XmgException): |
|
||||
pass |
|
||||
|
|
||||
class IdError(XmgException): |
|
||||
pass |
|
||||
|
|
||||
class NfoError(XmgException): |
|
||||
pass |
|
||||
|
|
||||
class MetaGen(): |
|
||||
def __init__(self, imdbid, imdbpy = None): |
|
||||
''' metagen is used to download metadata for a movie or tv show and then create |
|
||||
the necessary files for the media to be imported into XBMC. |
|
||||
|
|
||||
Arguments |
|
||||
=========== |
|
||||
fanart/poster_height/width_min: Sets lowest acceptable image resolution. 0 means |
|
||||
disregard. If no fanart available at specified resolution or greater, then |
|
||||
we disregard this setting, and download highest resolution that is available. |
|
||||
|
|
||||
name*: In the case of a movie, ideally this should be the full movie name |
|
||||
followed by the year of the movie in parentheses. e.g. "The Matrix (1999)". |
|
||||
If this is specific enough to generate only one search result then we'll |
|
||||
continue. Otherwise, we'll raise IdError. |
|
||||
|
|
||||
Because of the imprecise nature of this method of id, only use it if you |
|
||||
don't have the imdb_id or tmdb_id |
|
||||
|
|
||||
imdb_id: Use this argument if you know the imdb id of the show/movie. If |
|
||||
this is used, the tmdb_id argument is ignored. |
|
||||
|
|
||||
tmdb_id*: Use this argument if you know the tmdb id of the movie. If this |
|
||||
is used, the imdb_id argument is ignored. |
|
||||
|
|
||||
imdbpy: When xmg is used as a library, imdbpy may not be installed |
|
||||
system-wide, but included with your application. If this is the case, pass |
|
||||
your instance of imdb.IMDb() to metagen, so we can use it. |
|
||||
|
|
||||
* These arguments are not yet supported. |
|
||||
|
|
||||
''' |
|
||||
|
|
||||
|
|
||||
if imdbid[:2].lower() == 'tt': |
|
||||
self.imdbid = imdbid[2:] |
|
||||
else: |
|
||||
self.imdbid = imdbid |
|
||||
|
|
||||
self.nfo_string = 'http://www.imdb.com/title/' + imdbid + '/' |
|
||||
self.tmdb_data = self._get_tmdb_imdb() |
|
||||
self._validate_tmdb_json() |
|
||||
|
|
||||
#TODO: Search by movie name |
|
||||
#TODO: Search by tmdb_id |
|
||||
#TODO: Search by movie hash |
|
||||
|
|
||||
|
|
||||
def _validate_tmdb_json(self): |
|
||||
try: |
|
||||
_ = self._get_fanart(0,0) |
|
||||
except: |
|
||||
try: |
|
||||
_ = self._get_poster(0,0) |
|
||||
except: |
|
||||
raise ApiError("Unknown TMDB data format: %s" % self.tmdb_data) |
|
||||
|
|
||||
def write_nfo(self, path): |
|
||||
try: |
|
||||
f = open(path, 'w') |
|
||||
f.write(self.nfo_string) |
|
||||
f.close() |
|
||||
except: |
|
||||
raise NfoError("Couldn't write nfo") |
|
||||
|
|
||||
def _get_fanart(self, min_height, min_width): |
|
||||
''' Fetches the fanart for the specified imdb_id and saves it to dir. |
|
||||
Arguments |
|
||||
|
|
||||
min_height/width: Sets lowest acceptable resolution fanart. 0 means |
|
||||
disregard. If no fanart available at specified resolution or greater, then |
|
||||
we disregard. |
|
||||
''' |
|
||||
images = [image['image'] for image in self.tmdb_data['backdrops'] if image['image'].get('size') == 'original'] |
|
||||
if len(images) == 0: |
|
||||
raise ApiError("No fanart") |
|
||||
|
|
||||
return self._get_image(images, min_height, min_width) |
|
||||
|
|
||||
def get_fanart_url(self, min_height, min_width): |
|
||||
return self._get_fanart(min_height, min_width)['url'] |
|
||||
|
|
||||
def write_fanart(self, filename_root, path, min_height, min_width): |
|
||||
fanart_url = self.get_fanart_url(min_height, min_width) |
|
||||
#fetch and write to disk |
|
||||
dest = os.path.join(path, filename_root) |
|
||||
try: |
|
||||
f = open(dest, 'wb') |
|
||||
except: |
|
||||
raise IOError("Can't open for writing: %s" % dest) |
|
||||
|
|
||||
response = urllib2.urlopen(fanart_url) |
|
||||
f.write(response.read()) |
|
||||
f.close() |
|
||||
|
|
||||
return True |
|
||||
|
|
||||
def _get_poster(self, min_height, min_width): |
|
||||
''' Fetches the poster for the specified imdb_id and saves it to dir. |
|
||||
Arguments |
|
||||
|
|
||||
min_height/width: Sets lowest acceptable resolution poster. 0 means |
|
||||
disregard. If no poster available at specified resolution or greater, then |
|
||||
we disregard. |
|
||||
''' |
|
||||
images = [image['image'] for image in self.tmdb_data['posters'] if image['image'].get('size') == 'original'] |
|
||||
if len(images) == 0: |
|
||||
raise ApiError("No posters") |
|
||||
|
|
||||
return self._get_image(images, min_height, min_width) |
|
||||
|
|
||||
def get_poster_url(self, min_height, min_width): |
|
||||
return self._get_poster(min_height, min_width)['url'] |
|
||||
|
|
||||
def write_poster(self, filename_root, path, min_height, min_width): |
|
||||
poster_url = self.get_poster_url(min_height, min_width) |
|
||||
dest = os.path.join(path, filename_root) |
|
||||
|
|
||||
try: |
|
||||
f = open(dest, 'wb') |
|
||||
except: |
|
||||
raise IOError("Can't open for writing: %s" % dest) |
|
||||
|
|
||||
response = urllib2.urlopen(poster_url) |
|
||||
f.write(response.read()) |
|
||||
f.close() |
|
||||
|
|
||||
return True |
|
||||
|
|
||||
def _get_tmdb_imdb(self): |
|
||||
url = "http://api.themoviedb.org/2.1/Movie.imdbLookup/en/json/%s/%s" % (__tmdb_apikey__, "tt" + self.imdbid) |
|
||||
|
|
||||
count = 0 |
|
||||
while 1: |
|
||||
count += 1 |
|
||||
response = urllib2.urlopen(url) |
|
||||
json_string = response.read() |
|
||||
try: |
|
||||
tmdb_data = json.loads(json_string)[0] |
|
||||
return tmdb_data |
|
||||
except ValueError, e: |
|
||||
if count < 3: |
|
||||
continue |
|
||||
else: |
|
||||
raise ApiError("Invalid JSON: %s: %s" % (e, json_string)) |
|
||||
except: |
|
||||
ApiError("JSON error with: %s" % json_string) |
|
||||
|
|
||||
|
|
||||
def _get_image(self, image_list, min_height, min_width): |
|
||||
#Select image |
|
||||
images = [] |
|
||||
for image in image_list: |
|
||||
if not min_height or min_width: |
|
||||
images.append(image) |
|
||||
break |
|
||||
elif min_height and not min_width: |
|
||||
if image['height'] >= min_height: |
|
||||
images.append(image) |
|
||||
break |
|
||||
elif min_width and not min_height: |
|
||||
if image['width'] >= min_width: |
|
||||
images.append(image) |
|
||||
break |
|
||||
elif min_width and min_height: |
|
||||
if image['width'] >= min_width and image['height'] >= min_height: |
|
||||
images.append(image) |
|
||||
break |
|
||||
|
|
||||
#No image meets our resolution requirements, so disregard those requirements |
|
||||
if len(images) == 0 and min_height or min_width: |
|
||||
images.append(image_list[0]) |
|
||||
|
|
||||
return images[0] |
|
||||
|
|
||||
|
|
||||
if __name__ == "__main__": |
|
||||
import sys |
|
||||
try: |
|
||||
id = sys.argv[1] |
|
||||
except: |
|
||||
id = 'tt0111161' |
|
||||
|
|
||||
x = MetaGen(id) |
|
||||
x.write_nfo(".\movie.nfo") |
|
||||
x.write_fanart("fanart", ".", 0, 0) |
|
||||
x.write_poster("movie", ".", 0, 0) |
|
||||
Loading…
Reference in new issue