mirror of
https://github.com/jellyfin/jellyfin-kodi.git
synced 2025-06-23 00:20:31 +00:00
Tool black: auto-format Python code
This commit is contained in:
parent
e4d8084c25
commit
7763762212
54 changed files with 6545 additions and 4723 deletions
|
|
@ -14,100 +14,102 @@ LOG = LazyLogger(__name__)
|
|||
|
||||
class API(object):
|
||||
def __init__(self, item, server=None):
|
||||
|
||||
''' Get item information in special cases.
|
||||
server is the server address, provide if your functions requires it.
|
||||
'''
|
||||
"""Get item information in special cases.
|
||||
server is the server address, provide if your functions requires it.
|
||||
"""
|
||||
self.item = item
|
||||
self.server = server
|
||||
|
||||
def get_playcount(self, played, playcount):
|
||||
|
||||
''' Convert Jellyfin played/playcount into
|
||||
the Kodi equivalent. The playcount is tied to the watch status.
|
||||
'''
|
||||
"""Convert Jellyfin played/playcount into
|
||||
the Kodi equivalent. The playcount is tied to the watch status.
|
||||
"""
|
||||
return (playcount or 1) if played else None
|
||||
|
||||
def get_naming(self):
|
||||
|
||||
if self.item['Type'] == 'Episode' and 'SeriesName' in self.item:
|
||||
return "%s: %s" % (self.item['SeriesName'], self.item['Name'])
|
||||
if self.item["Type"] == "Episode" and "SeriesName" in self.item:
|
||||
return "%s: %s" % (self.item["SeriesName"], self.item["Name"])
|
||||
|
||||
elif self.item['Type'] == 'MusicAlbum' and 'AlbumArtist' in self.item:
|
||||
return "%s: %s" % (self.item['AlbumArtist'], self.item['Name'])
|
||||
elif self.item["Type"] == "MusicAlbum" and "AlbumArtist" in self.item:
|
||||
return "%s: %s" % (self.item["AlbumArtist"], self.item["Name"])
|
||||
|
||||
elif self.item['Type'] == 'Audio' and self.item.get('Artists'):
|
||||
return "%s: %s" % (self.item['Artists'][0], self.item['Name'])
|
||||
elif self.item["Type"] == "Audio" and self.item.get("Artists"):
|
||||
return "%s: %s" % (self.item["Artists"][0], self.item["Name"])
|
||||
|
||||
return self.item['Name']
|
||||
return self.item["Name"]
|
||||
|
||||
def get_actors(self):
|
||||
cast = []
|
||||
|
||||
if 'People' in self.item:
|
||||
self.get_people_artwork(self.item['People'])
|
||||
if "People" in self.item:
|
||||
self.get_people_artwork(self.item["People"])
|
||||
|
||||
for person in self.item['People']:
|
||||
for person in self.item["People"]:
|
||||
|
||||
if person['Type'] == "Actor":
|
||||
cast.append({
|
||||
'name': person['Name'],
|
||||
'role': person.get('Role', "Unknown"),
|
||||
'order': len(cast) + 1,
|
||||
'thumbnail': person['imageurl']
|
||||
})
|
||||
if person["Type"] == "Actor":
|
||||
cast.append(
|
||||
{
|
||||
"name": person["Name"],
|
||||
"role": person.get("Role", "Unknown"),
|
||||
"order": len(cast) + 1,
|
||||
"thumbnail": person["imageurl"],
|
||||
}
|
||||
)
|
||||
|
||||
return cast
|
||||
|
||||
def media_streams(self, video, audio, subtitles):
|
||||
return {
|
||||
'video': video or [],
|
||||
'audio': audio or [],
|
||||
'subtitle': subtitles or []
|
||||
}
|
||||
return {"video": video or [], "audio": audio or [], "subtitle": subtitles or []}
|
||||
|
||||
def video_streams(self, tracks, container=None):
|
||||
|
||||
if container:
|
||||
container = container.split(',')[0]
|
||||
container = container.split(",")[0]
|
||||
|
||||
for track in tracks:
|
||||
|
||||
if "DvProfile" in track:
|
||||
track['hdrtype'] = "dolbyvision"
|
||||
elif track.get('VideoRangeType', '') in ["HDR10", "HDR10Plus"]:
|
||||
track['hdrtype'] = "hdr10"
|
||||
elif "HLG" in track.get('VideoRangeType', ''):
|
||||
track['hdrtype'] = "hlg"
|
||||
track["hdrtype"] = "dolbyvision"
|
||||
elif track.get("VideoRangeType", "") in ["HDR10", "HDR10Plus"]:
|
||||
track["hdrtype"] = "hdr10"
|
||||
elif "HLG" in track.get("VideoRangeType", ""):
|
||||
track["hdrtype"] = "hlg"
|
||||
|
||||
track.update({
|
||||
'hdrtype': track.get('hdrtype', "").lower(),
|
||||
'codec': track.get('Codec', "").lower(),
|
||||
'profile': track.get('Profile', "").lower(),
|
||||
'height': track.get('Height'),
|
||||
'width': track.get('Width'),
|
||||
'3d': self.item.get('Video3DFormat'),
|
||||
'aspect': 1.85
|
||||
})
|
||||
track.update(
|
||||
{
|
||||
"hdrtype": track.get("hdrtype", "").lower(),
|
||||
"codec": track.get("Codec", "").lower(),
|
||||
"profile": track.get("Profile", "").lower(),
|
||||
"height": track.get("Height"),
|
||||
"width": track.get("Width"),
|
||||
"3d": self.item.get("Video3DFormat"),
|
||||
"aspect": 1.85,
|
||||
}
|
||||
)
|
||||
|
||||
if "msmpeg4" in track['codec']:
|
||||
track['codec'] = "divx"
|
||||
if "msmpeg4" in track["codec"]:
|
||||
track["codec"] = "divx"
|
||||
|
||||
elif "mpeg4" in track['codec'] and ("simple profile" in track['profile'] or not track['profile']):
|
||||
track['codec'] = "xvid"
|
||||
elif "mpeg4" in track["codec"] and (
|
||||
"simple profile" in track["profile"] or not track["profile"]
|
||||
):
|
||||
track["codec"] = "xvid"
|
||||
|
||||
elif "h264" in track['codec'] and container in ('mp4', 'mov', 'm4v'):
|
||||
track['codec'] = "avc1"
|
||||
elif "h264" in track["codec"] and container in ("mp4", "mov", "m4v"):
|
||||
track["codec"] = "avc1"
|
||||
|
||||
try:
|
||||
width, height = self.item.get('AspectRatio', track.get('AspectRatio', "0")).split(':')
|
||||
track['aspect'] = round(float(width) / float(height), 6)
|
||||
width, height = self.item.get(
|
||||
"AspectRatio", track.get("AspectRatio", "0")
|
||||
).split(":")
|
||||
track["aspect"] = round(float(width) / float(height), 6)
|
||||
except (ValueError, ZeroDivisionError):
|
||||
|
||||
if track['width'] and track['height']:
|
||||
track['aspect'] = round(float(track['width'] / track['height']), 6)
|
||||
if track["width"] and track["height"]:
|
||||
track["aspect"] = round(float(track["width"] / track["height"]), 6)
|
||||
|
||||
track['duration'] = self.get_runtime()
|
||||
track["duration"] = self.get_runtime()
|
||||
|
||||
return tracks
|
||||
|
||||
|
|
@ -115,28 +117,30 @@ class API(object):
|
|||
|
||||
for track in tracks:
|
||||
|
||||
track.update({
|
||||
'codec': track.get('Codec', "").lower(),
|
||||
'profile': track.get('Profile', "").lower(),
|
||||
'channels': track.get('Channels'),
|
||||
'language': track.get('Language')
|
||||
})
|
||||
track.update(
|
||||
{
|
||||
"codec": track.get("Codec", "").lower(),
|
||||
"profile": track.get("Profile", "").lower(),
|
||||
"channels": track.get("Channels"),
|
||||
"language": track.get("Language"),
|
||||
}
|
||||
)
|
||||
|
||||
if "dts-hd ma" in track['profile']:
|
||||
track['codec'] = "dtshd_ma"
|
||||
if "dts-hd ma" in track["profile"]:
|
||||
track["codec"] = "dtshd_ma"
|
||||
|
||||
elif "dts-hd hra" in track['profile']:
|
||||
track['codec'] = "dtshd_hra"
|
||||
elif "dts-hd hra" in track["profile"]:
|
||||
track["codec"] = "dtshd_hra"
|
||||
|
||||
return tracks
|
||||
|
||||
def get_runtime(self):
|
||||
|
||||
try:
|
||||
runtime = self.item['RunTimeTicks'] / 10000000.0
|
||||
runtime = self.item["RunTimeTicks"] / 10000000.0
|
||||
|
||||
except KeyError:
|
||||
runtime = self.item.get('CumulativeRunTimeTicks', 0) / 10000000.0
|
||||
runtime = self.item.get("CumulativeRunTimeTicks", 0) / 10000000.0
|
||||
|
||||
return runtime
|
||||
|
||||
|
|
@ -146,7 +150,7 @@ class API(object):
|
|||
resume = 0
|
||||
if resume_seconds:
|
||||
resume = round(float(resume_seconds), 6)
|
||||
jumpback = int(settings('resumeJumpBack'))
|
||||
jumpback = int(settings("resumeJumpBack"))
|
||||
if resume > jumpback:
|
||||
# To avoid negative bookmark
|
||||
resume = resume - jumpback
|
||||
|
|
@ -156,25 +160,25 @@ class API(object):
|
|||
def validate_studio(self, studio_name):
|
||||
# Convert studio for Kodi to properly detect them
|
||||
studios = {
|
||||
'abc (us)': "ABC",
|
||||
'fox (us)': "FOX",
|
||||
'mtv (us)': "MTV",
|
||||
'showcase (ca)': "Showcase",
|
||||
'wgn america': "WGN",
|
||||
'bravo (us)': "Bravo",
|
||||
'tnt (us)': "TNT",
|
||||
'comedy central': "Comedy Central (US)"
|
||||
"abc (us)": "ABC",
|
||||
"fox (us)": "FOX",
|
||||
"mtv (us)": "MTV",
|
||||
"showcase (ca)": "Showcase",
|
||||
"wgn america": "WGN",
|
||||
"bravo (us)": "Bravo",
|
||||
"tnt (us)": "TNT",
|
||||
"comedy central": "Comedy Central (US)",
|
||||
}
|
||||
return studios.get(studio_name.lower(), studio_name)
|
||||
|
||||
def get_overview(self, overview=None):
|
||||
|
||||
overview = overview or self.item.get('Overview')
|
||||
overview = overview or self.item.get("Overview")
|
||||
|
||||
if not overview:
|
||||
return
|
||||
|
||||
overview = overview.replace("\"", "\'")
|
||||
overview = overview.replace('"', "'")
|
||||
overview = overview.replace("\n", "[CR]")
|
||||
overview = overview.replace("\r", " ")
|
||||
overview = overview.replace("<br>", "[CR]")
|
||||
|
|
@ -183,7 +187,7 @@ class API(object):
|
|||
|
||||
def get_mpaa(self, rating=None):
|
||||
|
||||
mpaa = rating or self.item.get('OfficialRating', "")
|
||||
mpaa = rating or self.item.get("OfficialRating", "")
|
||||
|
||||
if mpaa in ("NR", "UR"):
|
||||
# Kodi seems to not like NR, but will accept Not Rated
|
||||
|
|
@ -197,112 +201,123 @@ class API(object):
|
|||
def get_file_path(self, path=None):
|
||||
|
||||
if path is None:
|
||||
path = self.item.get('Path')
|
||||
path = self.item.get("Path")
|
||||
|
||||
if not path:
|
||||
return ""
|
||||
|
||||
if path.startswith('\\\\'):
|
||||
path = path.replace('\\\\', "smb://", 1).replace('\\\\', "\\").replace('\\', "/")
|
||||
if path.startswith("\\\\"):
|
||||
path = (
|
||||
path.replace("\\\\", "smb://", 1)
|
||||
.replace("\\\\", "\\")
|
||||
.replace("\\", "/")
|
||||
)
|
||||
|
||||
if 'Container' in self.item:
|
||||
if "Container" in self.item:
|
||||
|
||||
if self.item['Container'] == 'dvd':
|
||||
if self.item["Container"] == "dvd":
|
||||
path = "%s/VIDEO_TS/VIDEO_TS.IFO" % path
|
||||
elif self.item['Container'] == 'bluray':
|
||||
elif self.item["Container"] == "bluray":
|
||||
path = "%s/BDMV/index.bdmv" % path
|
||||
|
||||
path = path.replace('\\\\', "\\")
|
||||
path = path.replace("\\\\", "\\")
|
||||
|
||||
if '\\' in path:
|
||||
path = path.replace('/', "\\")
|
||||
if "\\" in path:
|
||||
path = path.replace("/", "\\")
|
||||
|
||||
if '://' in path:
|
||||
protocol = path.split('://')[0]
|
||||
if "://" in path:
|
||||
protocol = path.split("://")[0]
|
||||
path = path.replace(protocol, protocol.lower())
|
||||
|
||||
return path
|
||||
|
||||
def get_user_artwork(self, user_id):
|
||||
|
||||
''' Get jellyfin user profile picture.
|
||||
'''
|
||||
"""Get jellyfin user profile picture."""
|
||||
return "%s/Users/%s/Images/Primary?Format=original" % (self.server, user_id)
|
||||
|
||||
def get_people_artwork(self, people):
|
||||
|
||||
''' Get people (actor, director, etc) artwork.
|
||||
'''
|
||||
"""Get people (actor, director, etc) artwork."""
|
||||
for person in people:
|
||||
|
||||
if 'PrimaryImageTag' in person:
|
||||
if "PrimaryImageTag" in person:
|
||||
|
||||
query = "&MaxWidth=400&MaxHeight=400&Index=0"
|
||||
person['imageurl'] = self.get_artwork(person['Id'], "Primary", person['PrimaryImageTag'], query)
|
||||
person["imageurl"] = self.get_artwork(
|
||||
person["Id"], "Primary", person["PrimaryImageTag"], query
|
||||
)
|
||||
else:
|
||||
person['imageurl'] = None
|
||||
person["imageurl"] = None
|
||||
|
||||
return people
|
||||
|
||||
def get_all_artwork(self, obj, parent_info=False):
|
||||
"""Get all artwork possible. If parent_info is True,
|
||||
it will fill missing artwork with parent artwork.
|
||||
|
||||
''' Get all artwork possible. If parent_info is True,
|
||||
it will fill missing artwork with parent artwork.
|
||||
|
||||
obj is from objects.Objects().map(item, 'Artwork')
|
||||
'''
|
||||
obj is from objects.Objects().map(item, 'Artwork')
|
||||
"""
|
||||
query = ""
|
||||
all_artwork = {
|
||||
'Primary': "",
|
||||
'BoxRear': "",
|
||||
'Art': "",
|
||||
'Banner': "",
|
||||
'Logo': "",
|
||||
'Thumb': "",
|
||||
'Disc': "",
|
||||
'Backdrop': []
|
||||
"Primary": "",
|
||||
"BoxRear": "",
|
||||
"Art": "",
|
||||
"Banner": "",
|
||||
"Logo": "",
|
||||
"Thumb": "",
|
||||
"Disc": "",
|
||||
"Backdrop": [],
|
||||
}
|
||||
|
||||
if settings('compressArt.bool'):
|
||||
if settings("compressArt.bool"):
|
||||
query = "&Quality=90"
|
||||
|
||||
if not settings('enableCoverArt.bool'):
|
||||
if not settings("enableCoverArt.bool"):
|
||||
query += "&EnableImageEnhancers=false"
|
||||
|
||||
art_maxheight = [360, 480, 600, 720, 1080, -1]
|
||||
maxheight = art_maxheight[int(settings('maxArtResolution') or 5)]
|
||||
maxheight = art_maxheight[int(settings("maxArtResolution") or 5)]
|
||||
if maxheight != -1:
|
||||
query += "&MaxHeight=%d" % maxheight
|
||||
|
||||
all_artwork['Backdrop'] = self.get_backdrops(obj['Id'], obj['BackdropTags'] or [], query)
|
||||
all_artwork["Backdrop"] = self.get_backdrops(
|
||||
obj["Id"], obj["BackdropTags"] or [], query
|
||||
)
|
||||
|
||||
for artwork in (obj['Tags'] or []):
|
||||
all_artwork[artwork] = self.get_artwork(obj['Id'], artwork, obj['Tags'][artwork], query)
|
||||
for artwork in obj["Tags"] or []:
|
||||
all_artwork[artwork] = self.get_artwork(
|
||||
obj["Id"], artwork, obj["Tags"][artwork], query
|
||||
)
|
||||
|
||||
if parent_info:
|
||||
|
||||
if not all_artwork['Backdrop'] and obj['ParentBackdropId']:
|
||||
all_artwork['Backdrop'] = self.get_backdrops(obj['ParentBackdropId'], obj['ParentBackdropTags'], query)
|
||||
if not all_artwork["Backdrop"] and obj["ParentBackdropId"]:
|
||||
all_artwork["Backdrop"] = self.get_backdrops(
|
||||
obj["ParentBackdropId"], obj["ParentBackdropTags"], query
|
||||
)
|
||||
|
||||
for art in ('Logo', 'Art', 'Thumb'):
|
||||
if not all_artwork[art] and obj['Parent%sId' % art]:
|
||||
all_artwork[art] = self.get_artwork(obj['Parent%sId' % art], art, obj['Parent%sTag' % art], query)
|
||||
for art in ("Logo", "Art", "Thumb"):
|
||||
if not all_artwork[art] and obj["Parent%sId" % art]:
|
||||
all_artwork[art] = self.get_artwork(
|
||||
obj["Parent%sId" % art], art, obj["Parent%sTag" % art], query
|
||||
)
|
||||
|
||||
if obj.get('SeriesTag'):
|
||||
all_artwork['Series.Primary'] = self.get_artwork(obj['SeriesId'], "Primary", obj['SeriesTag'], query)
|
||||
if obj.get("SeriesTag"):
|
||||
all_artwork["Series.Primary"] = self.get_artwork(
|
||||
obj["SeriesId"], "Primary", obj["SeriesTag"], query
|
||||
)
|
||||
|
||||
if not all_artwork['Primary']:
|
||||
all_artwork['Primary'] = all_artwork['Series.Primary']
|
||||
if not all_artwork["Primary"]:
|
||||
all_artwork["Primary"] = all_artwork["Series.Primary"]
|
||||
|
||||
elif not all_artwork['Primary'] and obj.get('AlbumId'):
|
||||
all_artwork['Primary'] = self.get_artwork(obj['AlbumId'], "Primary", obj['AlbumTag'], query)
|
||||
elif not all_artwork["Primary"] and obj.get("AlbumId"):
|
||||
all_artwork["Primary"] = self.get_artwork(
|
||||
obj["AlbumId"], "Primary", obj["AlbumTag"], query
|
||||
)
|
||||
|
||||
return all_artwork
|
||||
|
||||
def get_backdrops(self, item_id, tags, query=None):
|
||||
|
||||
''' Get backdrops based of "BackdropImageTags" in the jellyfin object.
|
||||
'''
|
||||
"""Get backdrops based of "BackdropImageTags" in the jellyfin object."""
|
||||
backdrops = []
|
||||
|
||||
if item_id is None:
|
||||
|
|
@ -310,15 +325,19 @@ class API(object):
|
|||
|
||||
for index, tag in enumerate(tags):
|
||||
|
||||
artwork = "%s/Items/%s/Images/Backdrop/%s?Format=original&Tag=%s%s" % (self.server, item_id, index, tag, (query or ""))
|
||||
artwork = "%s/Items/%s/Images/Backdrop/%s?Format=original&Tag=%s%s" % (
|
||||
self.server,
|
||||
item_id,
|
||||
index,
|
||||
tag,
|
||||
(query or ""),
|
||||
)
|
||||
backdrops.append(artwork)
|
||||
|
||||
return backdrops
|
||||
|
||||
def get_artwork(self, item_id, image, tag=None, query=None):
|
||||
|
||||
''' Get any type of artwork: Primary, Art, Banner, Logo, Thumb, Disc
|
||||
'''
|
||||
"""Get any type of artwork: Primary, Art, Banner, Logo, Thumb, Disc"""
|
||||
if item_id is None:
|
||||
return ""
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue