From 0522253194a24d5f12b3d45345f850bf51fba1ce Mon Sep 17 00:00:00 2001
From: herby2212 <12448284+herby2212@users.noreply.github.com>
Date: Wed, 1 Mar 2023 02:33:13 +0100
Subject: [PATCH] Add watch time & user stats for playlists (#2012)

---
 data/interfaces/default/info.html | 4 ++--
 plexpy/datafactory.py             | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/data/interfaces/default/info.html b/data/interfaces/default/info.html
index 471c8e7e..31c83c05 100644
--- a/data/interfaces/default/info.html
+++ b/data/interfaces/default/info.html
@@ -554,7 +554,7 @@ DOCUMENTATION :: END
                     </div>
                 </div>
                 % endif
-                % if data['media_type'] in ('movie', 'show', 'season', 'episode', 'artist', 'album', 'track', 'collection'):
+                % if data['media_type'] in ('movie', 'show', 'season', 'episode', 'artist', 'album', 'track', 'collection', 'playlist'):
                 <div class="col-md-12">
                     <div class="table-card-header">
                         <div class="header-bar">
@@ -937,7 +937,7 @@ DOCUMENTATION :: END
     });
 </script>
 % endif
-% if data['media_type'] in ('movie', 'show', 'season', 'episode', 'artist', 'album', 'track', 'collection'):
+% if data['media_type'] in ('movie', 'show', 'season', 'episode', 'artist', 'album', 'track', 'collection', 'playlist'):
 <script>
     // Populate watch time stats
     $.ajax({
diff --git a/plexpy/datafactory.py b/plexpy/datafactory.py
index 06f58c36..8eb1d39c 100644
--- a/plexpy/datafactory.py
+++ b/plexpy/datafactory.py
@@ -1210,7 +1210,7 @@ class DataFactory(object):
 
         group_by = 'session_history.reference_id' if grouping else 'session_history.id'
 
-        if media_type == 'collection':
+        if media_type in ('collection', 'playlist'):
             pms_connect = pmsconnect.PmsConnect()
             result = pms_connect.get_item_children(rating_key=rating_key, media_type=media_type)
             rating_keys = [child['rating_key'] for child in result['children_list']]
@@ -1294,7 +1294,7 @@ class DataFactory(object):
         
         group_by = 'session_history.reference_id' if grouping else 'session_history.id'
 
-        if media_type == 'collection':
+        if media_type in ('collection', 'playlist'):
             pms_connect = pmsconnect.PmsConnect()
             result = pms_connect.get_item_children(rating_key=rating_key, media_type=media_type)
             rating_keys = [child['rating_key'] for child in result['children_list']]