PlexKodiConnect/resources/lib/PlexAPI.py

1562 lines
55 KiB
Python
Raw Normal View History

2016-09-03 00:53:16 +10:00
# -*- coding: utf-8 -*-
"""
Taken from iBaa, https://github.com/iBaa/PlexConnect
Point of time: December 22, 2015
Collection of "connector functions" to Plex Media Server/MyPlex
PlexGDM:
loosely based on hippojay's plexGDM:
https://github.com/hippojay/script.plexbmc.helper... /resources/lib/plexgdm.py
Plex Media Server communication:
source (somewhat): https://github.com/hippojay/plugin.video.plexbmc
later converted from httplib to urllib2
Transcoder support:
PlexAPI_getTranscodePath() based on getTranscodeURL from pyplex/plexAPI
https://github.com/megawubs/pyplex/blob/master/plexAPI/info.py
MyPlex - Basic Authentication:
http://www.voidspace.org.uk/python/articles/urllib2.shtml
http://www.voidspace.org.uk/python/articles/authentication.shtml
http://stackoverflow.com/questions/2407126/python-urllib2-basic-auth-problem
http://stackoverflow.com/questions/111945/is-there-any-way-to-do-http-put-in-python
(and others...)
"""
2017-12-10 00:35:08 +11:00
from logging import getLogger
2017-01-25 04:00:35 +11:00
from re import compile as re_compile, sub
2018-02-11 03:59:20 +11:00
from urllib import urlencode, unquote
from os.path import basename, join
from os import makedirs
2015-12-29 04:47:16 +11:00
2018-02-11 03:59:20 +11:00
from xbmcgui import ListItem
from xbmcvfs import exists
2016-01-30 06:07:21 +11:00
2017-01-25 02:53:50 +11:00
import clientinfo as client
2018-02-11 03:59:20 +11:00
from downloadutils import DownloadUtils as DU
2018-02-11 22:59:04 +11:00
from utils import window, settings, language as lang, try_decode, try_encode, \
unix_date_to_kodi, exists_dir, slugify, dialog
2018-02-11 03:59:20 +11:00
import PlexFunctions as PF
import plexdb_functions as plexdb
import variables as v
import state
2016-09-03 00:53:16 +10:00
###############################################################################
2018-02-01 17:56:54 +11:00
LOG = getLogger("PLEX." + __name__)
2016-09-03 00:53:16 +10:00
2017-01-25 04:00:35 +11:00
REGEX_IMDB = re_compile(r'''/(tt\d+)''')
2017-02-19 00:45:31 +11:00
REGEX_TVDB = re_compile(r'''thetvdb:\/\/(.+?)\?''')
2018-02-12 00:42:49 +11:00
# Key of library: Plex-identifier. Value represents the Kodi/emby side
PEOPLE_OF_INTEREST = {
'Director': 'Director',
'Writer': 'Writer',
'Role': 'Actor',
'Producer': 'Producer'
}
# we need to use a little mapping between fanart.tv arttypes and kodi
# artttypes
FANART_TV_TYPES = [
("logo", "Logo"),
("musiclogo", "clearlogo"),
("disc", "Disc"),
("clearart", "Art"),
("banner", "Banner"),
("clearlogo", "Logo"),
("background", "fanart"),
("showbackground", "fanart"),
("characterart", "characterart")
]
2016-09-03 00:53:16 +10:00
###############################################################################
2018-02-12 00:42:49 +11:00
class API(object):
"""
API(item)
2016-01-02 06:36:02 +11:00
Processes a Plex media server's XML response
2015-12-29 04:47:16 +11:00
item: xml.etree.ElementTree element
"""
2015-12-29 04:47:16 +11:00
def __init__(self, item):
self.item = item
2016-01-04 05:17:59 +11:00
# which media part in the XML response shall we look at?
self.part = 0
self.mediastream = None
2016-09-03 00:53:16 +10:00
self.server = window('pms_server')
2015-12-29 04:47:16 +11:00
2018-02-12 00:42:49 +11:00
def set_part_number(self, number=None):
2016-01-04 05:17:59 +11:00
"""
Sets the part number to work with (used to deal with Movie with several
parts).
"""
2016-02-08 04:05:59 +11:00
self.part = number or 0
2016-01-04 05:17:59 +11:00
2018-02-12 00:42:49 +11:00
def plex_type(self):
"""
2016-01-30 06:07:21 +11:00
Returns the type of media, e.g. 'movie' or 'clip' for trailers
"""
2018-02-12 18:10:39 +11:00
return self.item.get('type')
2016-01-02 19:28:31 +11:00
2018-02-12 00:42:49 +11:00
def checksum(self):
2015-12-29 04:47:16 +11:00
"""
2018-02-12 00:42:49 +11:00
Returns a string, not int.
2016-03-28 01:57:35 +11:00
WATCH OUT - time in Plex, not Kodi ;-)
2015-12-29 04:47:16 +11:00
"""
# Include a letter to prohibit saving as an int!
2018-02-12 18:10:39 +11:00
return "K%s%s" % (self.plex_id(), self.item.get('updatedAt', ''))
2015-12-29 04:47:16 +11:00
2018-02-12 00:42:49 +11:00
def plex_id(self):
2015-12-30 23:25:37 +11:00
"""
2018-02-12 00:42:49 +11:00
Returns the Plex ratingKey such as '246922' as a string or None
2015-12-30 23:25:37 +11:00
"""
2018-02-12 18:10:39 +11:00
return self.item.get('ratingKey')
2015-12-30 00:13:32 +11:00
2018-02-12 00:42:49 +11:00
def path_and_plex_id(self):
2016-01-30 06:07:21 +11:00
"""
2018-02-12 00:42:49 +11:00
Returns the Plex key such as '/library/metadata/246922' or None
2016-01-30 06:07:21 +11:00
"""
2018-02-12 18:10:39 +11:00
return self.item.get('key')
2018-01-08 03:50:30 +11:00
def plex_media_streams(self):
"""
Returns the media streams directly from the PMS xml.
Mind self.mediastream to be set before and self.part!
"""
return self.item[self.mediastream][self.part]
2018-02-12 00:42:49 +11:00
def file_path(self, force_first_media=False):
"""
Returns the direct path to this item, e.g. '\\NAS\movies\movie.mkv'
or None
2018-02-12 00:42:49 +11:00
force_first_media=True:
will always use 1st media stream, e.g. when several different
files are present for the same PMS item
"""
2018-02-12 00:42:49 +11:00
if self.mediastream is None and force_first_media is False:
self.mediastream_number()
try:
2018-02-12 00:42:49 +11:00
if force_first_media is False:
ans = self.item[self.mediastream][self.part].attrib['file']
else:
ans = self.item[0][self.part].attrib['file']
2018-02-12 00:42:49 +11:00
except (TypeError, AttributeError, IndexError, KeyError):
ans = None
if ans is not None:
2016-04-27 03:19:52 +10:00
try:
2018-02-11 22:59:04 +11:00
ans = try_decode(unquote(ans))
except UnicodeDecodeError:
# Sometimes, Plex seems to have encoded in latin1
ans = unquote(ans).decode('latin1')
return ans
2017-03-19 22:14:16 +11:00
def get_picture_path(self):
"""
2017-03-26 23:09:43 +11:00
Returns the item's picture path (transcode, if necessary) as string.
Will always use addon paths, never direct paths
2017-03-19 22:14:16 +11:00
"""
extension = self.item[0][0].attrib['key'][self.item[0][0].attrib['key'].rfind('.'):].lower()
if (window('plex_force_transcode_pix') == 'true' or
extension not in v.KODI_SUPPORTED_IMAGES):
# Let Plex transcode
# max width/height supported by plex image transcoder is 1920x1080
2018-02-11 03:59:20 +11:00
path = self.server + PF.transcode_image_path(
2018-02-12 18:10:39 +11:00
self.item[0][0].get('key'),
2017-03-19 22:14:16 +11:00
window('pms_token'),
2018-02-12 18:10:39 +11:00
"%s%s" % (self.server, self.item[0][0].get('key')),
2017-03-19 22:14:16 +11:00
1920,
1080)
else:
2018-02-12 00:42:49 +11:00
path = self.attach_plex_token_to_url(
2017-03-26 23:09:43 +11:00
'%s%s' % (window('pms_server'),
self.item[0][0].attrib['key']))
# Attach Plex id to url to let it be picked up by our playqueue agent
# later
2018-02-12 00:42:49 +11:00
return try_encode('%s&plex_id=%s' % (path, self.plex_id()))
2017-03-19 22:14:16 +11:00
2018-02-12 00:42:49 +11:00
def tv_show_path(self):
"""
Returns the direct path to the TV show, e.g. '\\NAS\tv\series'
or None
"""
res = None
for child in self.item:
if child.tag == 'Location':
2018-02-12 18:10:39 +11:00
res = child.get('path')
return res
2016-01-30 06:07:21 +11:00
2018-02-12 00:42:49 +11:00
def season_number(self):
2016-01-10 02:14:02 +11:00
"""
Returns the 'index' of an PMS XML reply. Depicts e.g. season number.
"""
2018-02-12 18:10:39 +11:00
return self.item.get('index')
2016-01-10 02:14:02 +11:00
2018-02-12 00:42:49 +11:00
def date_created(self):
"""
2016-04-15 17:32:04 +10:00
Returns the date when this library item was created.
If not found, returns 2000-01-01 10:00:00
"""
2018-02-12 18:10:39 +11:00
res = self.item.get('addedAt')
if res is not None:
2018-02-11 22:59:04 +11:00
res = unix_date_to_kodi(res)
2016-04-15 17:32:04 +10:00
else:
res = '2000-01-01 10:00:00'
return res
2015-12-29 04:47:16 +11:00
2018-02-12 00:42:49 +11:00
def viewcount(self):
2018-01-22 04:31:49 +11:00
"""
Returns the play count for the item as an int or the int 0 if not found
"""
try:
return int(self.item.attrib['viewCount'])
except (KeyError, ValueError):
return 0
2018-02-12 00:42:49 +11:00
def userdata(self):
"""
Returns a dict with None if a value is missing
{
'Favorite': favorite, # False, because n/a in Plex
'PlayCount': playcount,
'Played': played, # True/False
'LastPlayedDate': lastPlayedDate,
'Resume': resume, # Resume time in seconds
'Runtime': runtime,
'Rating': rating
}
"""
item = self.item.attrib
2016-04-02 19:31:21 +11:00
# Default - attributes not found with Plex
2015-12-29 04:47:16 +11:00
favorite = False
try:
playcount = int(item['viewCount'])
except (KeyError, ValueError):
playcount = None
2017-02-03 01:23:54 +11:00
played = True if playcount else False
2015-12-29 04:47:16 +11:00
try:
2018-02-12 00:42:49 +11:00
last_played = unix_date_to_kodi(int(item['lastViewedAt']))
except (KeyError, ValueError):
2018-02-12 00:42:49 +11:00
last_played = None
2015-12-29 04:47:16 +11:00
if state.INDICATE_MEDIA_VERSIONS is True:
2017-02-03 01:23:54 +11:00
userrating = 0
2018-02-12 00:42:49 +11:00
for _ in self.item.findall('./Media'):
userrating += 1
# Don't show a value of '1'
userrating = 0 if userrating == 1 else userrating
else:
try:
userrating = int(float(item['userRating']))
except (KeyError, ValueError):
userrating = 0
2016-04-02 19:31:21 +11:00
try:
rating = float(item['audienceRating'])
except (KeyError, ValueError):
2016-04-02 19:31:21 +11:00
try:
rating = float(item['rating'])
except (KeyError, ValueError):
2016-04-02 19:31:21 +11:00
rating = 0.0
2018-02-12 00:42:49 +11:00
resume, runtime = self.resume_runtime()
2015-12-29 04:47:16 +11:00
return {
'Favorite': favorite,
'PlayCount': playcount,
'Played': played,
2018-02-12 00:42:49 +11:00
'LastPlayedDate': last_played,
2015-12-29 04:47:16 +11:00
'Resume': resume,
'Runtime': runtime,
'Rating': rating,
'UserRating': userrating
2015-12-29 04:47:16 +11:00
}
2018-02-12 00:42:49 +11:00
def collection_list(self):
2016-03-02 02:18:12 +11:00
"""
2016-03-02 02:28:48 +11:00
Returns a list of PMS collection tags or an empty list
2016-03-02 02:18:12 +11:00
"""
collections = []
for child in self.item:
if child.tag == 'Collection':
if child.attrib['tag']:
collections.append(child.attrib['tag'])
2016-03-02 02:18:12 +11:00
return collections
2018-02-12 00:42:49 +11:00
def people(self):
2015-12-29 04:47:16 +11:00
"""
Returns a dict of lists of people found.
2015-12-29 04:47:16 +11:00
{
'Director': list,
'Writer': list,
'Cast': list,
'Producer': list
2015-12-29 04:47:16 +11:00
}
"""
director = []
writer = []
cast = []
producer = []
for child in self.item:
try:
if child.tag == 'Director':
director.append(child.attrib['tag'])
elif child.tag == 'Writer':
writer.append(child.attrib['tag'])
elif child.tag == 'Role':
cast.append(child.attrib['tag'])
elif child.tag == 'Producer':
producer.append(child.attrib['tag'])
except KeyError:
2018-02-01 17:56:54 +11:00
LOG.warn('Malformed PMS answer for getPeople: %s: %s',
child.tag, child.attrib)
2015-12-29 04:47:16 +11:00
return {
'Director': director,
'Writer': writer,
'Cast': cast,
'Producer': producer
}
2018-02-12 00:42:49 +11:00
def people_list(self):
2015-12-29 04:47:16 +11:00
"""
Returns a list of people from item, with a list item of the form
{
'Name': xxx,
'Type': xxx,
'Id': xxx
'imageurl': url to picture, None otherwise
('Role': xxx for cast/actors only, None if not found)
2015-12-29 04:47:16 +11:00
}
"""
people = []
for child in self.item:
2018-02-12 00:42:49 +11:00
if child.tag in PEOPLE_OF_INTEREST.keys():
name = child.attrib['tag']
name_id = child.attrib['id']
2018-02-12 00:42:49 +11:00
typus = PEOPLE_OF_INTEREST[child.tag]
2018-02-12 18:10:39 +11:00
url = child.get('thumb')
role = child.get('role')
people.append({
'Name': name,
2018-02-12 00:42:49 +11:00
'Type': typus,
'Id': name_id,
'imageurl': url
})
if url:
people[-1].update({'imageurl': url})
2018-02-12 00:42:49 +11:00
if role:
people[-1].update({'Role': role})
2015-12-29 04:47:16 +11:00
return people
2018-02-12 00:42:49 +11:00
def genre_list(self):
2015-12-29 04:47:16 +11:00
"""
Returns a list of genres found. (Not a string)
2015-12-29 04:47:16 +11:00
"""
genre = []
for child in self.item:
if child.tag == 'Genre':
genre.append(child.attrib['tag'])
2015-12-29 04:47:16 +11:00
return genre
2018-02-12 00:42:49 +11:00
def provider(self, providername=None):
2015-12-29 04:47:16 +11:00
"""
providername: e.g. 'imdb', 'tvdb'
2015-12-29 04:47:16 +11:00
2016-02-03 23:01:13 +11:00
Return IMDB, e.g. "tt0903624". Returns None if not found
2015-12-29 04:47:16 +11:00
"""
2016-01-11 19:57:45 +11:00
try:
2018-02-12 18:10:39 +11:00
item = self.item.attrib['guid']
2016-01-11 19:57:45 +11:00
except KeyError:
return None
2016-01-10 02:14:02 +11:00
2016-02-03 23:01:13 +11:00
if providername == 'imdb':
2016-09-26 03:21:12 +10:00
regex = REGEX_IMDB
elif providername == 'tvdb':
# originally e.g. com.plexapp.agents.thetvdb://276564?lang=en
2016-09-26 03:21:12 +10:00
regex = REGEX_TVDB
2016-02-03 23:01:13 +11:00
else:
return None
2016-01-11 19:57:45 +11:00
provider = regex.findall(item)
2015-12-29 04:47:16 +11:00
try:
2016-01-10 02:14:02 +11:00
provider = provider[0]
2016-01-11 19:57:45 +11:00
except IndexError:
2015-12-29 04:47:16 +11:00
provider = None
return provider
2018-02-12 00:42:49 +11:00
def titles(self):
"""
Returns an item's name/title or "Missing Title Name".
2018-02-12 00:42:49 +11:00
Output is the tuple
title, sorttitle
sorttitle = title, if no sorttitle is found
"""
2018-02-12 18:10:39 +11:00
title = self.item.get('title', 'Missing Title Name')
sorttitle = self.item.get('titleSort', title)
2015-12-29 04:47:16 +11:00
return title, sorttitle
2018-02-12 00:42:49 +11:00
def plot(self):
"""
Returns the plot or None.
"""
2018-02-12 18:10:39 +11:00
return self.item.get('summary')
2018-02-12 00:42:49 +11:00
def tagline(self):
"""
Returns a shorter tagline or None
"""
2018-02-12 18:10:39 +11:00
return self.item.get('tagline')
2018-02-12 00:42:49 +11:00
def audience_rating(self):
"""
2016-04-02 19:31:21 +11:00
Returns the audience rating, 'rating' itself or 0.0
"""
2018-02-12 18:10:39 +11:00
res = self.item.get('audienceRating')
2016-03-18 02:15:48 +11:00
if res is None:
2018-02-12 18:10:39 +11:00
res = self.item.get('rating')
2016-04-17 21:36:41 +10:00
try:
res = float(res)
except (ValueError, TypeError):
res = 0.0
2016-03-18 02:15:48 +11:00
return res
2018-02-12 00:42:49 +11:00
def year(self):
"""
Returns the production(?) year ("year") or None
"""
2018-02-12 18:10:39 +11:00
return self.item.get('year')
2018-02-12 00:42:49 +11:00
def resume_point(self):
2018-01-22 21:20:37 +11:00
"""
Returns the resume point of time in seconds as int. 0 if not found
"""
try:
resume = float(self.item.attrib['viewOffset'])
except (KeyError, ValueError):
resume = 0.0
return int(resume * v.PLEX_TO_KODI_TIMEFACTOR)
2018-02-12 00:42:49 +11:00
def resume_runtime(self):
2015-12-29 04:47:16 +11:00
"""
Resume point of time and runtime/totaltime in rounded to seconds.
2016-01-02 06:36:02 +11:00
Time from Plex server is measured in milliseconds.
Kodi: seconds
2018-02-12 00:42:49 +11:00
Output is the tuple:
resume, runtime as ints. 0 if not found
2015-12-29 04:47:16 +11:00
"""
try:
2018-02-12 18:10:39 +11:00
runtime = float(self.item.attrib['duration'])
2016-04-17 21:36:41 +10:00
except (KeyError, ValueError):
runtime = 0.0
2015-12-29 04:47:16 +11:00
try:
2018-02-12 18:10:39 +11:00
resume = float(self.item.attrib['viewOffset'])
2016-04-17 21:36:41 +10:00
except (KeyError, ValueError):
2016-01-02 06:36:02 +11:00
resume = 0.0
runtime = int(runtime * v.PLEX_TO_KODI_TIMEFACTOR)
resume = int(resume * v.PLEX_TO_KODI_TIMEFACTOR)
2015-12-29 04:47:16 +11:00
return resume, runtime
2018-02-12 00:42:49 +11:00
def content_rating(self):
"""
Get the content rating or None
"""
2018-02-12 18:10:39 +11:00
mpaa = self.item.get('contentRating', None)
2015-12-29 04:47:16 +11:00
# Convert more complex cases
if mpaa in ("NR", "UR"):
# Kodi seems to not like NR, but will accept Rated Not Rated
mpaa = "Rated Not Rated"
return mpaa
2018-02-12 00:42:49 +11:00
def country_list(self):
2015-12-29 04:47:16 +11:00
"""
Returns a list of all countries found in item.
"""
country = []
for child in self.item:
if child.tag == 'Country':
country.append(child.attrib['tag'])
2015-12-29 04:47:16 +11:00
return country
2018-02-12 00:42:49 +11:00
def premiere_date(self):
"""
Returns the "originallyAvailableAt" or None
"""
2018-02-12 18:10:39 +11:00
return self.item.get('originallyAvailableAt')
2016-01-02 19:28:31 +11:00
2018-02-12 00:42:49 +11:00
def music_studio(self):
"""
Returns the 'studio' or None
"""
2018-02-12 18:10:39 +11:00
return self.item.get('studio')
2016-03-28 18:56:22 +11:00
2018-02-12 00:42:49 +11:00
def music_studio_list(self):
"""
Returns a list with a single entry for the studio, or an empty list
"""
2015-12-29 04:47:16 +11:00
studio = []
try:
2018-02-12 00:42:49 +11:00
studio.append(self.replace_studio(self.item.attrib['studio']))
2015-12-29 04:47:16 +11:00
except KeyError:
pass
return studio
2018-02-12 00:42:49 +11:00
@staticmethod
def replace_studio(studio_name):
"""
Convert studio for Kodi to properly detect them
"""
2015-12-29 04:47:16 +11:00
studios = {
'abc (us)': "ABC",
'fox (us)': "FOX",
'mtv (us)': "MTV",
'showcase (ca)': "Showcase",
'wgn america': "WGN"
}
2018-02-12 00:42:49 +11:00
return studios.get(studio_name.lower(), studio_name)
2015-12-29 04:47:16 +11:00
2018-02-12 00:42:49 +11:00
@staticmethod
def list_to_string(listobject):
2015-12-29 04:47:16 +11:00
"""
2018-02-12 18:10:39 +11:00
Smart-joins the listobject into a single string using a " / " separator
2015-12-29 04:47:16 +11:00
If the list is empty, smart_join returns an empty string.
"""
string = " / ".join(listobject)
return string
2018-02-12 00:42:49 +11:00
def parent_plex_id(self):
"""
Returns the 'parentRatingKey' as a string or None
"""
2018-02-12 18:10:39 +11:00
return self.item.get('parentRatingKey')
2016-02-03 23:01:13 +11:00
2018-02-12 00:42:49 +11:00
def episode_data(self):
2016-01-10 02:14:02 +11:00
"""
Call on a single episode.
Output: for the corresponding the TV show and season:
[
TV show key, Plex: 'grandparentRatingKey'
TV show title, Plex: 'grandparentTitle'
TV show season, Plex: 'parentIndex'
Episode number, Plex: 'index'
]
"""
2018-02-12 00:42:49 +11:00
return (self.item.get('grandparentRatingKey'),
self.item.get('grandparentTitle'),
self.item.get('parentIndex'),
self.item.get('index'))
2016-01-30 06:07:21 +11:00
2018-02-12 00:42:49 +11:00
@staticmethod
def attach_plex_token_to_url(url):
2016-02-20 02:10:19 +11:00
"""
Returns an extended URL with the Plex token included as 'X-Plex-Token='
url may or may not already contain a '?'
"""
2016-09-03 00:53:16 +10:00
if window('pms_token') == '':
2016-04-06 02:23:00 +10:00
return url
2016-02-20 02:10:19 +11:00
if '?' not in url:
2016-09-03 00:53:16 +10:00
url = "%s?X-Plex-Token=%s" % (url, window('pms_token'))
2016-02-20 02:10:19 +11:00
else:
2016-09-03 00:53:16 +10:00
url = "%s&X-Plex-Token=%s" % (url, window('pms_token'))
2016-02-20 02:10:19 +11:00
return url
2018-02-12 00:42:49 +11:00
def item_id(self):
"""
Returns current playQueueItemID or if unsuccessful the playListItemID
If not found, None is returned
"""
2018-02-12 18:10:39 +11:00
answ = self.item.get('playQueueItemID')
if answ is None:
answ = self.item.get('playListItemID')
return answ
2018-02-12 00:42:49 +11:00
def _data_from_part_or_media(self, key):
"""
Retrieves XML data 'key' first from the active part. If unsuccessful,
tries to retrieve the data from the Media response part.
If all fails, None is returned.
"""
2018-02-12 18:10:39 +11:00
answ = self.item[0][self.part].get(key)
if answ is None:
answ = self.item[0].get(key)
return answ
2018-02-12 00:42:49 +11:00
def video_codec(self):
"""
Returns the video codec and resolution for the child and part selected.
If any data is not found on a part-level, the Media-level data is
returned.
If that also fails (e.g. for old trailers, None is returned)
Output:
{
'videocodec': xxx, e.g. 'h264'
'resolution': xxx, e.g. '720' or '1080'
'height': xxx, e.g. '816'
'width': xxx, e.g. '1920'
'aspectratio': xxx, e.g. '1.78'
2016-05-16 00:30:09 +10:00
'bitrate': xxx, e.g. '10642'
'container': xxx e.g. 'mkv',
'bitDepth': xxx e.g. '8', '10'
}
"""
answ = {
2018-02-12 00:42:49 +11:00
'videocodec': self._data_from_part_or_media('videoCodec'),
'resolution': self._data_from_part_or_media('videoResolution'),
'height': self._data_from_part_or_media('height'),
'width': self._data_from_part_or_media('width'),
'aspectratio': self._data_from_part_or_media('aspectratio'),
'bitrate': self._data_from_part_or_media('bitrate'),
'container': self._data_from_part_or_media('container'),
}
try:
2018-02-12 18:10:39 +11:00
answ['bitDepth'] = self.item[0][self.part][self.mediastream].get(
'bitDepth')
2018-02-12 00:42:49 +11:00
except (TypeError, AttributeError, KeyError, IndexError):
answ['bitDepth'] = None
return answ
2018-02-12 18:10:39 +11:00
def trailer_id(self):
"""
Returns the ratingKey (plex_id) of the trailer or None
"""
for extra in self.item.iterfind('Extras'):
try:
2018-02-12 00:42:49 +11:00
typus = int(extra.attrib['extraType'])
except (KeyError, TypeError):
2018-02-12 00:42:49 +11:00
typus = None
if typus != 1:
continue
2018-02-12 18:10:39 +11:00
return extra.get('ratingKey')
2016-01-12 21:47:48 +11:00
2018-02-12 00:42:49 +11:00
def mediastreams(self):
"""
Returns the media streams for metadata purposes
Output: each track contains a dictionaries
{
2016-01-15 20:04:47 +11:00
'video': videotrack-list, 'codec', 'height', 'width',
'aspect', 'video3DFormat'
'audio': audiotrack-list, 'codec', 'channels',
'language'
'subtitle': list of subtitle languages (or "Unknown")
}
"""
2015-12-29 04:47:16 +11:00
videotracks = []
audiotracks = []
subtitlelanguages = []
try:
# Sometimes, aspectratio is on the "toplevel"
2018-02-12 18:10:39 +11:00
aspect = self.item[0].get('aspectRatio')
except IndexError:
# There is no stream info at all, returning empty
return {
'video': videotracks,
'audio': audiotracks,
'subtitle': subtitlelanguages
}
# Loop over parts
for child in self.item[0]:
2018-02-12 18:10:39 +11:00
container = child.get('container')
# Loop over Streams
for grandchild in child:
stream = grandchild.attrib
media_type = int(stream.get('streamType', 999))
track = {}
if media_type == 1: # Video streams
if 'codec' in stream:
track['codec'] = stream['codec'].lower()
if "msmpeg4" in track['codec']:
track['codec'] = "divx"
elif "mpeg4" in track['codec']:
# if "simple profile" in profile or profile == "":
# track['codec'] = "xvid"
pass
elif "h264" in track['codec']:
if container in ("mp4", "mov", "m4v"):
track['codec'] = "avc1"
track['height'] = stream.get('height')
track['width'] = stream.get('width')
# track['Video3DFormat'] = item.get('Video3DFormat')
track['aspect'] = stream.get('aspectRatio', aspect)
2018-02-12 00:42:49 +11:00
track['duration'] = self.resume_runtime()[1]
track['video3DFormat'] = None
videotracks.append(track)
elif media_type == 2: # Audio streams
if 'codec' in stream:
track['codec'] = stream['codec'].lower()
if ("dca" in track['codec'] and
"ma" in stream.get('profile', '').lower()):
track['codec'] = "dtshd_ma"
track['channels'] = stream.get('channels')
2016-03-10 22:47:30 +11:00
# 'unknown' if we cannot get language
track['language'] = stream.get(
2016-09-03 00:53:16 +10:00
'languageCode', lang(39310)).lower()
audiotracks.append(track)
elif media_type == 3: # Subtitle streams
2016-03-10 22:47:30 +11:00
# 'unknown' if we cannot get language
subtitlelanguages.append(
stream.get('languageCode', lang(39310)).lower())
return {
2015-12-29 04:47:16 +11:00
'video': videotracks,
'audio': audiotracks,
'subtitle': subtitlelanguages
}
2015-12-31 01:57:55 +11:00
2018-02-12 00:42:49 +11:00
def _one_artwork(self, entry):
if entry not in self.item.attrib:
return ''
artwork = self.item.attrib[entry]
if artwork.startswith('http'):
pass
else:
2018-02-12 00:42:49 +11:00
artwork = self.attach_plex_token_to_url(
'%s/photo/:/transcode?width=4000&height=4000&'
'minSize=1&upscale=0&url=%s' % (self.server, artwork))
return artwork
2018-02-12 00:42:49 +11:00
def artwork(self, parent_info=False):
"""
Gets the URLs to the Plex artwork, or empty string if not found.
2018-02-12 00:42:49 +11:00
parent_info=True will check for parent's artwork if None is found
2015-12-31 01:57:55 +11:00
Output:
{
'Primary'
'Art'
'Banner'
'Logo'
'Thumb'
'Disc'
'Backdrop' : LIST with the first entry xml key "art"
}
"""
2015-12-31 01:57:55 +11:00
allartworks = {
2018-02-12 18:10:39 +11:00
'Primary': self._one_artwork('thumb'),
2015-12-31 01:57:55 +11:00
'Art': "",
2018-02-12 18:10:39 +11:00
'Banner': self._one_artwork('banner'),
2015-12-31 01:57:55 +11:00
'Logo': "",
2018-02-12 18:10:39 +11:00
'Thumb': self._one_artwork('grandparentThumb'),
2015-12-31 01:57:55 +11:00
'Disc': "",
2018-02-12 18:10:39 +11:00
'Backdrop': [self._one_artwork('art')]
2015-12-31 01:57:55 +11:00
}
# Process parent items if the main item is missing artwork
2018-02-12 00:42:49 +11:00
if parent_info:
2015-12-31 01:57:55 +11:00
# Process parent backdrops
if not allartworks['Backdrop']:
2018-02-12 18:10:39 +11:00
allartworks['Backdrop'].append(self._one_artwork('parentArt'))
2015-12-31 01:57:55 +11:00
if not allartworks['Primary']:
2018-02-12 00:42:49 +11:00
allartworks['Primary'] = self._one_artwork('parentThumb')
return allartworks
2015-12-31 01:57:55 +11:00
2018-02-12 00:42:49 +11:00
def fanart_artwork(self, allartworks):
"""
Downloads additional fanart from third party sources (well, link to
fanart only).
allartworks = {
'Primary': "",
'Art': "",
'Banner': "",
'Logo': "",
'Thumb': "",
'Disc': "",
'Backdrop': []
}
"""
2018-02-12 00:42:49 +11:00
external_id = self.retrieve_external_item_id()
if external_id is not None:
allartworks = self.lookup_fanart_tv(external_id, allartworks)
return allartworks
2018-02-12 00:42:49 +11:00
def retrieve_external_item_id(self, collection=False):
"""
Returns the item's IMDB id for movies or tvdb id for TV shows
If not found in item's Plex metadata, check themovidedb.org
collection=True will try to return the three-tuple:
collection ID, poster-path, background-path
None is returned if unsuccessful
"""
item = self.item.attrib
media_type = item.get('type')
2018-02-12 00:42:49 +11:00
media_id = None
# Return the saved Plex id's, if applicable
# Always seek collection's ids since not provided by PMS
if collection is False:
if media_type == v.PLEX_TYPE_MOVIE:
2018-02-12 00:42:49 +11:00
media_id = self.provider('imdb')
elif media_type == v.PLEX_TYPE_SHOW:
2018-02-12 00:42:49 +11:00
media_id = self.provider('tvdb')
if media_id is not None:
return media_id
2018-02-01 17:56:54 +11:00
LOG.info('Plex did not provide ID for IMDB or TVDB. Start '
2016-09-03 00:53:16 +10:00
'lookup process')
else:
2018-02-12 18:10:39 +11:00
LOG.info('Start movie set/collection lookup on themoviedb with %s',
2018-02-01 17:56:54 +11:00
item.get('title', ''))
2018-02-12 00:42:49 +11:00
api_key = settings('themoviedbAPIKey')
if media_type == v.PLEX_TYPE_SHOW:
media_type = 'tv'
title = item.get('title', '')
# if the title has the year in remove it as tmdb cannot deal with it...
# replace e.g. 'The Americans (2015)' with 'The Americans'
2017-01-25 04:00:35 +11:00
title = sub(r'\s*\(\d{4}\)$', '', title, count=1)
2017-05-31 18:05:50 +10:00
url = 'https://api.themoviedb.org/3/search/%s' % media_type
parameters = {
2018-02-12 00:42:49 +11:00
'api_key': api_key,
'language': v.KODILANGUAGE,
2018-02-11 22:59:04 +11:00
'query': try_encode(title)
}
2018-02-11 03:59:20 +11:00
data = DU().downloadUrl(url,
authenticate=False,
parameters=parameters,
timeout=7)
try:
data.get('test')
2018-02-12 00:42:49 +11:00
except AttributeError:
2018-02-01 17:56:54 +11:00
LOG.error('Could not download data from FanartTV')
return
if data.get('results') is None:
2018-02-01 17:56:54 +11:00
LOG.info('No match found on themoviedb for type: %s, title: %s',
media_type, title)
return
year = item.get('year')
2018-02-12 00:42:49 +11:00
match_found = None
# find year match
if year is not None:
for entry in data["results"]:
if year in entry.get("first_air_date", ""):
2018-02-12 00:42:49 +11:00
match_found = entry
break
elif year in entry.get("release_date", ""):
2018-02-12 00:42:49 +11:00
match_found = entry
break
# find exact match based on title, if we haven't found a year match
2018-02-12 00:42:49 +11:00
if match_found is None:
2018-02-01 17:56:54 +11:00
LOG.info('No themoviedb match found using year %s', year)
replacements = (
' ',
'-',
'&',
',',
':',
';'
)
for entry in data["results"]:
name = entry.get("name", entry.get("title", ""))
original_name = entry.get("original_name", "")
title_alt = title.lower()
name_alt = name.lower()
org_name_alt = original_name.lower()
2018-02-12 00:42:49 +11:00
for replace_string in replacements:
title_alt = title_alt.replace(replace_string, '')
name_alt = name_alt.replace(replace_string, '')
org_name_alt = org_name_alt.replace(replace_string, '')
if name == title or original_name == title:
# match found for exact title name
2018-02-12 00:42:49 +11:00
match_found = entry
break
elif (name.split(" (")[0] == title or title_alt == name_alt
or title_alt == org_name_alt):
# match found with substituting some stuff
2018-02-12 00:42:49 +11:00
match_found = entry
break
# if a match was not found, we accept the closest match from TMDB
2018-02-12 18:10:39 +11:00
if match_found is None and data.get("results"):
2018-02-01 17:56:54 +11:00
LOG.info('Using very first match from themoviedb')
2018-02-12 00:42:49 +11:00
match_found = entry = data.get("results")[0]
2018-02-12 00:42:49 +11:00
if match_found is None:
2018-02-01 17:56:54 +11:00
LOG.info('Still no themoviedb match for type: %s, title: %s, '
'year: %s', media_type, title, year)
LOG.debug('themoviedb answer was %s', data['results'])
return
2018-02-01 17:56:54 +11:00
LOG.info('Found themoviedb match for %s: %s',
2018-02-12 00:42:49 +11:00
item.get('title'), match_found)
2018-02-12 00:42:49 +11:00
tmdb_id = str(entry.get("id", ""))
if tmdb_id == '':
2018-02-01 17:56:54 +11:00
LOG.error('No themoviedb ID found, aborting')
return
if media_type == "multi" and entry.get("media_type"):
media_type = entry.get("media_type")
name = entry.get("name", entry.get("title"))
2018-02-12 00:42:49 +11:00
# lookup external tmdb_id and perform artwork lookup on fanart.tv
parameters = {
2018-02-12 00:42:49 +11:00
'api_key': api_key
}
for language in [v.KODILANGUAGE, "en"]:
parameters['language'] = language
if media_type == "movie":
2018-02-12 00:42:49 +11:00
url = 'https://api.themoviedb.org/3/movie/%s' % tmdb_id
parameters['append_to_response'] = 'videos'
elif media_type == "tv":
2018-02-12 00:42:49 +11:00
url = 'https://api.themoviedb.org/3/tv/%s' % tmdb_id
parameters['append_to_response'] = 'external_ids,videos'
2018-02-11 03:59:20 +11:00
data = DU().downloadUrl(url,
authenticate=False,
parameters=parameters,
timeout=7)
try:
data.get('test')
2018-02-12 00:42:49 +11:00
except AttributeError:
2018-02-01 17:56:54 +11:00
LOG.error('Could not download %s with parameters %s',
url, parameters)
continue
2016-06-09 04:34:49 +10:00
if collection is False:
if data.get("imdb_id") is not None:
2018-02-12 00:42:49 +11:00
media_id = str(data.get("imdb_id"))
2016-06-07 06:10:58 +10:00
break
if data.get("external_ids") is not None:
2018-02-12 00:42:49 +11:00
media_id = str(data["external_ids"].get("tvdb_id"))
2016-06-07 06:10:58 +10:00
break
else:
if data.get("belongs_to_collection") is None:
continue
2018-02-12 00:42:49 +11:00
media_id = str(data.get("belongs_to_collection").get("id"))
2018-02-01 17:56:54 +11:00
LOG.debug('Retrieved collections tmdb id %s for %s',
2018-02-12 00:42:49 +11:00
media_id, title)
url = 'https://api.themoviedb.org/3/collection/%s' % media_id
2018-02-11 03:59:20 +11:00
data = DU().downloadUrl(url,
authenticate=False,
parameters=parameters,
timeout=7)
try:
data.get('poster_path')
except AttributeError:
2018-02-01 17:56:54 +11:00
LOG.info('Could not find TheMovieDB poster paths for %s in'
'the language %s', title, language)
continue
else:
2018-02-12 00:42:49 +11:00
poster = ('https://image.tmdb.org/t/p/original%s' %
data.get('poster_path'))
background = ('https://image.tmdb.org/t/p/original%s' %
data.get('backdrop_path'))
media_id = media_id, poster, background
break
2018-02-12 00:42:49 +11:00
return media_id
2016-06-07 06:10:58 +10:00
2018-02-12 00:42:49 +11:00
def lookup_fanart_tv(self, media_id, allartworks, set_info=False):
"""
perform artwork lookup on fanart.tv
2018-02-12 00:42:49 +11:00
media_id: IMDB id for movies, tvdb id for TV shows
"""
item = self.item.attrib
2016-09-03 00:53:16 +10:00
api_key = settings('FanArtTVAPIKey')
typus = item.get('type')
if typus == 'show':
typus = 'tv'
if typus == "movie":
url = 'http://webservice.fanart.tv/v3/movies/%s?api_key=%s' \
2018-02-12 00:42:49 +11:00
% (media_id, api_key)
elif typus == 'tv':
url = 'http://webservice.fanart.tv/v3/tv/%s?api_key=%s' \
2018-02-12 00:42:49 +11:00
% (media_id, api_key)
else:
# Not supported artwork
return allartworks
2018-02-11 03:59:20 +11:00
data = DU().downloadUrl(url,
authenticate=False,
timeout=15)
try:
data.get('test')
2018-02-12 00:42:49 +11:00
except AttributeError:
2018-02-01 17:56:54 +11:00
LOG.error('Could not download data from FanartTV')
return allartworks
2018-02-12 00:42:49 +11:00
fanart_tv_types = list(FANART_TV_TYPES)
if typus == "artist":
2018-02-12 00:42:49 +11:00
fanart_tv_types.append(("thumb", "folder"))
else:
2018-02-12 00:42:49 +11:00
fanart_tv_types.append(("thumb", "Thumb"))
if set_info:
fanart_tv_types.append(("poster", "Primary"))
2016-06-07 06:10:58 +10:00
prefixes = (
"hd" + typus,
"hd",
typus,
"",
)
2018-02-12 00:42:49 +11:00
for fanarttype in fanart_tv_types:
# Skip the ones we already have
if allartworks.get(fanarttype[1]):
continue
for prefix in prefixes:
fanarttvimage = prefix + fanarttype[0]
if fanarttvimage not in data:
continue
# select image in preferred language
for entry in data[fanarttvimage]:
if entry.get("lang") == v.KODILANGUAGE:
2018-02-12 18:10:39 +11:00
allartworks[fanarttype[1]] = \
entry.get("url", "").replace(' ', '%20')
break
# just grab the first english OR undefinded one as fallback
# (so we're actually grabbing the more popular one)
if not allartworks.get(fanarttype[1]):
for entry in data[fanarttvimage]:
if entry.get("lang") in ("en", "00"):
2018-02-12 18:10:39 +11:00
allartworks[fanarttype[1]] = \
entry.get("url", "").replace(' ', '%20')
break
# grab extrafanarts in list
maxfanarts = 10
fanartcount = 0
for prefix in prefixes:
fanarttvimage = prefix + 'background'
if fanarttvimage not in data:
continue
for entry in data[fanarttvimage]:
if entry.get("url") is None:
continue
if fanartcount > maxfanarts:
break
allartworks['Backdrop'].append(
entry['url'].replace(' ', '%20'))
fanartcount += 1
return allartworks
2016-01-02 00:40:40 +11:00
2018-02-12 00:42:49 +11:00
def set_artwork(self):
2016-06-07 06:10:58 +10:00
"""
Gets the URLs to the Plex artwork, or empty string if not found.
parentInfo=True will check for parent's artwork if None is found
Only call on movies
2016-06-07 06:10:58 +10:00
Output:
{
'Primary'
'Art'
'Banner'
'Logo'
'Thumb'
'Disc'
'Backdrop' : LIST with the first entry xml key "art"
}
"""
allartworks = {
'Primary': "",
'Art': "",
'Banner': "",
'Logo': "",
'Thumb': "",
'Disc': "",
'Backdrop': []
}
# Plex does not get much artwork - go ahead and get the rest from
# fanart tv only for movie or tv show
2018-02-12 00:42:49 +11:00
external_id = self.retrieve_external_item_id(collection=True)
if external_id is not None:
try:
2018-02-12 00:42:49 +11:00
external_id, poster, background = external_id
except TypeError:
poster, background = None, None
if poster is not None:
allartworks['Primary'] = poster
if background is not None:
allartworks['Backdrop'].append(background)
2018-02-12 00:42:49 +11:00
allartworks = self.lookup_fanart_tv(external_id,
allartworks,
set_info=True)
else:
2018-02-01 17:56:54 +11:00
LOG.info('Did not find a set/collection ID on TheMovieDB using %s.'
2018-02-12 00:42:49 +11:00
' Artwork will be missing.', self.titles()[0])
2017-05-31 20:57:53 +10:00
return allartworks
2016-06-07 06:10:58 +10:00
2018-02-12 00:42:49 +11:00
def should_stream(self):
2016-04-17 21:36:41 +10:00
"""
Returns True if the item's 'optimizedForStreaming' is set, False other-
wise
"""
2018-02-12 18:10:39 +11:00
return self.item[0].get('optimizedForStreaming') == '1'
2016-04-17 21:36:41 +10:00
2018-02-12 00:42:49 +11:00
def mediastream_number(self):
"""
Returns the Media stream as an int (mostly 0). Will let the user choose
if several media streams are present for a PMS item (if settings are
set accordingly)
"""
# How many streams do we have?
count = 0
2018-02-12 18:10:39 +11:00
for entry in self.item.iterfind('./Media'):
count += 1
if (count > 1 and (
2018-02-12 00:42:49 +11:00
(self.plex_type() != 'clip' and
2016-09-03 00:53:16 +10:00
settings('bestQuality') == 'false')
2018-02-12 18:10:39 +11:00
or
2018-02-12 00:42:49 +11:00
(self.plex_type() == 'clip' and
2016-09-03 00:53:16 +10:00
settings('bestTrailer') == 'false'))):
# Several streams/files available.
dialoglist = []
2018-02-12 18:10:39 +11:00
for entry in self.item.iterfind('./Media'):
# Get additional info (filename / languages)
2017-04-03 04:10:10 +10:00
filename = None
if 'file' in entry[0].attrib:
filename = basename(entry[0].attrib['file'])
2017-04-03 04:10:10 +10:00
# Languages of audio streams
languages = []
for stream in entry[0]:
if (stream.attrib['streamType'] == '1' and
'language' in stream.attrib):
languages.append(stream.attrib['language'])
languages = ', '.join(languages)
if filename:
2018-02-11 22:59:04 +11:00
option = try_encode(filename)
2017-04-03 04:10:10 +10:00
if languages:
if option:
2018-02-11 22:59:04 +11:00
option = '%s (%s): ' % (option, try_encode(languages))
2017-04-03 04:10:10 +10:00
else:
2018-02-11 22:59:04 +11:00
option = '%s: ' % try_encode(languages)
2017-04-03 04:10:10 +10:00
if 'videoResolution' in entry.attrib:
option = '%s%sp ' % (option,
2018-02-12 18:10:39 +11:00
entry.get('videoResolution'))
2017-04-03 04:10:10 +10:00
if 'videoCodec' in entry.attrib:
option = '%s%s' % (option,
2018-02-12 18:10:39 +11:00
entry.get('videoCodec'))
2017-04-03 04:10:10 +10:00
option = option.strip() + ' - '
if 'audioProfile' in entry.attrib:
option = '%s%s ' % (option,
2018-02-12 18:10:39 +11:00
entry.get('audioProfile'))
2017-04-03 04:10:10 +10:00
if 'audioCodec' in entry.attrib:
option = '%s%s ' % (option,
2018-02-12 18:10:39 +11:00
entry.get('audioCodec'))
2017-04-03 04:10:10 +10:00
dialoglist.append(option)
2018-02-11 03:59:20 +11:00
media = dialog('select', 'Select stream', dialoglist)
else:
media = 0
self.mediastream = media
return media
2018-02-12 00:42:49 +11:00
def transcode_video_path(self, action, quality=None):
2016-01-02 00:40:40 +11:00
"""
2016-02-03 23:01:13 +11:00
To be called on a VIDEO level of PMS xml response!
Transcode Video support; returns the URL to get a media started
2016-01-02 00:40:40 +11:00
Input:
2016-04-12 02:57:20 +10:00
action 'DirectStream' or 'Transcode'
2016-01-04 05:17:59 +11:00
quality: {
2016-02-05 06:23:04 +11:00
'videoResolution': e.g. '1024x768',
'videoQuality': e.g. '60',
'maxVideoBitrate': e.g. '2000' (in kbits)
}
(one or several of these options)
2016-01-02 00:40:40 +11:00
Output:
final URL to pull in PMS transcoder
2016-01-04 05:17:59 +11:00
TODO: mediaIndex
2016-01-02 00:40:40 +11:00
"""
if self.mediastream is None:
2018-02-12 00:42:49 +11:00
self.mediastream_number()
if quality is None:
quality = {}
2017-01-25 02:53:50 +11:00
xargs = client.getXArgsDeviceInfo()
2016-02-05 06:23:04 +11:00
# For DirectPlay, path/key of PART is needed
# trailers are 'clip' with PMS xmls
2016-04-12 02:57:20 +10:00
if action == "DirectStream":
path = self.item[self.mediastream][self.part].attrib['key']
2016-02-03 23:01:13 +11:00
url = self.server + path
2016-02-07 22:38:50 +11:00
# e.g. Trailers already feature an '?'!
2016-02-03 23:01:13 +11:00
if '?' in url:
url += '&' + urlencode(xargs)
else:
url += '?' + urlencode(xargs)
return url
2016-04-14 00:14:55 +10:00
# For Transcoding
headers = {
'X-Plex-Platform': 'Android',
'X-Plex-Platform-Version': '7.0',
'X-Plex-Product': 'Plex for Android',
'X-Plex-Version': '5.8.0.475'
}
2016-02-05 06:23:04 +11:00
# Path/key to VIDEO item of xml PMS response is needed, not part
path = self.item.attrib['key']
2018-02-12 00:42:49 +11:00
transcode_path = self.server + \
2016-02-07 22:38:50 +11:00
'/video/:/transcode/universal/start.m3u8?'
2016-01-02 00:40:40 +11:00
args = {
'audioBoost': settings('audioBoost'),
'autoAdjustQuality': 0,
'directPlay': 0,
'directStream': 1,
'protocol': 'hls', # seen in the wild: 'dash', 'http', 'hls'
2018-02-12 18:10:39 +11:00
'session': window('plex_client_Id'),
'fastSeek': 1,
2016-01-02 00:40:40 +11:00
'path': path,
'mediaIndex': self.mediastream,
'partIndex': self.part,
'hasMDE': 1,
'location': 'lan',
2017-05-02 03:51:10 +10:00
'subtitleSize': settings('subtitleSize')
2016-01-02 00:40:40 +11:00
}
# Look like Android to let the PMS use the transcoding profile
xargs.update(headers)
2018-02-01 17:56:54 +11:00
LOG.debug("Setting transcode quality to: %s", quality)
args.update(quality)
2018-02-12 00:42:49 +11:00
url = transcode_path + urlencode(xargs) + '&' + urlencode(args)
return url
2016-01-02 00:40:40 +11:00
2018-02-12 00:42:49 +11:00
def cache_external_subs(self):
"""
Downloads external subtitles temporarily to Kodi and returns a list
of their paths
"""
2016-01-02 00:40:40 +11:00
externalsubs = []
try:
2016-11-06 02:38:56 +11:00
mediastreams = self.item[0][self.part]
2016-01-02 00:40:40 +11:00
except (TypeError, KeyError, IndexError):
return
kodiindex = 0
fileindex = 0
2016-01-02 00:40:40 +11:00
for stream in mediastreams:
# Since plex returns all possible tracks together, have to pull
# only external subtitles - only for these a 'key' exists
2018-02-12 18:10:39 +11:00
if stream.get('streamType') != "3":
# Not a subtitle
continue
# Only set for additional external subtitles NOT lying beside video
2018-02-12 18:10:39 +11:00
key = stream.get('key')
# Only set for dedicated subtitle files lying beside video
# ext = stream.attrib.get('format')
if key:
# We do know the language - temporarily download
2018-02-12 18:10:39 +11:00
if stream.get('languageCode') is not None:
2017-05-02 03:51:10 +10:00
path = self.download_external_subtitles(
"{server}%s" % key,
"subtitle%02d.%s.%s" % (fileindex,
stream.attrib['languageCode'],
stream.attrib['codec']))
fileindex += 1
# We don't know the language - no need to download
else:
2018-02-12 00:42:49 +11:00
path = self.attach_plex_token_to_url(
"%s%s" % (self.server, key))
externalsubs.append(path)
2016-01-02 00:40:40 +11:00
kodiindex += 1
2018-02-01 17:56:54 +11:00
LOG.info('Found external subs: %s', externalsubs)
2016-01-02 00:40:40 +11:00
return externalsubs
2016-03-15 03:47:05 +11:00
@staticmethod
2017-05-02 03:51:10 +10:00
def download_external_subtitles(url, filename):
"""
One cannot pass the subtitle language for ListItems. Workaround; will
download the subtitle at url to the Kodi PKC directory in a temp dir
Returns the path to the downloaded subtitle or None
"""
if not exists_dir(v.EXTERNAL_SUBTITLE_TEMP_PATH):
makedirs(v.EXTERNAL_SUBTITLE_TEMP_PATH)
path = join(v.EXTERNAL_SUBTITLE_TEMP_PATH, filename)
2018-02-12 00:42:49 +11:00
response = DU().downloadUrl(url, return_response=True)
try:
2018-02-12 00:42:49 +11:00
response.status_code
except AttributeError:
2018-02-01 17:56:54 +11:00
LOG.error('Could not temporarily download subtitle %s', url)
return
else:
2018-02-01 17:56:54 +11:00
LOG.debug('Writing temp subtitle to %s', path)
try:
2018-02-12 00:42:49 +11:00
with open(path, 'wb') as filer:
filer.write(response.content)
except UnicodeEncodeError:
2018-02-01 17:56:54 +11:00
LOG.debug('Need to slugify the filename %s', path)
2018-01-08 03:50:30 +11:00
path = slugify(path)
2018-02-12 00:42:49 +11:00
with open(path, 'wb') as filer:
filer.write(response.content)
return path
2018-02-12 00:42:49 +11:00
def kodi_premiere_date(self):
2016-06-13 01:22:22 +10:00
"""
Takes Plex' originallyAvailableAt of the form "yyyy-mm-dd" and returns
2018-02-12 00:42:49 +11:00
Kodi's "dd.mm.yyyy" or None
2016-06-13 01:22:22 +10:00
"""
2018-02-12 00:42:49 +11:00
date = self.premiere_date()
if date is None:
return
2016-06-13 01:22:22 +10:00
try:
2017-01-25 04:00:35 +11:00
date = sub(r'(\d+)-(\d+)-(\d+)', r'\3.\2.\1', date)
2016-06-13 01:22:22 +10:00
except:
date = None
return date
2016-06-13 01:22:22 +10:00
2018-02-12 00:42:49 +11:00
def create_listitem(self, listitem=None, append_show_title=False,
append_sxxexx=False):
"""
Return a xbmcgui.ListItem() for this Plex item
"""
if self.plex_type() == v.PLEX_TYPE_PHOTO:
listitem = self._create_photo_listitem(listitem)
2017-03-19 22:14:16 +11:00
# Only set the bare minimum of artwork
2018-02-12 00:42:49 +11:00
listitem.setArt({'icon': 'DefaultPicture.png',
'fanart': self._one_artwork('thumb')})
2017-03-19 22:14:16 +11:00
else:
2018-02-12 00:42:49 +11:00
listitem = self._create_video_listitem(listitem,
append_show_title,
append_sxxexx)
self.add_video_streams(listitem)
self.set_listitem_artwork(listitem)
return listitem
2017-03-19 22:14:16 +11:00
2018-02-12 00:42:49 +11:00
def _create_photo_listitem(self, listitem=None):
"""
Use for photo items only
2016-03-15 03:47:05 +11:00
"""
2018-02-12 00:42:49 +11:00
title, _ = self.titles()
if listitem is None:
listitem = ListItem(title)
else:
2018-02-12 00:42:49 +11:00
listitem.setLabel(title)
metadata = {
2018-02-12 00:42:49 +11:00
'date': self.kodi_premiere_date(),
2018-02-12 18:10:39 +11:00
'size': long(self.item[0][0].get('size', 0)),
'exif:width': self.item[0].get('width', ''),
'exif:height': self.item[0].get('height', ''),
}
2018-02-12 00:42:49 +11:00
listitem.setInfo(type='image', infoLabels=metadata)
listitem.setProperty('plot', self.plot())
listitem.setProperty('plexid', self.plex_id())
return listitem
2018-02-12 00:42:49 +11:00
def _create_video_listitem(self,
listitem=None,
append_show_title=False,
append_sxxexx=False):
"""
Use for video items only
2016-03-15 03:47:05 +11:00
Call on a child level of PMS xml response (e.g. in a for loop)
2018-02-12 00:42:49 +11:00
listitem : existing xbmcgui.ListItem to work with
otherwise, a new one is created
2018-02-12 00:42:49 +11:00
append_show_title : True to append TV show title to episode title
append_sxxexx : True to append SxxExx to episode title
2016-03-15 03:47:05 +11:00
Returns XBMC listitem for this PMS library item
"""
2018-02-12 00:42:49 +11:00
title, sorttitle = self.titles()
typus = self.plex_type()
2016-03-15 03:47:05 +11:00
2018-02-12 00:42:49 +11:00
if listitem is None:
listitem = ListItem(title)
2017-03-19 22:14:16 +11:00
else:
2018-02-12 00:42:49 +11:00
listitem.setLabel(title)
2017-03-19 22:54:29 +11:00
# Necessary; Kodi won't start video otherwise!
2018-02-12 00:42:49 +11:00
listitem.setProperty('IsPlayable', 'true')
# Video items, e.g. movies and episodes or clips
2018-02-12 00:42:49 +11:00
people = self.people()
userdata = self.userdata()
metadata = {
2018-02-12 00:42:49 +11:00
'genre': self.list_to_string(self.genre_list()),
'year': self.year(),
'rating': self.audience_rating(),
'playcount': userdata['PlayCount'],
'cast': people['Cast'],
2018-02-12 00:42:49 +11:00
'director': self.list_to_string(people.get('Director')),
'plot': self.plot(),
'sorttitle': sorttitle,
'duration': userdata['Runtime'],
2018-02-12 00:42:49 +11:00
'studio': self.list_to_string(self.music_studio_list()),
'tagline': self.tagline(),
'writer': self.list_to_string(people.get('Writer')),
'premiered': self.premiere_date(),
'dateadded': self.date_created(),
'lastplayed': userdata['LastPlayedDate'],
2018-02-12 00:42:49 +11:00
'mpaa': self.content_rating(),
'aired': self.premiere_date()
}
2018-01-22 21:20:37 +11:00
# Do NOT set resumetime - otherwise Kodi always resumes at that time
# even if the user chose to start element from the beginning
2018-02-12 00:42:49 +11:00
# listitem.setProperty('resumetime', str(userdata['Resume']))
listitem.setProperty('totaltime', str(userdata['Runtime']))
2016-03-15 03:47:05 +11:00
2017-03-19 22:14:16 +11:00
if typus == v.PLEX_TYPE_EPISODE:
2018-02-12 00:42:49 +11:00
_, show, season, episode = self.episode_data()
season = -1 if season is None else int(season)
episode = -1 if episode is None else int(episode)
metadata['episode'] = episode
metadata['season'] = season
2016-03-15 03:47:05 +11:00
metadata['tvshowtitle'] = show
if season and episode:
2018-02-12 00:42:49 +11:00
listitem.setProperty('episodeno',
"s%.2de%.2d" % (season, episode))
2018-02-12 00:42:49 +11:00
if append_sxxexx is True:
title = "S%.2dE%.2d - %s" % (season, episode, title)
2018-02-12 00:42:49 +11:00
listitem.setArt({'icon': 'DefaultTVShows.png'})
if append_show_title is True:
title = "%s - %s " % (show, title)
2018-02-12 00:42:49 +11:00
if append_show_title or append_sxxexx:
listitem.setLabel(title)
2017-03-19 22:14:16 +11:00
elif typus == v.PLEX_TYPE_MOVIE:
2018-02-12 00:42:49 +11:00
listitem.setArt({'icon': 'DefaultMovies.png'})
else:
2016-06-13 01:22:22 +10:00
# E.g. clips, trailers, ...
2018-02-12 00:42:49 +11:00
listitem.setArt({'icon': 'DefaultVideo.png'})
2018-02-12 00:42:49 +11:00
plex_id = self.plex_id()
listitem.setProperty('plexid', plex_id)
with plexdb.Get_Plex_DB() as plex_db:
try:
2018-02-12 00:42:49 +11:00
listitem.setProperty('dbid',
str(plex_db.getItem_byId(plex_id)[0]))
except TypeError:
pass
# Expensive operation
metadata['title'] = title
2018-02-12 00:42:49 +11:00
listitem.setInfo('video', infoLabels=metadata)
2017-08-03 02:54:05 +10:00
try:
# Add context menu entry for information screen
2018-02-12 00:42:49 +11:00
listitem.addContextMenuItems([(lang(30032), 'XBMC.Action(Info)',)])
2017-08-03 02:54:05 +10:00
except TypeError:
# Kodi fuck-up
pass
2018-02-12 00:42:49 +11:00
return listitem
2016-03-15 03:47:05 +11:00
2018-02-12 00:42:49 +11:00
def add_video_streams(self, listitem):
2016-03-15 03:47:05 +11:00
"""
Add media stream information to xbmcgui.ListItem
"""
2018-02-12 00:42:49 +11:00
for key, value in self.mediastreams().iteritems():
if value:
2018-02-12 00:42:49 +11:00
listitem.addStreamInfo(key, value)
2016-03-17 04:33:18 +11:00
2018-02-12 00:42:49 +11:00
def validate_playurl(self, path, typus, force_check=False, folder=False,
omit_check=False):
2016-03-17 04:33:18 +11:00
"""
Returns a valid path for Kodi, e.g. with '\' substituted to '\\' in
Unicode. Returns None if this is not possible
path : Unicode
typus : Plex type from PMS xml
2018-02-12 00:42:49 +11:00
force_check : Will always try to check validity of path
Will also skip confirmation dialog if path not found
folder : Set to True if path is a folder
2018-02-12 00:42:49 +11:00
omit_check : Will entirely omit validity check if True
2016-03-17 04:33:18 +11:00
"""
if path is None:
return None
typus = v.REMAP_TYPE_FROM_PLEXTYPE[typus]
if state.REMAP_PATH is True:
path = path.replace(getattr(state, 'remapSMB%sOrg' % typus),
getattr(state, 'remapSMB%sNew' % typus),
1)
2016-03-17 04:33:18 +11:00
# There might be backslashes left over:
path = path.replace('\\', '/')
elif state.REPLACE_SMB_PATH is True:
if path.startswith('\\\\'):
path = 'smb:' + path.replace('\\', '/')
2018-02-12 00:42:49 +11:00
if ((state.PATH_VERIFIED and force_check is False) or
omit_check is True):
return path
# exist() needs a / or \ at the end to work for directories
if folder is False:
# files
2018-02-11 22:59:04 +11:00
check = exists(try_encode(path))
else:
# directories
if "\\" in path:
if not path.endswith('\\'):
# Add the missing backslash
check = exists_dir(path + "\\")
else:
check = exists_dir(path)
else:
if not path.endswith('/'):
check = exists_dir(path + "/")
else:
check = exists_dir(path)
if not check:
2018-02-12 00:42:49 +11:00
if force_check is False:
# Validate the path is correct with user intervention
2018-02-12 00:42:49 +11:00
if self.ask_to_validate(path):
state.STOP_SYNC = True
path = None
2017-08-19 23:03:19 +10:00
state.PATH_VERIFIED = True
else:
path = None
2018-02-12 00:42:49 +11:00
elif force_check is False:
2017-08-19 23:03:19 +10:00
# Only set the flag if we were not force-checking the path
state.PATH_VERIFIED = True
return path
2016-03-17 04:33:18 +11:00
2018-02-12 00:42:49 +11:00
@staticmethod
def ask_to_validate(url):
2016-03-17 04:33:18 +11:00
"""
Displays a YESNO dialog box:
Kodi can't locate file: <url>. Please verify the path.
You may need to verify your network credentials in the
add-on settings or use different Plex paths. Stop syncing?
Returns True if sync should stop, else False
"""
2018-02-01 17:56:54 +11:00
LOG.warn('Cannot access file: %s', url)
2018-02-11 03:59:20 +11:00
resp = dialog('yesno',
heading=lang(29999),
line1=lang(39031) + url,
line2=lang(39032))
2016-03-17 04:33:18 +11:00
return resp
2017-01-09 01:03:41 +11:00
def set_listitem_artwork(self, listitem):
"""
Set all artwork to the listitem
"""
2018-02-12 00:42:49 +11:00
allartwork = self.artwork(parent_info=True)
2017-01-09 01:03:41 +11:00
arttypes = {
'poster': "Primary",
'tvshow.poster': "Thumb",
'clearart': "Primary",
'tvshow.clearart': "Primary",
'clearlogo': "Logo",
'tvshow.clearlogo': "Logo",
'discart': "Disc",
'fanart_image': "Backdrop",
'landscape': "Backdrop",
"banner": "Banner"
}
for arttype in arttypes:
art = arttypes[arttype]
if art == "Backdrop":
2018-02-12 00:42:49 +11:00
# Backdrop is a list, grab the first backdrop
self._set_listitem_artprop(listitem,
arttype,
allartwork[art][0])
2017-01-09 01:03:41 +11:00
else:
self._set_listitem_artprop(listitem, arttype, allartwork[art])
2018-02-12 00:42:49 +11:00
@staticmethod
def _set_listitem_artprop(listitem, arttype, path):
2017-01-09 01:03:41 +11:00
if arttype in (
'thumb', 'fanart_image', 'small_poster', 'tiny_poster',
'medium_landscape', 'medium_poster', 'small_fanartimage',
'medium_fanartimage', 'fanart_noindicators'):
listitem.setProperty(arttype, path)
else:
listitem.setArt({arttype: path})