PlexKodiConnect/resources/lib/player.py

412 lines
15 KiB
Python
Raw Normal View History

2015-12-25 07:07:00 +11:00
# -*- coding: utf-8 -*-
###############################################################################
2016-09-02 03:41:55 +10:00
import logging
2015-12-25 07:07:00 +11:00
import json
import xbmc
import xbmcgui
2016-09-02 03:41:55 +10:00
from utils import window, settings, language as lang, DateToKodi, \
getUnixTimestamp
2015-12-25 07:07:00 +11:00
import clientinfo
import downloadutils
import embydb_functions as embydb
import kodidb_functions as kodidb
2015-12-25 07:07:00 +11:00
###############################################################################
2015-12-25 07:07:00 +11:00
2016-09-02 03:41:55 +10:00
log = logging.getLogger("PLEX."+__name__)
###############################################################################
2015-12-25 07:07:00 +11:00
class Player(xbmc.Player):
# Borg - multiple instances, shared state
_shared_state = {}
played_info = {}
playStats = {}
currentFile = None
def __init__(self):
self.__dict__ = self._shared_state
self.clientInfo = clientinfo.ClientInfo()
2016-02-17 19:13:37 +11:00
self.doUtils = downloadutils.DownloadUtils().downloadUrl
2015-12-25 07:07:00 +11:00
2016-07-22 23:54:03 +10:00
xbmc.Player.__init__(self)
2016-09-02 03:41:55 +10:00
log.info("Started playback monitor.")
2015-12-25 07:07:00 +11:00
def GetPlayStats(self):
return self.playStats
2016-02-17 19:13:37 +11:00
def onPlayBackStarted(self):
"""
2016-09-02 03:41:55 +10:00
Will be called when xbmc starts playing a file.
Window values need to have been set in Kodimonitor.py
"""
2015-12-25 07:07:00 +11:00
self.stopAll()
# Get current file (in utf-8!)
2015-12-25 07:07:00 +11:00
try:
currentFile = self.getPlayingFile()
2015-12-25 07:07:00 +11:00
xbmc.sleep(300)
except:
currentFile = ""
count = 0
while not currentFile:
xbmc.sleep(100)
try:
currentFile = self.getPlayingFile()
except:
pass
if count == 20:
2015-12-25 07:07:00 +11:00
break
else:
count += 1
if not currentFile:
2016-09-02 03:41:55 +10:00
log.warn('Error getting currently playing file; abort reporting')
return
# Save currentFile for cleanup later and for references
self.currentFile = currentFile
2016-09-02 03:41:55 +10:00
window('plex_lastPlayedFiled', value=currentFile)
# We may need to wait for info to be set in kodi monitor
itemId = window("emby_%s.itemid" % currentFile)
count = 0
while not itemId:
xbmc.sleep(200)
2016-02-17 19:13:37 +11:00
itemId = window("emby_%s.itemid" % currentFile)
2016-07-22 23:33:27 +10:00
if count == 5:
2016-09-02 03:41:55 +10:00
log.warn("Could not find itemId, cancelling playback report!")
return
count += 1
2016-09-02 03:41:55 +10:00
log.info("ONPLAYBACK_STARTED: %s itemid: %s" % (currentFile, itemId))
embyitem = "emby_%s" % currentFile
runtime = window("%s.runtime" % embyitem)
refresh_id = window("%s.refreshid" % embyitem)
playMethod = window("%s.playmethod" % embyitem)
itemType = window("%s.type" % embyitem)
try:
playcount = int(window("%s.playcount" % embyitem))
except ValueError:
playcount = 0
window('emby_skipWatched%s' % itemId, value="true")
2016-09-02 03:41:55 +10:00
log.debug("Playing itemtype is: %s" % itemType)
2016-05-31 16:06:42 +10:00
customseek = window('plex_customplaylist.seektime')
if customseek:
2016-09-02 03:41:55 +10:00
# Start at, when using custom playlist (play to Kodi from
# webclient)
log.info("Seeking to: %s" % customseek)
2016-06-01 03:13:29 +10:00
try:
self.seekTime(int(customseek))
2016-06-01 03:13:29 +10:00
except:
2016-09-02 03:41:55 +10:00
log.error('Could not seek!')
2016-05-31 16:06:42 +10:00
window('plex_customplaylist.seektime', clear=True)
try:
seekTime = self.getTime()
except RuntimeError:
2016-09-02 03:41:55 +10:00
log.error('Could not get current seektime from xbmc player')
seekTime = 0
# Get playback volume
volume_query = {
"jsonrpc": "2.0",
"id": 1,
"method": "Application.GetProperties",
"params": {
"properties": ["volume", "muted"]
}
}
result = xbmc.executeJSONRPC(json.dumps(volume_query))
result = json.loads(result)
result = result.get('result')
2016-09-02 03:41:55 +10:00
volume = result.get('volume')
muted = result.get('muted')
# Postdata structure to send to Emby server
url = "{server}/:/timeline?"
postdata = {
'QueueableMediaTypes': "Video",
'CanSeek': True,
'ItemId': itemId,
'MediaSourceId': itemId,
'PlayMethod': playMethod,
'VolumeLevel': volume,
'PositionTicks': int(seekTime * 10000000),
'IsMuted': muted
}
# Get the current audio track and subtitles
if playMethod == "Transcode":
# property set in PlayUtils.py
postdata['AudioStreamIndex'] = window("%sAudioStreamIndex" % currentFile)
postdata['SubtitleStreamIndex'] = window("%sSubtitleStreamIndex" % currentFile)
else:
# Get the current kodi audio and subtitles and convert to Emby equivalent
tracks_query = {
"jsonrpc": "2.0",
"id": 1,
"method": "Player.GetProperties",
"params": {
"playerid": 1,
"properties": ["currentsubtitle","currentaudiostream","subtitleenabled"]
2015-12-25 07:07:00 +11:00
}
}
result = xbmc.executeJSONRPC(json.dumps(tracks_query))
result = json.loads(result)
result = result.get('result')
2015-12-25 07:07:00 +11:00
try: # Audio tracks
indexAudio = result['currentaudiostream']['index']
except (KeyError, TypeError):
indexAudio = 0
try: # Subtitles tracks
indexSubs = result['currentsubtitle']['index']
except (KeyError, TypeError):
indexSubs = 0
try: # If subtitles are enabled
subsEnabled = result['subtitleenabled']
except (KeyError, TypeError):
subsEnabled = ""
# Postdata for the audio
postdata['AudioStreamIndex'] = indexAudio + 1
# Postdata for the subtitles
if subsEnabled and len(xbmc.Player().getAvailableSubtitleStreams()) > 0:
# Number of audiotracks to help get Emby Index
audioTracks = len(xbmc.Player().getAvailableAudioStreams())
mapping = window("%s.indexMapping" % embyitem)
2015-12-25 07:07:00 +11:00
if mapping: # Set in playbackutils.py
2015-12-25 07:07:00 +11:00
2016-09-02 03:41:55 +10:00
log.debug("Mapping for external subtitles index: %s"
% mapping)
externalIndex = json.loads(mapping)
if externalIndex.get(str(indexSubs)):
# If the current subtitle is in the mapping
postdata['SubtitleStreamIndex'] = externalIndex[str(indexSubs)]
2015-12-25 07:07:00 +11:00
else:
# Internal subtitle currently selected
subindex = indexSubs - len(externalIndex) + audioTracks + 1
postdata['SubtitleStreamIndex'] = subindex
2015-12-25 07:07:00 +11:00
else: # Direct paths enabled scenario or no external subtitles set
postdata['SubtitleStreamIndex'] = indexSubs + audioTracks + 1
else:
postdata['SubtitleStreamIndex'] = ""
2015-12-25 07:07:00 +11:00
# Post playback to server
# log("Sending POST play started: %s." % postdata, 2)
# self.doUtils(url, postBody=postdata, type="POST")
# Ensure we do have a runtime
try:
runtime = int(runtime)
except ValueError:
2016-06-01 03:13:29 +10:00
try:
runtime = self.getTotalTime()
2016-09-02 03:41:55 +10:00
log.error("Runtime is missing, Kodi runtime: %s" % runtime)
2016-06-01 03:13:29 +10:00
except:
2016-09-02 03:41:55 +10:00
log.error('Could not get kodi runtime, setting to zero')
2016-06-01 03:13:29 +10:00
runtime = 0
with embydb.GetEmbyDB() as emby_db:
emby_dbitem = emby_db.getItem_byId(itemId)
try:
fileid = emby_dbitem[1]
except TypeError:
2016-09-02 03:41:55 +10:00
log.info("Could not find fileid in plex db.")
fileid = None
# Save data map for updates and position calls
data = {
'runtime': runtime,
'item_id': itemId,
'refresh_id': refresh_id,
'currentfile': currentFile,
'AudioStreamIndex': postdata['AudioStreamIndex'],
'SubtitleStreamIndex': postdata['SubtitleStreamIndex'],
'playmethod': playMethod,
'Type': itemType,
2016-08-11 03:36:08 +10:00
'currentPosition': int(seekTime),
'fileid': fileid,
'itemType': itemType,
'playcount': playcount
}
2016-09-02 03:41:55 +10:00
self.played_info[currentFile] = data
2016-09-02 03:41:55 +10:00
log.info("ADDING_FILE: %s" % data)
# log some playback stats
'''if(itemType != None):
if(self.playStats.get(itemType) != None):
count = self.playStats.get(itemType) + 1
self.playStats[itemType] = count
else:
self.playStats[itemType] = 1
2015-12-25 07:07:00 +11:00
if(playMethod != None):
if(self.playStats.get(playMethod) != None):
count = self.playStats.get(playMethod) + 1
self.playStats[playMethod] = count
else:
self.playStats[playMethod] = 1'''
2015-12-25 07:07:00 +11:00
2016-02-21 10:21:39 +11:00
def onPlayBackPaused(self):
2015-12-25 07:07:00 +11:00
currentFile = self.currentFile
2016-09-02 03:41:55 +10:00
log.info("PLAYBACK_PAUSED: %s" % currentFile)
2015-12-25 07:07:00 +11:00
if self.played_info.get(currentFile):
self.played_info[currentFile]['paused'] = True
2016-02-21 10:21:39 +11:00
def onPlayBackResumed(self):
2015-12-25 07:07:00 +11:00
currentFile = self.currentFile
2016-09-02 03:41:55 +10:00
log.info("PLAYBACK_RESUMED: %s" % currentFile)
2015-12-25 07:07:00 +11:00
if self.played_info.get(currentFile):
self.played_info[currentFile]['paused'] = False
2016-02-21 10:21:39 +11:00
def onPlayBackSeek(self, time, seekOffset):
2015-12-25 07:07:00 +11:00
# Make position when seeking a bit more accurate
currentFile = self.currentFile
2016-09-02 03:41:55 +10:00
self.info("PLAYBACK_SEEK: %s" % currentFile)
2015-12-25 07:07:00 +11:00
if self.played_info.get(currentFile):
try:
position = self.getTime()
except RuntimeError:
# When Kodi is not playing
return
2016-08-11 03:36:08 +10:00
self.played_info[currentFile]['currentPosition'] = position
2016-02-21 10:21:39 +11:00
def onPlayBackStopped(self):
2016-03-17 04:01:07 +11:00
# Will be called when user stops xbmc playing a file
2016-09-02 03:41:55 +10:00
log.info("ONPLAYBACK_STOPPED")
2016-03-17 04:01:07 +11:00
2015-12-25 07:07:00 +11:00
self.stopAll()
2016-03-17 04:01:07 +11:00
window('Plex_currently_playing_itemid', clear=True)
2016-05-31 16:06:42 +10:00
window('plex_customplaylist', clear=True)
window('plex_customplaylist.seektime', clear=True)
window('plex_customplaylist.seektime', clear=True)
2016-09-02 03:41:55 +10:00
log.debug("Cleared playlist properties.")
2016-03-17 04:01:07 +11:00
def onPlayBackEnded(self):
# Will be called when xbmc stops playing a file, because the file ended
2016-09-02 03:41:55 +10:00
log.info("ONPLAYBACK_ENDED")
2016-03-17 04:01:07 +11:00
self.onPlayBackStopped()
2015-12-25 07:07:00 +11:00
def stopAll(self):
if not self.played_info:
2016-09-02 03:41:55 +10:00
return
log.info("Played_information: %s" % self.played_info)
2015-12-25 07:07:00 +11:00
# Process each items
for item in self.played_info:
data = self.played_info.get(item)
if data:
2016-09-02 03:41:55 +10:00
log.debug("Item path: %s" % item)
log.debug("Item data: %s" % data)
2015-12-25 07:07:00 +11:00
runtime = data['runtime']
currentPosition = data['currentPosition']
itemid = data['item_id']
refresh_id = data['refresh_id']
currentFile = data['currentfile']
media_type = data['Type']
2015-12-25 07:07:00 +11:00
playMethod = data['playmethod']
# Prevent manually mark as watched in Kodi monitor
2016-09-02 03:41:55 +10:00
window('emby_skipWatched%s' % itemid, value="true")
2015-12-25 07:07:00 +11:00
if currentPosition and runtime:
try:
percentComplete = float(currentPosition) / float(runtime)
2015-12-25 07:07:00 +11:00
except ZeroDivisionError:
# Runtime is 0.
percentComplete = 0
markPlayed = 0.90
2016-09-02 03:41:55 +10:00
log.info("Percent complete: %s Mark played at: %s"
% (percentComplete, markPlayed))
if percentComplete >= markPlayed:
# Tell Kodi that we've finished watching (Plex knows)
if (data['fileid'] is not None and
data['itemType'] in ('movie', 'episode')):
with kodidb.GetKodiDB('video') as kodi_db:
kodi_db.addPlaystate(
data['fileid'],
None,
None,
data['playcount'] + 1,
2016-09-02 03:41:55 +10:00
DateToKodi(getUnixTimestamp()))
2015-12-25 07:07:00 +11:00
# Send the delete action to the server.
offerDelete = False
if media_type == "Episode" and settings('deleteTV') == "true":
2015-12-25 07:07:00 +11:00
offerDelete = True
elif media_type == "Movie" and settings('deleteMovies') == "true":
2015-12-25 07:07:00 +11:00
offerDelete = True
2016-02-17 19:13:37 +11:00
if settings('offerDelete') != "true":
2015-12-25 07:07:00 +11:00
# Delete could be disabled, even if the subsetting is enabled.
offerDelete = False
2016-02-07 22:38:50 +11:00
# Plex: never delete
offerDelete = False
if percentComplete >= markPlayed and offerDelete:
2016-03-07 23:38:45 +11:00
resp = xbmcgui.Dialog().yesno(
lang(30091),
lang(33015),
2016-03-07 23:38:45 +11:00
autoclose=120000)
if not resp:
2016-09-02 03:41:55 +10:00
log.info("User skipped deletion.")
continue
2015-12-25 07:07:00 +11:00
url = "{server}/emby/Items/%s?format=json" % itemid
2016-09-02 03:41:55 +10:00
log.info("Deleting request: %s" % itemid)
2016-04-26 22:41:58 +10:00
self.doUtils(url, action_type="DELETE")
2016-02-07 22:38:50 +11:00
# Clean the WINDOW properties
for filename in self.played_info:
cleanup = (
'emby_%s.itemid' % filename,
'emby_%s.runtime' % filename,
'emby_%s.refreshid' % filename,
'emby_%s.playmethod' % filename,
'emby_%s.type' % filename,
'emby_%s.runtime' % filename,
'emby_%s.playcount' % filename,
2016-08-12 06:11:00 +10:00
'plex_%s.playlistPosition' % filename
)
for item in cleanup:
2016-09-02 03:41:55 +10:00
window(item, clear=True)
2016-02-07 22:38:50 +11:00
2016-04-14 00:14:55 +10:00
# Stop transcoding
if playMethod == "Transcode":
2016-09-02 03:41:55 +10:00
log.info("Transcoding for %s terminating" % itemid)
2016-04-26 22:41:58 +10:00
self.doUtils(
2016-04-14 00:14:55 +10:00
"{server}/video/:/transcode/universal/stop",
parameters={'session': self.clientInfo.getDeviceId()})
2015-12-25 07:07:00 +11:00
self.played_info.clear()