From 07a9bdbde31e22c38c675de2b5f1a1a5ffdcad46 Mon Sep 17 00:00:00 2001
From: JonnyWong16 <9099342+JonnyWong16@users.noreply.github.com>
Date: Tue, 4 Aug 2020 20:15:41 -0700
Subject: [PATCH] Add more stream export attributes

---
 plexpy/exporter.py | 16 ++++++++++++++++
 plexpy/helpers.py  |  5 +++++
 2 files changed, 21 insertions(+)

diff --git a/plexpy/exporter.py b/plexpy/exporter.py
index 2225c20e..5a2ea629 100644
--- a/plexpy/exporter.py
+++ b/plexpy/exporter.py
@@ -132,6 +132,7 @@ MOVIE_ATTRS = {
             'videoStreams': {
                 'codec': None,
                 'codecID': None,
+                'default': None,
                 'displayTitle': None,
                 'extendedDisplayTitle': None,
                 'id': None,
@@ -155,8 +156,11 @@ MOVIE_ATTRS = {
                 'frameRate': None,
                 'frameRateMode': None,
                 'hasScalingMatrix': None,
+                'hdr': lambda i: helpers.is_hdr(getattr(i, 'bitDepth', 0), getattr(i, 'colorSpace', '')),
                 'height': None,
                 'level': None,
+                'pixelAspectRatio': None,
+                'pixelFormat': None,
                 'profile': None,
                 'refFrames': None,
                 'requiredBandwidths': lambda e: [int(b) for b in e.split(',')] if e else None,
@@ -167,6 +171,7 @@ MOVIE_ATTRS = {
             'audioStreams': {
                 'codec': None,
                 'codecID': None,
+                'default': None,
                 'displayTitle': None,
                 'extendedDisplayTitle': None,
                 'id': None,
@@ -191,6 +196,7 @@ MOVIE_ATTRS = {
             'subtitleStreams': {
                 'codec': None,
                 'codecID': None,
+                'default': None,
                 'displayTitle': None,
                 'extendedDisplayTitle': None,
                 'id': None,
@@ -204,6 +210,7 @@ MOVIE_ATTRS = {
                 'type': None,
                 'forced': None,
                 'format': None,
+                'headerCompression': None,
                 'key': None
             }
         }
@@ -404,6 +411,7 @@ EPISODE_ATTRS = {
             'videoStreams': {
                 'codec': None,
                 'codecID': None,
+                'default': None,
                 'displayTitle': None,
                 'extendedDisplayTitle': None,
                 'id': None,
@@ -427,8 +435,11 @@ EPISODE_ATTRS = {
                 'frameRate': None,
                 'frameRateMode': None,
                 'hasScalingMatrix': None,
+                'hdr': lambda i: helpers.is_hdr(getattr(i, 'bitDepth', 0), getattr(i, 'colorSpace', '')),
                 'height': None,
                 'level': None,
+                'pixelAspectRatio': None,
+                'pixelFormat': None,
                 'profile': None,
                 'refFrames': None,
                 'requiredBandwidths': lambda e: [int(b) for b in e.split(',')] if e else None,
@@ -439,6 +450,7 @@ EPISODE_ATTRS = {
             'audioStreams': {
                 'codec': None,
                 'codecID': None,
+                'default': None,
                 'displayTitle': None,
                 'extendedDisplayTitle': None,
                 'id': None,
@@ -463,6 +475,7 @@ EPISODE_ATTRS = {
             'subtitleStreams': {
                 'codec': None,
                 'codecID': None,
+                'default': None,
                 'displayTitle': None,
                 'extendedDisplayTitle': None,
                 'id': None,
@@ -476,6 +489,7 @@ EPISODE_ATTRS = {
                 'type': None,
                 'forced': None,
                 'format': None,
+                'headerCompression': None,
                 'key': None
             }
         }
@@ -654,6 +668,7 @@ TRACK_ATTRS = {
             'audioStreams': {
                 'codec': None,
                 'codecID': None,
+                'default': None,
                 'displayTitle': None,
                 'extendedDisplayTitle': None,
                 'id': None,
@@ -681,6 +696,7 @@ TRACK_ATTRS = {
             'lyricStreams': {
                 'codec': None,
                 'codecID': None,
+                'default': None,
                 'displayTitle': None,
                 'extendedDisplayTitle': None,
                 'id': None,
diff --git a/plexpy/helpers.py b/plexpy/helpers.py
index b0e486a1..4ca630a4 100644
--- a/plexpy/helpers.py
+++ b/plexpy/helpers.py
@@ -1292,6 +1292,11 @@ def dict_merge(a, b, path=None):
     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):
     endpoints = {
         'pms_image_proxy': pms_image_proxy,