Add more stream export attributes

This commit is contained in:
JonnyWong16 2020-08-04 20:15:41 -07:00
parent 588b1b1bc3
commit 07a9bdbde3
No known key found for this signature in database
GPG key ID: B1F1F9807184697A
2 changed files with 21 additions and 0 deletions

View file

@ -132,6 +132,7 @@ MOVIE_ATTRS = {
'videoStreams': { 'videoStreams': {
'codec': None, 'codec': None,
'codecID': None, 'codecID': None,
'default': None,
'displayTitle': None, 'displayTitle': None,
'extendedDisplayTitle': None, 'extendedDisplayTitle': None,
'id': None, 'id': None,
@ -155,8 +156,11 @@ MOVIE_ATTRS = {
'frameRate': None, 'frameRate': None,
'frameRateMode': None, 'frameRateMode': None,
'hasScalingMatrix': None, 'hasScalingMatrix': None,
'hdr': lambda i: helpers.is_hdr(getattr(i, 'bitDepth', 0), getattr(i, 'colorSpace', '')),
'height': None, 'height': None,
'level': None, 'level': None,
'pixelAspectRatio': None,
'pixelFormat': None,
'profile': None, 'profile': None,
'refFrames': None, 'refFrames': None,
'requiredBandwidths': lambda e: [int(b) for b in e.split(',')] if e else None, 'requiredBandwidths': lambda e: [int(b) for b in e.split(',')] if e else None,
@ -167,6 +171,7 @@ MOVIE_ATTRS = {
'audioStreams': { 'audioStreams': {
'codec': None, 'codec': None,
'codecID': None, 'codecID': None,
'default': None,
'displayTitle': None, 'displayTitle': None,
'extendedDisplayTitle': None, 'extendedDisplayTitle': None,
'id': None, 'id': None,
@ -191,6 +196,7 @@ MOVIE_ATTRS = {
'subtitleStreams': { 'subtitleStreams': {
'codec': None, 'codec': None,
'codecID': None, 'codecID': None,
'default': None,
'displayTitle': None, 'displayTitle': None,
'extendedDisplayTitle': None, 'extendedDisplayTitle': None,
'id': None, 'id': None,
@ -204,6 +210,7 @@ MOVIE_ATTRS = {
'type': None, 'type': None,
'forced': None, 'forced': None,
'format': None, 'format': None,
'headerCompression': None,
'key': None 'key': None
} }
} }
@ -404,6 +411,7 @@ EPISODE_ATTRS = {
'videoStreams': { 'videoStreams': {
'codec': None, 'codec': None,
'codecID': None, 'codecID': None,
'default': None,
'displayTitle': None, 'displayTitle': None,
'extendedDisplayTitle': None, 'extendedDisplayTitle': None,
'id': None, 'id': None,
@ -427,8 +435,11 @@ EPISODE_ATTRS = {
'frameRate': None, 'frameRate': None,
'frameRateMode': None, 'frameRateMode': None,
'hasScalingMatrix': None, 'hasScalingMatrix': None,
'hdr': lambda i: helpers.is_hdr(getattr(i, 'bitDepth', 0), getattr(i, 'colorSpace', '')),
'height': None, 'height': None,
'level': None, 'level': None,
'pixelAspectRatio': None,
'pixelFormat': None,
'profile': None, 'profile': None,
'refFrames': None, 'refFrames': None,
'requiredBandwidths': lambda e: [int(b) for b in e.split(',')] if e else None, 'requiredBandwidths': lambda e: [int(b) for b in e.split(',')] if e else None,
@ -439,6 +450,7 @@ EPISODE_ATTRS = {
'audioStreams': { 'audioStreams': {
'codec': None, 'codec': None,
'codecID': None, 'codecID': None,
'default': None,
'displayTitle': None, 'displayTitle': None,
'extendedDisplayTitle': None, 'extendedDisplayTitle': None,
'id': None, 'id': None,
@ -463,6 +475,7 @@ EPISODE_ATTRS = {
'subtitleStreams': { 'subtitleStreams': {
'codec': None, 'codec': None,
'codecID': None, 'codecID': None,
'default': None,
'displayTitle': None, 'displayTitle': None,
'extendedDisplayTitle': None, 'extendedDisplayTitle': None,
'id': None, 'id': None,
@ -476,6 +489,7 @@ EPISODE_ATTRS = {
'type': None, 'type': None,
'forced': None, 'forced': None,
'format': None, 'format': None,
'headerCompression': None,
'key': None 'key': None
} }
} }
@ -654,6 +668,7 @@ TRACK_ATTRS = {
'audioStreams': { 'audioStreams': {
'codec': None, 'codec': None,
'codecID': None, 'codecID': None,
'default': None,
'displayTitle': None, 'displayTitle': None,
'extendedDisplayTitle': None, 'extendedDisplayTitle': None,
'id': None, 'id': None,
@ -681,6 +696,7 @@ TRACK_ATTRS = {
'lyricStreams': { 'lyricStreams': {
'codec': None, 'codec': None,
'codecID': None, 'codecID': None,
'default': None,
'displayTitle': None, 'displayTitle': None,
'extendedDisplayTitle': None, 'extendedDisplayTitle': None,
'id': None, 'id': None,

View file

@ -1292,6 +1292,11 @@ def dict_merge(a, b, path=None):
return a return a
def is_hdr(bit_depth, color_space):
bit_depth = cast_to_int(bit_depth)
return bit_depth > 8 and color_space == 'bt2020nc'
def page(endpoint, *args, **kwargs): def page(endpoint, *args, **kwargs):
endpoints = { endpoints = {
'pms_image_proxy': pms_image_proxy, 'pms_image_proxy': pms_image_proxy,