403 lines
14 KiB
Python
403 lines
14 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
###############################################################################
|
|
|
|
import logging
|
|
from downloadutils import DownloadUtils
|
|
|
|
from utils import window, settings, tryEncode, language as lang, dialog
|
|
import variables as v
|
|
import PlexAPI
|
|
|
|
###############################################################################
|
|
|
|
log = logging.getLogger("PLEX."+__name__)
|
|
|
|
###############################################################################
|
|
|
|
|
|
class PlayUtils():
|
|
|
|
def __init__(self, item):
|
|
|
|
self.item = item
|
|
self.API = PlexAPI.API(item)
|
|
self.doUtils = DownloadUtils().downloadUrl
|
|
|
|
self.userid = window('currUserId')
|
|
self.server = window('pms_server')
|
|
self.machineIdentifier = window('plex_machineIdentifier')
|
|
|
|
def getPlayUrl(self, partNumber=None):
|
|
"""
|
|
Returns the playurl for the part with number partNumber
|
|
(movie might consist of several files)
|
|
|
|
playurl is utf-8 encoded!
|
|
"""
|
|
self.API.setPartNumber(partNumber)
|
|
self.API.getMediastreamNumber()
|
|
playurl = self.isDirectPlay()
|
|
|
|
if playurl is not None:
|
|
log.info("File is direct playing.")
|
|
playurl = tryEncode(playurl)
|
|
# Set playmethod property
|
|
window('plex_%s.playmethod' % playurl, "DirectPlay")
|
|
|
|
elif self.isDirectStream():
|
|
log.info("File is direct streaming.")
|
|
playurl = tryEncode(
|
|
self.API.getTranscodeVideoPath('DirectStream'))
|
|
# Set playmethod property
|
|
window('plex_%s.playmethod' % playurl, "DirectStream")
|
|
|
|
else:
|
|
log.info("File is transcoding.")
|
|
playurl = tryEncode(self.API.getTranscodeVideoPath(
|
|
'Transcode',
|
|
quality={
|
|
'maxVideoBitrate': self.get_bitrate(),
|
|
'videoResolution': self.get_resolution(),
|
|
'videoQuality': '100',
|
|
'mediaBufferSize': int(settings('kodi_video_cache'))/1024,
|
|
}))
|
|
# Set playmethod property
|
|
window('plex_%s.playmethod' % playurl, value="Transcode")
|
|
|
|
log.info("The playurl is: %s" % playurl)
|
|
return playurl
|
|
|
|
def isDirectPlay(self):
|
|
"""
|
|
Returns the path/playurl if we can direct play, None otherwise
|
|
"""
|
|
# True for e.g. plex.tv watch later
|
|
if self.API.shouldStream() is True:
|
|
log.info("Plex item optimized for direct streaming")
|
|
return
|
|
# set to either 'Direct Stream=1' or 'Transcode=2'
|
|
# and NOT to 'Direct Play=0'
|
|
if settings('playType') != "0":
|
|
# User forcing to play via HTTP
|
|
log.info("User chose to not direct play")
|
|
return
|
|
if self.mustTranscode():
|
|
return
|
|
return self.API.validatePlayurl(self.API.getFilePath(),
|
|
self.API.getType(),
|
|
forceCheck=True)
|
|
|
|
def directPlay(self):
|
|
|
|
try:
|
|
playurl = self.item['MediaSources'][0]['Path']
|
|
except (IndexError, KeyError):
|
|
playurl = self.item['Path']
|
|
|
|
if self.item.get('VideoType'):
|
|
# Specific format modification
|
|
if self.item['VideoType'] == "Dvd":
|
|
playurl = "%s/VIDEO_TS/VIDEO_TS.IFO" % playurl
|
|
elif self.item['VideoType'] == "BluRay":
|
|
playurl = "%s/BDMV/index.bdmv" % playurl
|
|
|
|
# Assign network protocol
|
|
if playurl.startswith('\\\\'):
|
|
playurl = playurl.replace("\\\\", "smb://")
|
|
playurl = playurl.replace("\\", "/")
|
|
|
|
if "apple.com" in playurl:
|
|
USER_AGENT = "QuickTime/7.7.4"
|
|
playurl += "?|User-Agent=%s" % USER_AGENT
|
|
|
|
return playurl
|
|
|
|
def mustTranscode(self):
|
|
"""
|
|
Returns True if we need to transcode because
|
|
- codec is in h265
|
|
- 10bit video codec
|
|
- HEVC codec
|
|
- window variable 'plex_forcetranscode' set to 'true'
|
|
(excepting trailers etc.)
|
|
- video bitrate above specified settings bitrate
|
|
if the corresponding file settings are set to 'true'
|
|
"""
|
|
if self.API.getType() in (v.PLEX_TYPE_CLIP, v.PLEX_TYPE_SONG):
|
|
log.info('Plex clip or music track, not transcoding')
|
|
return False
|
|
videoCodec = self.API.getVideoCodec()
|
|
log.info("videoCodec: %s" % videoCodec)
|
|
if window('plex_forcetranscode') == 'true':
|
|
log.info('User chose to force-transcode')
|
|
return True
|
|
if (settings('transcodeHi10P') == 'true' and
|
|
videoCodec['bitDepth'] == '10'):
|
|
log.info('Option to transcode 10bit video content enabled.')
|
|
return True
|
|
codec = videoCodec['videocodec']
|
|
if codec is None:
|
|
# e.g. trailers. Avoids TypeError with "'h265' in codec"
|
|
log.info('No codec from PMS, not transcoding.')
|
|
return False
|
|
try:
|
|
bitrate = int(videoCodec['bitrate'])
|
|
except (TypeError, ValueError):
|
|
log.info('No video bitrate from PMS, not transcoding.')
|
|
return False
|
|
if bitrate > self.get_max_bitrate():
|
|
log.info('Video bitrate of %s is higher than the maximal video'
|
|
'bitrate of %s that the user chose. Transcoding'
|
|
% (bitrate, self.get_max_bitrate()))
|
|
return True
|
|
try:
|
|
resolution = int(videoCodec['resolution'])
|
|
except (TypeError, ValueError):
|
|
log.info('No video resolution from PMS, not transcoding.')
|
|
return False
|
|
if 'h265' in codec or 'hevc' in codec:
|
|
if resolution >= self.getH265():
|
|
log.info("Option to transcode h265/HEVC enabled. Resolution "
|
|
"of the media: %s, transcoding limit resolution: %s"
|
|
% (str(resolution), str(self.getH265())))
|
|
return True
|
|
return False
|
|
|
|
def isDirectStream(self):
|
|
# Never transcode Music
|
|
if self.API.getType() == 'track':
|
|
return True
|
|
# set to 'Transcode=2'
|
|
if settings('playType') == "2":
|
|
# User forcing to play via HTTP
|
|
log.info("User chose to transcode")
|
|
return False
|
|
if self.mustTranscode():
|
|
return False
|
|
return True
|
|
|
|
def get_max_bitrate(self):
|
|
# get the addon video quality
|
|
videoQuality = settings('maxVideoQualities')
|
|
bitrate = {
|
|
'0': 320,
|
|
'1': 720,
|
|
'2': 1500,
|
|
'3': 2000,
|
|
'4': 3000,
|
|
'5': 4000,
|
|
'6': 8000,
|
|
'7': 10000,
|
|
'8': 12000,
|
|
'9': 20000,
|
|
'10': 40000,
|
|
'11': 99999999 # deactivated
|
|
}
|
|
# max bit rate supported by server (max signed 32bit integer)
|
|
return bitrate.get(videoQuality, 2147483)
|
|
|
|
def getH265(self):
|
|
"""
|
|
Returns the user settings for transcoding h265: boundary resolutions
|
|
of 480, 720 or 1080 as an int
|
|
|
|
OR 9999999 (int) if user chose not to transcode
|
|
"""
|
|
H265 = {
|
|
'0': 99999999,
|
|
'1': 480,
|
|
'2': 720,
|
|
'3': 1080
|
|
}
|
|
return H265[settings('transcodeH265')]
|
|
|
|
def get_bitrate(self):
|
|
"""
|
|
Get the desired transcoding bitrate from the settings
|
|
"""
|
|
videoQuality = settings('transcoderVideoQualities')
|
|
bitrate = {
|
|
'0': 320,
|
|
'1': 720,
|
|
'2': 1500,
|
|
'3': 2000,
|
|
'4': 3000,
|
|
'5': 4000,
|
|
'6': 8000,
|
|
'7': 10000,
|
|
'8': 12000,
|
|
'9': 20000,
|
|
'10': 40000,
|
|
}
|
|
# max bit rate supported by server (max signed 32bit integer)
|
|
return bitrate.get(videoQuality, 2147483)
|
|
|
|
def get_resolution(self):
|
|
"""
|
|
Get the desired transcoding resolutions from the settings
|
|
"""
|
|
chosen = settings('transcoderVideoQualities')
|
|
res = {
|
|
'0': '420x420',
|
|
'1': '576x320',
|
|
'2': '720x480',
|
|
'3': '1024x768',
|
|
'4': '1280x720',
|
|
'5': '1280x720',
|
|
'6': '1920x1080',
|
|
'7': '1920x1080',
|
|
'8': '1920x1080',
|
|
'9': '1920x1080',
|
|
'10': '1920x1080',
|
|
}
|
|
return res[chosen]
|
|
|
|
def audioSubsPref(self, listitem, url, part=None):
|
|
"""
|
|
For transcoding only
|
|
|
|
Called at the very beginning of play; used to change audio and subtitle
|
|
stream by a PUT request to the PMS
|
|
"""
|
|
# Set media and part where we're at
|
|
if self.API.mediastream is None:
|
|
self.API.getMediastreamNumber()
|
|
if part is None:
|
|
part = 0
|
|
try:
|
|
mediastreams = self.item[self.API.mediastream][part]
|
|
except (TypeError, IndexError):
|
|
log.error('Could not get media %s, part %s'
|
|
% (self.API.mediastream, part))
|
|
return
|
|
part_id = mediastreams.attrib['id']
|
|
audio_streams_list = []
|
|
audio_streams = []
|
|
subtitle_streams_list = []
|
|
# No subtitles as an option
|
|
subtitle_streams = [lang(39706)]
|
|
downloadable_streams = []
|
|
download_subs = []
|
|
# selectAudioIndex = ""
|
|
select_subs_index = ""
|
|
audio_numb = 0
|
|
# Remember 'no subtitles'
|
|
sub_num = 1
|
|
default_sub = None
|
|
|
|
for stream in mediastreams:
|
|
# Since Plex returns all possible tracks together, have to sort
|
|
# them.
|
|
index = stream.attrib.get('id')
|
|
typus = stream.attrib.get('streamType')
|
|
# Audio
|
|
if typus == "2":
|
|
codec = stream.attrib.get('codec')
|
|
channelLayout = stream.attrib.get('audioChannelLayout', "")
|
|
try:
|
|
track = "%s %s - %s %s" % (audio_numb+1,
|
|
stream.attrib['language'],
|
|
codec,
|
|
channelLayout)
|
|
except:
|
|
track = "%s %s - %s %s" % (audio_numb+1,
|
|
lang(39707), # unknown
|
|
codec,
|
|
channelLayout)
|
|
audio_streams_list.append(index)
|
|
audio_streams.append(tryEncode(track))
|
|
audio_numb += 1
|
|
|
|
# Subtitles
|
|
elif typus == "3":
|
|
try:
|
|
track = "%s %s" % (sub_num+1, stream.attrib['language'])
|
|
except KeyError:
|
|
track = "%s %s (%s)" % (sub_num+1,
|
|
lang(39707), # unknown
|
|
stream.attrib.get('codec'))
|
|
default = stream.attrib.get('default')
|
|
forced = stream.attrib.get('forced')
|
|
downloadable = stream.attrib.get('key')
|
|
|
|
if default:
|
|
track = "%s - %s" % (track, lang(39708)) # Default
|
|
if forced:
|
|
track = "%s - %s" % (track, lang(39709)) # Forced
|
|
if downloadable:
|
|
# We do know the language - temporarily download
|
|
if 'language' in stream.attrib:
|
|
path = self.API.download_external_subtitles(
|
|
'{server}%s' % stream.attrib['key'],
|
|
"subtitle.%s.%s" % (stream.attrib['language'],
|
|
stream.attrib['codec']))
|
|
# We don't know the language - no need to download
|
|
else:
|
|
path = self.API.addPlexCredentialsToUrl(
|
|
"%s%s" % (self.server, stream.attrib['key']))
|
|
downloadable_streams.append(index)
|
|
download_subs.append(tryEncode(path))
|
|
else:
|
|
track = "%s (%s)" % (track, lang(39710)) # burn-in
|
|
if stream.attrib.get('selected') == '1' and downloadable:
|
|
# Only show subs without asking user if they can be
|
|
# turned off
|
|
default_sub = index
|
|
|
|
subtitle_streams_list.append(index)
|
|
subtitle_streams.append(tryEncode(track))
|
|
sub_num += 1
|
|
|
|
if audio_numb > 1:
|
|
resp = dialog('select', lang(33013), audio_streams)
|
|
if resp > -1:
|
|
# User selected some audio track
|
|
args = {
|
|
'audioStreamID': audio_streams_list[resp],
|
|
'allParts': 1
|
|
}
|
|
self.doUtils('{server}/library/parts/%s' % part_id,
|
|
action_type='PUT',
|
|
parameters=args)
|
|
|
|
if sub_num == 1:
|
|
# No subtitles
|
|
return
|
|
|
|
select_subs_index = None
|
|
if (settings('pickPlexSubtitles') == 'true' and
|
|
default_sub is not None):
|
|
log.info('Using default Plex subtitle: %s' % default_sub)
|
|
select_subs_index = default_sub
|
|
else:
|
|
resp = dialog('select', lang(33014), subtitle_streams)
|
|
if resp > 0:
|
|
select_subs_index = subtitle_streams_list[resp-1]
|
|
else:
|
|
# User selected no subtitles or backed out of dialog
|
|
select_subs_index = ''
|
|
|
|
log.debug('Adding external subtitles: %s' % download_subs)
|
|
# Enable Kodi to switch autonomously to downloadable subtitles
|
|
if download_subs:
|
|
listitem.setSubtitles(download_subs)
|
|
|
|
if select_subs_index in downloadable_streams:
|
|
for i, stream in enumerate(downloadable_streams):
|
|
if stream == select_subs_index:
|
|
# Set the correct subtitle
|
|
window('plex_%s.subtitle' % tryEncode(url), value=str(i))
|
|
break
|
|
# Don't additionally burn in subtitles
|
|
select_subs_index = ''
|
|
else:
|
|
window('plex_%s.subtitle' % tryEncode(url), value='None')
|
|
|
|
args = {
|
|
'subtitleStreamID': select_subs_index,
|
|
'allParts': 1
|
|
}
|
|
self.doUtils('{server}/library/parts/%s' % part_id,
|
|
action_type='PUT',
|
|
parameters=args)
|