jellyfin-kodi/resources/lib/context_entry.py

221 lines
6.9 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
#################################################################################################
import logging
import sys
from datetime import timedelta
import xbmc
import xbmcaddon
import api
import read_embyserver as embyserver
import playbackutils as pbutils
import embydb_functions as embydb
import musicutils as musicutils
from utils import settings, dialog, language as lang
from dialogs import context, resume
from database import DatabaseConn
#################################################################################################
log = logging.getLogger("EMBY."+__name__)
addon = xbmcaddon.Addon('plugin.video.emby')
XML_PATH = (addon.getAddonInfo('path'), "default", "1080i")
OPTIONS = {
'Refresh': lang(30410),
'Delete': lang(30409),
'Addon': lang(30408),
'AddFav': lang(30405),
'RemoveFav': lang(30406),
'RateSong': lang(30407),
'Transcode': lang(30412)
}
#################################################################################################
class ContextMenu(object):
_selected_option = None
def __init__(self, force_transcode=False):
self.emby = embyserver.Read_EmbyServer()
self.kodi_id = sys.listitem.getVideoInfoTag().getDbId()
self.item_type = self._get_item_type()
2017-12-30 22:53:03 +00:00
self.item_id = self._get_item_id(self.kodi_id, self.item_type)
log.info("Found item_id: %s item_type: %s", self.item_id, self.item_type)
if self.item_id:
self.item = self.emby.getItem(self.item_id)
self.api = api.API(self.item)
if force_transcode:
self._force_transcode()
elif self._select_menu():
self._action_menu()
if self._selected_option in (OPTIONS['Delete'], OPTIONS['AddFav'],
OPTIONS['RemoveFav'], OPTIONS['RateSong']):
log.info("refreshing container")
xbmc.sleep(500)
xbmc.executebuiltin('Container.Refresh')
@classmethod
def _get_item_type(cls):
item_type = sys.listitem.getVideoInfoTag().getMediaType().decode('utf-8')
if not item_type:
if xbmc.getCondVisibility('Container.Content(albums)'):
item_type = "album"
elif xbmc.getCondVisibility('Container.Content(artists)'):
item_type = "artist"
elif xbmc.getCondVisibility('Container.Content(songs)'):
item_type = "song"
elif xbmc.getCondVisibility('Container.Content(pictures)'):
item_type = "picture"
else:
log.info("item_type is unknown")
return item_type
@classmethod
def _get_item_id(cls, kodi_id, item_type):
item_id = xbmc.getInfoLabel('ListItem.Property(embyid)')
if not item_id and kodi_id and item_type:
with DatabaseConn('emby') as cursor:
emby_db = embydb.Embydb_Functions(cursor)
item = emby_db.getItem_byKodiId(kodi_id, item_type)
try:
item_id = item[0]
except TypeError:
pass
return item_id
def _select_menu(self):
# Display select dialog
userdata = self.api.get_userdata()
options = []
if userdata['Favorite']:
# Remove from emby favourites
options.append(OPTIONS['RemoveFav'])
else:
# Add to emby favourites
options.append(OPTIONS['AddFav'])
if self.item_type == "song":
# Set custom song rating
options.append(OPTIONS['RateSong'])
# Refresh item
options.append(OPTIONS['Refresh'])
# Delete item
options.append(OPTIONS['Delete'])
# Addon settings
options.append(OPTIONS['Addon'])
context_menu = context.ContextMenu("script-emby-context.xml", *XML_PATH)
2016-09-27 06:33:09 +00:00
context_menu.set_options(options)
context_menu.doModal()
if context_menu.is_selected():
self._selected_option = context_menu.get_selected()
return self._selected_option
def _action_menu(self):
2017-04-04 02:51:25 +00:00
selected = self._selected_option.decode('utf-8')
if selected == OPTIONS['Refresh']:
self.emby.refreshItem(self.item_id)
elif selected == OPTIONS['AddFav']:
self.emby.updateUserRating(self.item_id, favourite=True)
elif selected == OPTIONS['RemoveFav']:
self.emby.updateUserRating(self.item_id, favourite=False)
elif selected == OPTIONS['RateSong']:
self._rate_song()
elif selected == OPTIONS['Addon']:
xbmc.executebuiltin('Addon.OpenSettings(plugin.video.emby)')
elif selected == OPTIONS['Delete']:
self._delete_item()
def _rate_song(self):
with DatabaseConn('music') as cursor_music:
query = "SELECT rating FROM song WHERE idSong = ?"
cursor_music.execute(query, (self.kodi_id,))
try:
value = cursor_music.fetchone()[0]
current_value = int(round(float(value), 0))
except TypeError:
pass
else:
new_value = dialog("numeric", 0, lang(30411), str(current_value))
if new_value > -1:
new_value = int(new_value)
if new_value > 5:
new_value = 5
if settings('enableUpdateSongRating') == "true":
musicutils.updateRatingToFile(new_value, self.api.get_file_path())
query = "UPDATE song SET rating = ? WHERE idSong = ?"
cursor_music.execute(query, (new_value, self.kodi_id,))
def _delete_item(self):
delete = True
if settings('skipContextMenu') != "true":
if not dialog(type_="yesno", heading="{emby}", line1=lang(33041)):
log.info("User skipped deletion for: %s", self.item_id)
delete = False
if delete:
log.info("Deleting request: %s", self.item_id)
self.emby.deleteItem(self.item_id)
def _force_transcode(self):
log.info("Force transcode called.")
seektime = self.api.adjust_resume(self.api.get_userdata()['Resume'])
if seektime:
log.info("Resume dialog called.")
dialog = resume.ResumeDialog("script-emby-resume.xml", *XML_PATH)
dialog.set_resume_point("Resume from %s" % str(timedelta(seconds=seektime)).split(".")[0])
dialog.doModal()
if dialog.is_selected():
if not dialog.get_selected(): # Start from beginning selected.
self.item['UserData']['PlaybackPositionTicks'] = 0
else: # User backed out
log.info("User exited without a selection.")
return
pbutils.PlaybackUtils(self.item).play(self.item['Id'], self.kodi_id, True)