jellyfin-kodi/resources/lib/playlist.py

# -*- coding: utf-8 -*-

#################################################################################################

import logging

import xbmc
import xbmcgui

import playutils
import playbackutils
import embydb_functions as embydb
import read_embyserver as embyserver
from utils import window, kodiSQL, JSONRPC
from database import DatabaseConn
from contextlib import closing

#################################################################################################

log = logging.getLogger("EMBY."+__name__)

#################################################################################################


class Playlist(object):


    def __init__(self):
        self.emby = embyserver.Read_EmbyServer()


    def play_all(self, item_ids, start_at):

        with DatabaseConn('emby') as conn:
            with closing(conn.cursor()) as cursor:
                emby_db = embydb.Embydb_Functions(cursor)

                player = xbmc.Player()
                playlist = xbmc.PlayList(xbmc.PLAYLIST_VIDEO)
                playlist.clear()

                log.info("---*** PLAY ALL ***---")
                log.info("Items: %s and start at: %s", item_ids, start_at)

                started = False
                window('emby_customplaylist', value="true")

                if start_at:
                    # Seek to the starting position
                    window('emby_customplaylist.seektime', str(start_at))

                for item_id in item_ids:

                    log.info("Adding %s to playlist", item_id)
                    item = emby_db.getItem_byId(item_id)
                    try:
                        db_id = item[0]
                        media_type = item[4]

                    except TypeError:
                        # Item is not found in our database, add item manually
                        log.info("Item was not found in the database, manually adding item")
                        item = self.emby.getItem(item_id)
                        self.add_to_xbmc_playlist(playlist, item)

                    else: # Add to playlist
                        self.add_to_playlist(db_id, media_type)

                    if not started:
                        started = True
                        player.play(playlist)

                self.verify_playlist()

    def modify_playlist(self, item_ids):

        with DatabaseConn('emby') as conn:
            with closing(conn.cursor()) as cursor:
                emby_db = embydb.Embydb_Functions(cursor)

                log.info("---*** ADD TO PLAYLIST ***---")
                log.info("Items: %s", item_ids)

                playlist = xbmc.PlayList(xbmc.PLAYLIST_VIDEO)

                for item_id in item_ids:

                    log.info("Adding %s to playlist", item_id)
                    item = emby_db.getItem_byId(item_id)
                    try:
                        db_id = item[0]
                        media_type = item[4]

                    except TypeError:
                        # Item is not found in our database, add item manually
                        item = self.emby.getItem(item_id)
                        self.add_to_xbmc_playlist(playlist, item)

                    else: # Add to playlist
                        self.add_to_playlist(db_id, media_type)

                self.verify_playlist()
            
        return playlist

    @classmethod
    def add_to_xbmc_playlist(cls, playlist, item):

        playurl = playutils.PlayUtils(item).getPlayUrl()
        if not playurl:
            log.info("Failed to retrieve playurl")
            return

        log.info("Playurl: %s", playurl)

        listitem = xbmcgui.ListItem()
        playbackutils.PlaybackUtils(item).setProperties(playurl, listitem)
        playlist.add(playurl, listitem)

    @classmethod
    def add_to_playlist(cls, db_id=None, media_type=None, url=None):

        params = {

            'playlistid': 1
        }
        if db_id is not None:
            params['item'] = {'%sid' % media_type: int(db_id)}
        else:
            params['item'] = {'file': url}

        log.debug(JSONRPC('Playlist.Add').execute(params))

    @classmethod
    def insert_to_playlist(cls, position, db_id=None, media_type=None, url=None):

        params = {

            'playlistid': 1,
            'position': position
        }
        if db_id is not None:
            params['item'] = {'%sid' % media_type: int(db_id)}
        else:
            params['item'] = {'file': url}

        log.debug(JSONRPC('Playlist.Insert').execute(params))

    @classmethod
    def verify_playlist(cls):
        log.debug(JSONRPC('Playlist.GetItems').execute({'playlistid': 1}))

    @classmethod
    def remove_from_playlist(cls, position):

        params = {

            'playlistid': 1,
            'position': position
        }
        log.debug(JSONRPC('Playlist.Remove').execute(params))