[RELEASE] Movie Set Artwork Automator

PHP Code:

import re, urllib, urllib2
from BeautifulSoup import BeautifulSoup

baseurl = 'http://www.themoviedb.org'

movie_set_name = 'X-Men Collection' # this will be replaced by the script

query = baseurl + '/search/movie_collection?query=' + urllib.quote_plus( movie_set_name )
search = BeautifulSoup(urllib2.urlopen( query ).read())
search_result = search.find('ul', 'search_results')

if len(search_result.findAll('li')) == 1:
  link = search_result.a['href']
  collection = BeautifulSoup(urllib2.urlopen( baseurl + link ).read())
  title = collection.h2.a.string
  poster = collection.find('a', 'poster').find('img', 'shadow')['src']
  fanart = collection.find('div', id='backdrops').img['src']

  poster = re.sub('/w[0-9]+/', '/original/', poster)
  fanart = re.sub('/w[0-9]+/', '/original/', fanart)

  print poster # this links to the artwork! download, rename to poster.jpg and update DB
  print fanart # this links to the artwork! download, rename to fanart.jpg and update DB

else:
  print 'Error while matching collection.'

PHP Code:

import urllib, urllib2, json
import xbmc, xbmcaddon
from log import log
from database import ARTWORK_TYPE_POSTER, ARTWORK_TYPE_FANART

__addon__     = xbmcaddon.Addon( "script.moviesetart" )
download_path = xbmc.translatePath( __addon__.getSetting( "download_folder_path" ) ).decode('utf-8')

apiurl = 'http://api.themoviedb.org/3/'
apikey = '4be68d7eab1fbd1b6fd8a3b80a65a95e'
base_image_url = None

def _download_getjson( url, params=(), queryparams={} ):
  url = apiurl + url % params + '?' + _download_format_query(queryparams)
  request = urllib2.Request(url)
  request.add_header('Accept', 'application/json')
  data = json.loads(urllib2.urlopen(request).read())
  return data

def _download_base_url():
  global base_image_url
  if not base_image_url:
    config = _download_getjson( 'configuration' )
    base_image_url = config['images']['base_url'] + 'original'
  return base_image_url

def _download_format_query( queryparams ):
  query = ['api_key=' + apikey]
  for key, value in queryparams.items():
    query.append( key + '=' + urllib.quote_plus(value) )
  return '&'.join([ str(i) for i in query ])

def download( movieset, artwork_type_filenames ):
  log( 'downloader: Searching for collection: ' + movieset['label'] )
  baseurl = _download_base_url()
  artwork = {}
  
  search = _download_getjson( 'search/collection', queryparams={'query': movieset['label']} )
  if search['total_results'] == 1:
    data = search['results'][0]
    
    '''
    Might be a nice implementation later -
    present a choice to the user
    
    data = _download_getjson( 'collection/%s/images', data['id'] )
    '''
    
    if data['poster_path']:
      artwork[ARTWORK_TYPE_POSTER] = baseurl + data['poster_path']
    if data['backdrop_path']:
      artwork[ARTWORK_TYPE_FANART] = baseurl + data['backdrop_path']
  else:
    log( 'downloader: Error while matching collection: ' + movieset['label'] )
    return 0
  
  if bool(artwork):
    for artwork_type, artwork_link in artwork.items():
      log( 'downloader: Found type ' + artwork_type + ' link: ' + artwork_link )
    return artwork
  else:
    log( 'downloader: No artwork in collection: ' + movieset['label'] )
    return 0