417 lines
17 KiB
Python
417 lines
17 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
from __future__ import absolute_import, division, unicode_literals
|
|
from logging import getLogger
|
|
import Queue
|
|
import copy
|
|
|
|
import xbmcgui
|
|
|
|
from cProfile import Profile
|
|
from pstats import Stats
|
|
from StringIO import StringIO
|
|
|
|
from .get_metadata import GetMetadataTask, reset_collections
|
|
from . import common, sections
|
|
from .. import utils, timing, backgroundthread, variables as v, app
|
|
from .. import plex_functions as PF, itemtypes
|
|
from ..plex_db import PlexDB
|
|
|
|
if (v.PLATFORM != 'Microsoft UWP' and
|
|
utils.settings('enablePlaylistSync') == 'true'):
|
|
# Xbox cannot use watchdog, a dependency for PKC playlist features
|
|
from .. import playlists
|
|
PLAYLIST_SYNC_ENABLED = True
|
|
else:
|
|
PLAYLIST_SYNC_ENABLED = False
|
|
|
|
LOG = getLogger('PLEX.sync.full_sync')
|
|
# How many items will be put through the processing chain at once?
|
|
BATCH_SIZE = 2000
|
|
# Safety margin to filter PMS items - how many seconds to look into the past?
|
|
UPDATED_AT_SAFETY = 60 * 5
|
|
LAST_VIEWED_AT_SAFETY = 60 * 5
|
|
|
|
|
|
def tag_last(iterable):
|
|
"""
|
|
Given some iterable, returns (last, item), where last is only True if you
|
|
are on the final iteration.
|
|
"""
|
|
iterator = iter(iterable)
|
|
gotone = False
|
|
try:
|
|
lookback = next(iterator)
|
|
gotone = True
|
|
while True:
|
|
cur = next(iterator)
|
|
yield False, lookback
|
|
lookback = cur
|
|
except StopIteration:
|
|
if gotone:
|
|
yield True, lookback
|
|
raise StopIteration()
|
|
|
|
|
|
class InitNewSection(object):
|
|
"""
|
|
Throw this into the queue used for ProcessMetadata to tell it which
|
|
Plex library section we're looking at
|
|
"""
|
|
def __init__(self, context, total_number_of_items, section_name,
|
|
section_id, plex_type):
|
|
self.context = context
|
|
self.total = total_number_of_items
|
|
self.name = section_name
|
|
self.id = section_id
|
|
self.plex_type = plex_type
|
|
|
|
|
|
class FullSync(common.libsync_mixin):
|
|
def __init__(self, repair, callback, show_dialog):
|
|
"""
|
|
repair=True: force sync EVERY item
|
|
"""
|
|
self._canceled = False
|
|
self.repair = repair
|
|
self.callback = callback
|
|
self.queue = None
|
|
self.process_thread = None
|
|
self.current_sync = None
|
|
self.plexdb = None
|
|
self.plex_type = None
|
|
self.section_type = None
|
|
self.worker_count = int(utils.settings('syncThreadNumber'))
|
|
self.item_count = 0
|
|
# For progress dialog
|
|
self.show_dialog = show_dialog
|
|
self.show_dialog_userdata = utils.settings('playstate_sync_indicator') == 'true'
|
|
self.dialog = None
|
|
self.total = 0
|
|
self.current = 0
|
|
self.processed = 0
|
|
self.title = ''
|
|
self.section = None
|
|
self.section_name = None
|
|
self.install_sync_done = utils.settings('SyncInstallRunDone') == 'true'
|
|
self.threader = backgroundthread.ThreaderManager(
|
|
worker=backgroundthread.NonstoppingBackgroundWorker,
|
|
worker_count=self.worker_count)
|
|
super(FullSync, self).__init__()
|
|
|
|
def update_progressbar(self):
|
|
if self.show_dialog:
|
|
try:
|
|
progress = int(float(self.current) / float(self.total) * 100.0)
|
|
except ZeroDivisionError:
|
|
progress = 0
|
|
self.dialog.update(progress,
|
|
'%s (%s)' % (self.section_name, self.section_type_text),
|
|
'%s/%s %s'
|
|
% (self.current, self.total, self.title))
|
|
|
|
def process_item(self, xml_item):
|
|
"""
|
|
Processes a single library item
|
|
"""
|
|
plex_id = int(xml_item.get('ratingKey'))
|
|
if not self.repair and self.plexdb.checksum(plex_id, self.plex_type) == \
|
|
int('%s%s' % (plex_id,
|
|
xml_item.get('updatedAt',
|
|
xml_item.get('addedAt', 1541572987)))):
|
|
return
|
|
self.threader.addTask(GetMetadataTask(self.queue,
|
|
plex_id,
|
|
self.plex_type,
|
|
self.get_children))
|
|
self.item_count += 1
|
|
|
|
def update_library(self):
|
|
LOG.debug('Writing changes to Kodi library now')
|
|
i = 0
|
|
if not self.section:
|
|
self.section = self.queue.get()
|
|
self.queue.task_done()
|
|
while not self.isCanceled() and self.item_count > 0:
|
|
section = self.section
|
|
if not section:
|
|
break
|
|
LOG.debug('Start or continue processing section %s (%ss)',
|
|
section.name, section.plex_type)
|
|
self.processed = 0
|
|
self.total = section.total
|
|
self.section_name = section.name
|
|
self.section_type_text = utils.lang(
|
|
v.TRANSLATION_FROM_PLEXTYPE[section.plex_type])
|
|
with section.context(self.current_sync) as context:
|
|
while not self.isCanceled() and self.item_count > 0:
|
|
try:
|
|
item = self.queue.get(block=False)
|
|
except backgroundthread.Queue.Empty:
|
|
if self.threader.threader.working():
|
|
app.APP.monitor.waitForAbort(0.02)
|
|
continue
|
|
else:
|
|
# Try again, in case a thread just finished
|
|
i += 1
|
|
if i == 3:
|
|
break
|
|
continue
|
|
i = 0
|
|
self.queue.task_done()
|
|
if isinstance(item, dict):
|
|
context.add_update(item['xml'][0],
|
|
section_name=section.name,
|
|
section_id=section.id,
|
|
children=item['children'])
|
|
self.title = item['xml'][0].get('title')
|
|
self.processed += 1
|
|
elif isinstance(item, InitNewSection) or item is None:
|
|
self.section = item
|
|
break
|
|
else:
|
|
raise ValueError('Unknown type %s' % type(item))
|
|
self.item_count -= 1
|
|
self.current += 1
|
|
self.update_progressbar()
|
|
if self.processed == 500:
|
|
self.processed = 0
|
|
context.commit()
|
|
LOG.debug('Done writing changes to Kodi library')
|
|
|
|
@utils.log_time
|
|
def addupdate_section(self, section):
|
|
LOG.debug('Processing library section for new or changed items %s',
|
|
section)
|
|
if not self.install_sync_done:
|
|
app.SYNC.path_verified = False
|
|
try:
|
|
# Sync new, updated and deleted items
|
|
iterator = section['iterator']
|
|
# Tell the processing thread about this new section
|
|
queue_info = InitNewSection(section['context'],
|
|
iterator.total,
|
|
iterator.get('librarySectionTitle'),
|
|
section['section_id'],
|
|
section['plex_type'])
|
|
self.queue.put(queue_info)
|
|
last = True
|
|
# To keep track of the item-number in order to kill while loops
|
|
self.item_count = 0
|
|
self.current = 0
|
|
while True:
|
|
# Check Plex DB to see what we need to add/update
|
|
with PlexDB() as self.plexdb:
|
|
for i, (last, xml_item) in enumerate(tag_last(iterator)):
|
|
if self.isCanceled():
|
|
return False
|
|
self.process_item(xml_item)
|
|
if self.item_count == BATCH_SIZE:
|
|
break
|
|
# Make sure Plex DB above is closed before adding/updating
|
|
if self.item_count == BATCH_SIZE:
|
|
self.update_library()
|
|
if last:
|
|
break
|
|
self.update_library()
|
|
reset_collections()
|
|
return True
|
|
except RuntimeError:
|
|
LOG.error('Could not entirely process section %s', section)
|
|
return False
|
|
|
|
@utils.log_time
|
|
def playstate_per_section(self, section):
|
|
LOG.debug('Processing playstate for library section %s', section)
|
|
try:
|
|
# Sync new, updated and deleted items
|
|
iterator = section['iterator']
|
|
# Tell the processing thread about this new section
|
|
queue_info = InitNewSection(section['context'],
|
|
iterator.total,
|
|
section['section_name'],
|
|
section['section_id'],
|
|
section['plex_type'])
|
|
self.queue.put(queue_info)
|
|
self.total = iterator.total
|
|
self.section_name = section['section_name']
|
|
self.section_type_text = utils.lang(
|
|
v.TRANSLATION_FROM_PLEXTYPE[section['plex_type']])
|
|
self.current = 0
|
|
with section['context'](self.current_sync) as itemtype:
|
|
for xml_item in iterator:
|
|
if self.isCanceled():
|
|
return False
|
|
itemtype.update_userdata(xml_item, section['plex_type'])
|
|
itemtype.plexdb.update_last_sync(int(xml_item.attrib['ratingKey']),
|
|
section['plex_type'],
|
|
self.current_sync)
|
|
self.current += 1
|
|
self.update_progressbar()
|
|
return True
|
|
except RuntimeError:
|
|
LOG.error('Could not entirely process section %s', section)
|
|
return False
|
|
|
|
def threaded_get_iterators(self, kinds, queue, updated_at=None,
|
|
last_viewed_at=None):
|
|
"""
|
|
PF.SectionItems is costly, so let's do it asynchronous
|
|
"""
|
|
if self.repair:
|
|
updated_at = None
|
|
last_viewed_at = None
|
|
else:
|
|
updated_at = updated_at - UPDATED_AT_SAFETY if updated_at else None
|
|
last_viewed_at = last_viewed_at - LAST_VIEWED_AT_SAFETY \
|
|
if last_viewed_at else None
|
|
try:
|
|
for kind in kinds:
|
|
for section in (x for x in sections.SECTIONS
|
|
if x['plex_type'] == kind[1]):
|
|
if self.isCanceled():
|
|
return
|
|
element = copy.deepcopy(section)
|
|
element['section_type'] = element['plex_type']
|
|
element['plex_type'] = kind[0]
|
|
element['element_type'] = kind[1]
|
|
element['context'] = kind[2]
|
|
element['get_children'] = kind[3]
|
|
element['iterator'] = PF.SectionItems(section['section_id'],
|
|
plex_type=kind[0],
|
|
updated_at=updated_at,
|
|
last_viewed_at=last_viewed_at)
|
|
queue.put(element)
|
|
finally:
|
|
queue.put(None)
|
|
|
|
def full_library_sync(self):
|
|
"""
|
|
"""
|
|
kinds = [
|
|
(v.PLEX_TYPE_MOVIE, v.PLEX_TYPE_MOVIE, itemtypes.Movie, False),
|
|
(v.PLEX_TYPE_SHOW, v.PLEX_TYPE_SHOW, itemtypes.Show, False),
|
|
(v.PLEX_TYPE_SEASON, v.PLEX_TYPE_SHOW, itemtypes.Season, False),
|
|
(v.PLEX_TYPE_EPISODE, v.PLEX_TYPE_SHOW, itemtypes.Episode, False)
|
|
]
|
|
if app.SYNC.enable_music:
|
|
kinds.extend([
|
|
(v.PLEX_TYPE_ARTIST, v.PLEX_TYPE_ARTIST, itemtypes.Artist, False),
|
|
(v.PLEX_TYPE_ALBUM, v.PLEX_TYPE_ARTIST, itemtypes.Album, True),
|
|
])
|
|
# ADD NEW ITEMS
|
|
# Already start setting up the iterators. We need to enforce
|
|
# syncing e.g. show before season before episode
|
|
iterator_queue = Queue.Queue()
|
|
updated_at = int(utils.settings('lastfullsync')) or None
|
|
task = backgroundthread.FunctionAsTask(self.threaded_get_iterators,
|
|
None,
|
|
kinds,
|
|
iterator_queue,
|
|
updated_at=updated_at)
|
|
backgroundthread.BGThreader.addTask(task)
|
|
while True:
|
|
section = iterator_queue.get()
|
|
iterator_queue.task_done()
|
|
if section is None:
|
|
break
|
|
# Setup our variables
|
|
self.plex_type = section['plex_type']
|
|
self.section_type = section['section_type']
|
|
self.context = section['context']
|
|
self.get_children = section['get_children']
|
|
# Now do the heavy lifting
|
|
if self.isCanceled() or not self.addupdate_section(section):
|
|
return False
|
|
common.update_kodi_library(video=True, music=True)
|
|
# SYNC PLAYSTATE of ALL items (otherwise we won't pick up on items that
|
|
# were set to unwatched). Also mark all items on the PMS to be able
|
|
# to delete the ones still in Kodi
|
|
LOG.info('Start synching playstate and userdata for every item')
|
|
self.threader.shutdown()
|
|
self.threader = None
|
|
if not self.show_dialog_userdata and self.dialog:
|
|
# Close the progress indicator dialog
|
|
self.dialog.close()
|
|
self.dialog = None
|
|
task = backgroundthread.FunctionAsTask(self.threaded_get_iterators,
|
|
None,
|
|
kinds,
|
|
iterator_queue)
|
|
backgroundthread.BGThreader.addTask(task)
|
|
while True:
|
|
section = iterator_queue.get()
|
|
iterator_queue.task_done()
|
|
if section is None:
|
|
break
|
|
# Setup our variables
|
|
self.plex_type = section['plex_type']
|
|
self.section_type = section['section_type']
|
|
self.context = section['context']
|
|
self.get_children = section['get_children']
|
|
# Now do the heavy lifting
|
|
if self.isCanceled() or not self.playstate_per_section(section):
|
|
return False
|
|
# Delete movies that are not on Plex anymore
|
|
LOG.info('Looking for items to delete')
|
|
with section['context'](self.current_sync) as context:
|
|
for plex_id in context.plexdb.plex_id_by_last_sync(self.plex_type,
|
|
self.current_sync):
|
|
if self.isCanceled():
|
|
return False
|
|
context.remove(plex_id, self.plex_type)
|
|
LOG.debug('Done deleting')
|
|
return True
|
|
|
|
@utils.log_time
|
|
def run(self):
|
|
profile = Profile()
|
|
profile.enable()
|
|
self.current_sync = timing.plex_now()
|
|
# Delete playlist and video node files from Kodi
|
|
utils.delete_playlists()
|
|
utils.delete_nodes()
|
|
# Get latest Plex libraries and build playlist and video node files
|
|
if not sections.sync_from_pms():
|
|
return
|
|
successful = False
|
|
try:
|
|
self.queue = backgroundthread.Queue.Queue()
|
|
if self.show_dialog:
|
|
self.dialog = xbmcgui.DialogProgressBG()
|
|
self.dialog.create(utils.lang(39714))
|
|
|
|
# Actual syncing - do only new items first
|
|
LOG.info('Running full_library_sync with repair=%s',
|
|
self.repair)
|
|
if not self.full_library_sync():
|
|
return
|
|
if self.isCanceled():
|
|
return
|
|
if PLAYLIST_SYNC_ENABLED and not playlists.full_sync():
|
|
return
|
|
successful = True
|
|
finally:
|
|
common.update_kodi_library(video=True, music=True)
|
|
if self.dialog:
|
|
self.dialog.close()
|
|
if self.threader:
|
|
self.threader.shutdown()
|
|
if successful:
|
|
utils.settings('lastfullsync', value=str(int(self.current_sync)))
|
|
if self.callback:
|
|
self.callback(successful)
|
|
LOG.info('Done full_sync')
|
|
profile.disable()
|
|
string_io = StringIO()
|
|
stats = Stats(profile, stream=string_io).sort_stats('cumulative')
|
|
stats.print_stats()
|
|
LOG.info('cProfile result: ')
|
|
LOG.info(string_io.getvalue())
|
|
|
|
|
|
def start(show_dialog, repair=False, callback=None):
|
|
"""
|
|
"""
|
|
# FullSync(repair, callback, show_dialog).start()
|
|
FullSync(repair, callback, show_dialog).run()
|