PlexKodiConnect/resources/lib/library_sync/process_metadata.py

# -*- coding: utf-8 -*-
from __future__ import absolute_import, division, unicode_literals
from logging import getLogger
import xbmcgui

from cProfile import Profile
from pstats import Stats
from StringIO import StringIO

from . import common
from .. import backgroundthread, utils, variables as v

LOG = getLogger('PLEX.sync.process_metadata')


class InitNewSection(object):
    """
    Throw this into the queue used for ProcessMetadata to tell it which
    Plex library section we're looking at

    context: itemtypes.Movie, itemtypes.Episode, etc.
    """
    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 UpdateLastSync(object):
    def __init__(self, plex_id):
        self.plex_id = plex_id


class ProcessMetadata(backgroundthread.KillableThread, common.libsync_mixin):
    """
    Not yet implemented for more than 1 thread - if ever. Only to be called by
    ONE thread!
    Processes the XML metadata in the queue

    Input:
        queue:      Queue.Queue() object that you'll need to fill up with
                    the downloaded XML eTree objects
        item_class: as used to call functions in itemtypes.py e.g. 'Movies' =>
                    itemtypes.Movies()
    """
    def __init__(self, queue, last_sync, show_dialog):
        self._canceled = False
        self.queue = queue
        self.last_sync = last_sync
        self.show_dialog = show_dialog
        self.total = 0
        self.current = 1
        self.processed = 0
        self.title = ''
        self.section_name = None
        self.dialog = None
        super(ProcessMetadata, 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 run(self):
        LOG.debug('Processing thread started')
        try:
            self._run()
        except:
            utils.ERROR(txt='process_metadata crashed',
                        notify=True,
                        cancel_sync=True)
        finally:
            if self.dialog:
                self.dialog.close()
            while not self.queue.empty():
                # We need to empty the queue to let full_sync finish join()
                self.queue.get()
                self.queue.task_done()
            LOG.debug('Processing thread terminated')

    def _run(self):
        """
        Do the work
        """
        if self.show_dialog:
            self.dialog = xbmcgui.DialogProgressBG()
            self.dialog.create(utils.lang(39714))
        # Init with the very first library section. This will block!
        section = self.queue.get()
        self.queue.task_done()
        if section is None:
            return
        while not self.isCanceled():
            if section is None:
                break
            LOG.debug('Start processing section %s (%ss)',
                      section.name, section.plex_type)
            self.current = 1
            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])
            profile = Profile()
            profile.enable()
            with section.context(self.last_sync) as context:
                while not self.isCanceled():
                    # grabs item from queue. This will block!
                    item = self.queue.get()
                    if isinstance(item, InitNewSection) or item is None:
                        section = item
                        self.queue.task_done()
                        break
                    elif isinstance(item, UpdateLastSync):
                        context.plexdb.update_last_sync(item.plex_id,
                                                        section.plex_type,
                                                        self.last_sync)
                    else:
                        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
                    self.update_progressbar()
                    self.current += 1
                    if self.processed == 500:
                        self.processed = 0
                        context.commit()
                    self.queue.task_done()
            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())