CouchPotatoServer/couchpotato/core/media/show/episode.py

from couchpotato import get_db
from couchpotato.core.event import addEvent, fireEvent, fireEventAsync
from couchpotato.core.logger import CPLog
from couchpotato.core.helpers.variable import tryInt
from couchpotato.core.media import MediaBase


log = CPLog(__name__)

autoload = 'Episode'


class Episode(MediaBase):

    def __init__(self):
        addEvent('show.episode.add', self.add)
        addEvent('show.episode.update_info', self.updateInfo)

    def add(self, parent_id, info = None, update_after = True):
        if not info: info = {}

        identifiers = info.get('identifiers')
        try: del info['identifiers']
        except: pass

        # Add Season
        episode_info = {
            '_t': 'media',
            'type': 'show.episode',
            'identifiers': identifiers,
            'parent_id': parent_id,
            'info': info, # Returned dict by providers
        }

        # Check if season already exists
        existing_episode = fireEvent('media.with_identifiers', identifiers, with_doc = True, single = True)

        db = get_db()

        if existing_episode:
            s = existing_episode['doc']
            s.update(episode_info)
            episode = db.update(s)
        else:
            episode = db.insert(episode_info)

        # Update library info
        if update_after is not False:
            handle = fireEventAsync if update_after is 'async' else fireEvent
            handle('show.season.update_info', episode.get('_id'), info = info, single = True)

        return episode

    def updateInfo(self, media_id = None, info = None, force = False):
        if not info: info = {}

        if self.shuttingDown():
            return

        db = get_db()

        episode = db.get('id', media_id)

        # Get new info
        if not info:
            season = db.get('id', episode['parent_id'])
            show = db.get('id', season['parent_id'])

            info = fireEvent(
                'episode.info', show.get('identifiers'), {
                    'season_identifier': season.get('info', {}).get('number'),
                    'episode_identifier': episode.get('identifiers')
                },
                merge = True
            )

        # Update/create media
        if force:

            episode['identifiers'].update(info['identifiers'])
            if 'identifiers' in info:
                del info['identifiers']

            episode.update({'info': info})
            e = db.update(episode)
            episode.update(e)

        # Get images
        image_urls = info.get('images', [])
        existing_files = episode.get('files', {})
        self.getPoster(image_urls, existing_files)

        return episode