def changeCategory(self):
m = ChoiceSlideout(T(32028))
m.addItem(0,T(32031))
m.addItem(-1,T(32038))
m.addItem(-2,T(32039))
for cat_id, cat in self.getCollectionCategoriesToShow(): # @UnusedVariable
m.addItem(cat_id,cat)
m.addSep()
m.addItem('refresh','[B]Update Collection[/B]')
if export.exporting():
m.addItem('export_cancel','[COLOR FFFF9999][B]Cancel Export[/B][/COLOR]')
else:
m.addItem('export','[B]Export[/B]')
m.setAutoSelectID(self.lastCategory)
ID = m.getResult()
if ID is None: return
if ID == 'refresh':
d = xbmcgui.DialogProgress()
d.create('Updating...','Getting collection data...')
try:
cats = util.getCollectionCategories(0,self)
API.refreshCollection(cats,callback=d.update)
finally:
d.close()
self.refresh(category=self.lastCategory)
elif ID == 'export':
export.exportBG(self.lastCategory)
elif ID == 'export_cancel':
export.cancel()
else:
self.refresh(category=ID)
def showReviews(self,page=0,category=None,filterChange=False,force=False):
self.setProperty('filterstring',self.filterString)
self.loadingOn()
try:
paging = (None,None)
if self.mode == 'SEARCH':
self.loadingOff()
w = openWindow(BluRaySearch,'bluray-com-search.xml',return_window=True)
self.loadingOn()
if w.canceled:
if not self.hasSearched: self.doClose()
return
if not w.keywords:
if not self.hasSearched: self.doClose()
return
results = API.search(w.keywords,w.section,w.country)
del w
if not results:
xbmcgui.Dialog().ok(T(32008),T(32009))
if self.hasSearched: return
self.doClose()
return
self.hasSearched = True
self.reviewList.reset()
elif self.mode == 'RELEASES':
results = API.getReleases()
elif self.mode == 'DEALS':
results, paging = API.getDeals(page)
elif self.mode == 'COLLECTION':
if filterChange:
results = self.currentResults
else:
try:
results = API.getCollection(categories=util.getCollectionCategories(category,self),force_refresh=force)
except bluraycomapi.LoginError, e:
error = 'Unknown'
if e.error == 'userpass': error = 'Bad Blu-ray.com name or password.'
xbmcgui.Dialog().ok('Error','Login Error:','',error)
return
if category != None: self.lastCategory = category
elif self.mode == 'PRICETRACKER':
try:
results = API.getPriceTracking()
except bluraycomapi.LoginError,e:
error = 'Unknown'
if e.error == 'userpass': error = 'Bad Blu-ray.com name or password.'
xbmcgui.Dialog().ok('Error','Login Error:','',error)
return
def _export(results=None,category=None):
util.initAPI()
if not results:
try:
cats = util.getCollectionCategories(int(category))
results = util.API.getCollection(categories=cats)
print repr(category),repr(results), repr(cats)
except util.bluraycomapi.LoginError: #TODO: Implement with notifiacation
#error = 'Unknown'
#if e.error == 'userpass': error = 'Bad Blu-ray.com name or password.'
#xbmcgui.Dialog().ok('Error','Login Error:','',error)
return
tag = u'<tag>{0}</tag>'
genre = u'<genre>{0}</genre>'
actor = u' <actor><name>{name}</name><role>{role}</role><thumb>{thumb}</thumb></actor>\n'
set_ = u'<set>{0}</set>'
path = util.getSetting('export_path')
if not path or not xbmcvfs.exists(path):
path = xbmcgui.Dialog().browse(3,util.T(32054),'files')
if not path: return
util.ADDON.setSetting('export_path',path)
sep = u'/'
if '\\' in path: sep = u'\\'
video = os.path.join(xbmc.translatePath(util.ADDON.getAddonInfo('path')).decode('utf-8'),'resources','video.mp4')
baseTags = tag.format('blu-ray.com')
if util.getSetting('export_offline_tag',True):
baseTags += tag.format('offline')
genreTable = {}
for g in util.API.genres: genreTable[g[2]] = g[1]
catTable = {}
for c in util.API.categories: catTable[c[0]] = c[1]
total = float(len(results))
progress = xbmcgui.DialogProgressBG()
progress.create(util.T(32051))
import tmdbsimple as tmdb
tmdb.API_KEY = '99ccac3e0d7fd2c7a076beea141c1057'
config = tmdb.Configuration()
config.info()
tmdbBaseImageURL = config.images['base_url'] + u'original{0}'
try:
for idx,r in enumerate(results):
progress.update(int((idx/total)*100),r.title,' ')
if canceled() or xbmc.abortRequested: return
searchTitle = r.titles[0].replace('3D','').strip()
cleanTitle = cleanFilename(r.title)
fanart = ''
#Write .strm file
f = xbmcvfs.File(path+sep+u'{0}.strm'.format(cleanTitle),'w')
f.write(video.encode('utf-8'))
f.close()
#Write .nfo file
if util.getSetting('export_write_nfo',True):
tags = baseTags
tags += tag.format(catTable.get(r.categoryID,''))
if r.is3D: tags += tag.format('3D')
genres = ''
for i in r.genreIDs:
if i in genreTable:
genres += genre.format(genreTable[i])
actors = ''
collection = ''
plot = ''
if util.getSetting('export_get_tmdb',True):
from xml.sax.saxutils import escape
progress.update(int((idx/total)*100),r.title,u'TMDB: {0}'.format(searchTitle))
search = tmdb.Search()
tryTwice(search.movie,query=searchTitle,year=r.year)
if canceled() or xbmc.abortRequested: return
if not search.results:
if ':' in searchTitle:
searchTitle = searchTitle.split(':',1)[0]
progress.update(int((idx/total)*100),r.title,u'TMDB: {0}'.format(searchTitle))
tryTwice(search.movie,query=searchTitle,year=r.year or None)
if canceled() or xbmc.abortRequested: return
if search.results:
movie = tmdb.Movies(search.results[0]['id'])
tryTwice(movie.info,append_to_response='credits')
if canceled() or xbmc.abortRequested: return
fanart = movie.backdrop_path and tmdbBaseImageURL.format(movie.backdrop_path) or ''
if movie.belongs_to_collection:
collection = set_.format(escape(movie.belongs_to_collection['name']))
if r.uniqueMovies:
bd = movie.belongs_to_collection['backdrop_path']
fanart = bd and tmdbBaseImageURL.format(bd) or fanart
for c in movie.credits.get('cast',[]):
actors += actor.format(name=escape(c['name']),role=escape(c['character']),thumb=escape(tmdbBaseImageURL.format(c['profile_path'])))
plot = movie.overview
f = xbmcvfs.File(path+sep+u'{0}.nfo'.format(cleanTitle),'w')
f.write(
NFO.format(
title=escape(r.title),
sort=escape(r.sortTitle or r.title),
rating=r.rating.split(' ',1)[-1],
#plot=r.description or r.info,
path=escape(video),
runtime=r.runtime,
plot=escape(plot),
year=r.year,
thumb=escape(r.icon.replace('_medium.','_front.')),
fanart=escape(fanart),
watched=r.watched and '0' or '',
collection=collection,
genres=genres,
actors=actors,
tags=tags
).encode('utf-8')
)
f.close()
fanartOutPath = path+sep+u'{0}-fanart.jpg'.format(cleanTitle)
if util.getSetting('export_get_fanart',True) and fanart and not xbmcvfs.exists(fanartOutPath):
progress.update(int((idx/total)*100),r.title,u'Getting fanart')
f = xbmcvfs.File(fanartOutPath,'w')
try:
r = requests.get(fanart, stream=True)
if canceled() or xbmc.abortRequested: return
if r.status_code == 200:
for chunk in r.iter_content(1024):
f.write(chunk)
finally:
f.close()
#Tag existing movies online
if util.getSetting('export_offline_tag',True) and util.getSetting('export_online_tag',True):
progress.update(100,util.T(32052))
response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.GetMovies", "params": {"properties":["tag"]}, "id": 1}')
try:
data = json.loads(response)
if 'result' in data and 'movies' in data['result']:
for i in data['result']['movies']:
tags = i['tag']
if not 'offline' in tags and not 'online' in tags:
tags.append('online')
xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.SetMovieDetails", "params":{"movieid":%s,"tag":%s},"id": 1}' % (i['movieid'],json.dumps(tags)))
except:
util.ERROR()
finally:
progress.close()
#Trigger library scan of export path
if util.getSetting('export_trigger_scan',True):
xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "VideoLibrary.Scan", "params": {"directory":"%s"}, "id": 1}' % path)
xbmcgui.Dialog().ok(util.T(32048),'',util.T(32053).format(int(total)))
