dockeruser
/
CouchPotatoServer
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

Remove XMG, using own MetaData class

pull/51/merge
Ruud 14 years ago
parent
4de1df79ad
commit
b8fb68f05d
2 changed files with 0 additions and 206 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 0
      libs/xmg/__init__.py
  2. 206
      libs/xmg/xmg.py

0
libs/xmg/__init__.py
View File

206
libs/xmg/xmg.py
View File

@ -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)
Write Preview
Loading…
Cancel
Save