Remove XMG, using own MetaData class
This commit is contained in:
Ruud committed
1 parent
4de1df79ad
commit
b8fb68f05d
2 files changed
-206
No files matched your search
Whitespace-only changes.
-206
@@ -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)
|
|
||||||
Reference in new issue
Block a user