PlexKodiConnect/resources/lib/library_sync/get_metadata.py

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

from . import common
from ..plex_api import API
from .. import backgroundthread, plex_functions as PF, utils, variables as v

LOG = getLogger('PLEX.sync.get_metadata')
LOCK = backgroundthread.threading.Lock()


class GetMetadataThread(common.LibrarySyncMixin,
                        backgroundthread.KillableThread):
    """
    Threaded download of Plex XML metadata for a certain library item.
    Fills the queue with the downloaded etree XML objects
    """
    def __init__(self, get_metadata_queue, processing_queue):
        self.get_metadata_queue = get_metadata_queue
        self.processing_queue = processing_queue
        super(GetMetadataThread, self).__init__()

    def _collections(self, item):
        api = API(item['xml'][0])
        collection_match = item['section'].collection_match
        collection_xmls = item['section'].collection_xmls
        if collection_match is None:
            collection_match = PF.collections(api.library_section_id())
            if collection_match is None:
                LOG.error('Could not download collections')
                return
            # Extract what we need to know
            collection_match = \
                [(utils.cast(int, x.get('index')),
                  utils.cast(int, x.get('ratingKey'))) for x in collection_match]
        item['children'] = {}
        for plex_set_id, set_name in api.collections():
            if self.should_cancel():
                return
            if plex_set_id not in collection_xmls:
                # Get Plex metadata for collections - a pain
                for index, collection_plex_id in collection_match:
                    if index == plex_set_id:
                        collection_xml = PF.GetPlexMetadata(collection_plex_id)
                        try:
                            collection_xml[0].attrib
                        except (TypeError, IndexError, AttributeError):
                            LOG.error('Could not get collection %s %s',
                                      collection_plex_id, set_name)
                            continue
                        collection_xmls[plex_set_id] = collection_xml
                        break
                else:
                    LOG.error('Did not find Plex collection %s %s',
                              plex_set_id, set_name)
                    continue
            item['children'][plex_set_id] = collection_xmls[plex_set_id]

    def _process_abort(self, count, section):
        # Make sure other threads will also receive sentinel
        self.get_metadata_queue.put(None)
        if count is not None:
            self._process_skipped_item(count, section)

    def _process_skipped_item(self, count, section):
        section.sync_successful = False
        # Add a "dummy" item so we're not skipping a beat
        self.processing_queue.put((count, {'section': section, 'xml': None}))

    def _run(self):
        while True:
            item = self.get_metadata_queue.get()
            try:
                if item is None or self.should_cancel():
                    self._process_abort(item[0] if item else None,
                                        item[2] if item else None)
                    break
                count, plex_id, section = item
                item = {
                    'xml': PF.GetPlexMetadata(plex_id),  # This will block
                    'children': None,
                    'section': section
                }
                if item['xml'] is None:
                    # Did not receive a valid XML - skip that item for now
                    LOG.error("Could not get metadata for %s. Skipping item "
                              "for now", plex_id)
                    self._process_skipped_item(count, section)
                    continue
                elif item['xml'] == 401:
                    LOG.error('HTTP 401 returned by PMS. Too much strain? '
                              'Cancelling sync for now')
                    utils.window('plex_scancrashed', value='401')
                    self._process_abort(count, section)
                    break
                if section.plex_type == v.PLEX_TYPE_MOVIE:
                    # Check for collections/sets
                    collections = False
                    for child in item['xml'][0]:
                        if child.tag == 'Collection':
                            collections = True
                            break
                    if collections:
                        with LOCK:
                            self._collections(item)
                if section.get_children:
                    if self.should_cancel():
                        self._process_abort(count, section)
                        break
                    children_xml = PF.GetAllPlexChildren(plex_id)  # Will block
                    try:
                        children_xml[0].attrib
                    except (TypeError, IndexError, AttributeError):
                        LOG.error('Could not get children for Plex id %s',
                                  plex_id)
                        self._process_skipped_item(count, section)
                        continue
                    else:
                        item['children'] = children_xml
                self.processing_queue.put((count, item))
            finally:
                self.get_metadata_queue.task_done()