From 1dd1c6f67ffe3b6ac706f644ab8a464a3afa89a7 Mon Sep 17 00:00:00 2001
From: JonnyWong16 <JonnyWong16@users.noreply.github.com>
Date: Sat, 22 Feb 2020 14:02:03 -0800
Subject: [PATCH] Add fake Live TV library

---
 data/interfaces/default/library.html |  8 ++++++++
 plexpy/activity_processor.py         |  4 ++++
 plexpy/common.py                     |  3 +++
 plexpy/config.py                     |  1 +
 plexpy/libraries.py                  | 23 +++++++++++++++++++++++
 plexpy/pmsconnect.py                 |  3 +++
 6 files changed, 42 insertions(+)

diff --git a/data/interfaces/default/library.html b/data/interfaces/default/library.html
index 431e8931..03883a3d 100644
--- a/data/interfaces/default/library.html
+++ b/data/interfaces/default/library.html
@@ -36,6 +36,7 @@ DOCUMENTATION :: END
 <%def name="body()">
 % if data:
 <%
+    from plexpy.common import LIVE_TV_SECTION_ID
     from plexpy.helpers import page
 %>
 <div class="container-fluid">
@@ -78,8 +79,10 @@ DOCUMENTATION :: END
                                     <li class="active"><a href="#tabs-profile" role="tab" data-toggle="tab">Profile</a></li>
                                     <li><a id="history-tab-btn" href="#tabs-history" role="tab" data-toggle="tab">History</a></li>
                                     % if _session['user_group'] == 'admin':
+                                    % if data['section_id'] != LIVE_TV_SECTION_ID:
                                     <li><a id="media-info-tab-btn" href="#tabs-mediainfo" role="tab" data-toggle="tab">Media Info</a></li>
                                     % endif
+                                    % endif
                                 </ul>
                             </div>
                         </div>
@@ -146,6 +149,7 @@ DOCUMENTATION :: END
                                 </div>
                             </div>
                         </div>
+                        % if data['section_id'] != LIVE_TV_SECTION_ID:
                         <div class="container-fluid">
                             <div class="row">
                                 <div class="col-md-12">
@@ -171,6 +175,7 @@ DOCUMENTATION :: END
                                 </div>
                             </div>
                         </div>
+                        % endif
                     </div>
                     <div role="tabpanel" class="tab-pane" id="tabs-history">
                         <div class="container-fluid">
@@ -351,6 +356,7 @@ DOCUMENTATION :: END
 <script src="${http_root}js/dataTables.bootstrap.min.js"></script>
 <script src="${http_root}js/dataTables.bootstrap.pagination.js"></script>
 % if data:
+<% from plexpy.common import LIVE_TV_SECTION_ID %>
 <script>
     % if str(data['section_id']).isdigit():
     var section_id = ${data['section_id']};
@@ -529,7 +535,9 @@ DOCUMENTATION :: END
     }
 
     recentlyWatched();
+    % if data['section_id'] != LIVE_TV_SECTION_ID:
     recentlyAdded();
+    % endif
 
     function highlightWatchedScrollerButton() {
         var scroller = $("#recently-watched-row-scroller");
diff --git a/plexpy/activity_processor.py b/plexpy/activity_processor.py
index 3c96d4aa..6d1c2c35 100644
--- a/plexpy/activity_processor.py
+++ b/plexpy/activity_processor.py
@@ -149,6 +149,10 @@ class ActivityProcessor(object):
                 timestamp = {'started': started}
                 self.db.upsert('sessions', timestamp, keys)
 
+                # Add Live TV library if it hasn't been added
+                if values['live']:
+                    libraries.add_live_tv_library()
+
                 return True
 
     def write_session_history(self, session=None, import_metadata=None, is_import=False, import_ignore_interval=0):
diff --git a/plexpy/common.py b/plexpy/common.py
index 6f22edad..f00976b9 100644
--- a/plexpy/common.py
+++ b/plexpy/common.py
@@ -42,6 +42,9 @@ ONLINE_POSTER_THUMB = "https://tautulli.com/images/poster.png"
 ONLINE_COVER_THUMB = "https://tautulli.com/images/cover.png"
 ONLINE_ART = "https://tautulli.com/images/art.png"
 
+LIVE_TV_SECTION_ID = 999999  # Fake section_id for Live TV library
+LIVE_TV_SECTION_NAME = "Live TV"  # Fake section_name for Live TV library
+
 DEFAULT_IMAGES = {
     'poster': DEFAULT_POSTER_THUMB,
     'cover': DEFAULT_COVER_THUMB,
diff --git a/plexpy/config.py b/plexpy/config.py
index 82159e0d..644ebf18 100644
--- a/plexpy/config.py
+++ b/plexpy/config.py
@@ -68,6 +68,7 @@ _CONFIG_DEFINITIONS = {
     'PMS_UPDATE_CHECK_INTERVAL': (int, 'Advanced', 24),
     'PMS_WEB_URL': (str, 'PMS', 'https://app.plex.tv/desktop'),
     'TIME_FORMAT': (str, 'General', 'HH:mm'),
+    'ADD_LIVE_TV_LIBRARY': (int, 'Advanced', 1),
     'ANON_REDIRECT': (str, 'General', 'http://www.nullrefer.com/?'),
     'API_ENABLED': (int, 'General', 1),
     'API_KEY': (str, 'General', ''),
diff --git a/plexpy/libraries.py b/plexpy/libraries.py
index f1e7dfd1..24869451 100644
--- a/plexpy/libraries.py
+++ b/plexpy/libraries.py
@@ -88,6 +88,29 @@ def refresh_libraries():
         return False
 
 
+def add_live_tv_library():
+    if not plexpy.CONFIG.ADD_LIVE_TV_LIBRARY:
+        return
+
+    logger.info(u"Tautulli Libraries :: Adding Live TV library to the database.")
+
+    monitor_db = database.MonitorDatabase()
+
+    section_keys = {'server_id': plexpy.CONFIG.PMS_IDENTIFIER,
+                    'section_id': common.LIVE_TV_SECTION_ID}
+    section_values = {'server_id': plexpy.CONFIG.PMS_IDENTIFIER,
+                      'section_id': common.LIVE_TV_SECTION_ID,
+                      'section_name': common.LIVE_TV_SECTION_NAME,
+                      'section_type': 'live'
+                      }
+
+    result = monitor_db.upsert('library_sections', key_dict=section_keys, value_dict=section_values)
+
+    if result == 'insert':
+        plexpy.CONFIG.__setattr__('ADD_LIVE_TV_LIBRARY', 0)
+        plexpy.CONFIG.write()
+
+
 def update_section_ids():
     plexpy.CONFIG.UPDATE_SECTION_IDS = -1
 
diff --git a/plexpy/pmsconnect.py b/plexpy/pmsconnect.py
index e081e8fd..f715d838 100644
--- a/plexpy/pmsconnect.py
+++ b/plexpy/pmsconnect.py
@@ -1267,6 +1267,9 @@ class PmsConnect(object):
 
         # Get additional metadata from metadata.provider.plex.tv
         if not plex_guid and metadata['live']:
+            metadata['section_id'] = common.LIVE_TV_SECTION_ID
+            metadata['library_name'] = common.LIVE_TV_SECTION_NAME
+
             plextv_metadata = self.get_metadata_details(plex_guid=metadata['guid'])
             if plextv_metadata:
                 keys_to_update = ['summary', 'rating', 'thumb', 'grandparent_thumb', 'duration',