From 52d38883dcd521bcff7a4b0ee79bbf553da35212 Mon Sep 17 00:00:00 2001
From: JonnyWong16 <9099342+JonnyWong16@users.noreply.github.com>
Date: Thu, 30 Apr 2020 20:46:58 -0700
Subject: [PATCH] Add Tautulli database import to the settings page

---
 data/interfaces/default/app_import.html | 60 +++++++++++++++++++------
 data/interfaces/default/settings.html   |  3 +-
 plexpy/webserve.py                      | 35 ++++++++++++---
 3 files changed, 77 insertions(+), 21 deletions(-)

diff --git a/data/interfaces/default/app_import.html b/data/interfaces/default/app_import.html
index 0f0c8351..fb730cb3 100644
--- a/data/interfaces/default/app_import.html
+++ b/data/interfaces/default/app_import.html
@@ -5,6 +5,7 @@
             <h4 class="modal-title">Import ${app} Database</h4>
         </div>
         <div class="modal-body" id="modal-text">
+            % if app in ('PlexWatch', 'Plexivity'):
             <p class="help-block">
                 <%
                     v = ''
@@ -15,26 +16,54 @@
                 %>
                 <strong>Please ensure your ${app} database is at version ${v} or higher.</strong>
             </p>
+            % endif
             <div class="form-group">
-                <label for="db_location">Database Location</label>
+                <label for="import_database_path">Database Location</label>
                 <div class="row">
-                    <div class="col-xs-8">
-                        <input type="text" class="form-control" id="db_location" name="db_location" value="" required>
+                    <div class="col-xs-12">
+                        <input type="text" class="form-control" id="import_database_path" name="import_database_path" value="" required>
                     </div>
                 </div>
-                <p class="help-block">Enter the path and file name for the ${app} database you wish to import.</p>
+                <p class="help-block">Enter the full path to the ${app} database you wish to import.</p>
             </div>
+            % if app == 'Tautulli':
             <div class="form-group">
-                <label for="table_name">Table Name</label>
+                <label for="table_name">Import Method</label>
                 <div class="row">
                     <div class="col-xs-4">
-                        <select id="table_name" class="form-control" name="table_name">
-                            <option value="processed">processed</option>
-                            <option value="grouped">grouped</option>
+                        <select class="form-control" id="import_method" name="import_method">
+                            <option value="merge">Merge</option>
+                            <option value="append">Append</option>
+                            <option value="overwrite">Oerwrite</option>
                         </select>
                     </div>
                 </div>
-                <p class="help-block">The table name from which you wish to import. Only import one of these, importing both will result in duplicated data.</p>
+                <p class="help-block">Select how you would like to import the Tautulli history.</p>
+                <ul class="help-block" style="padding-inline-start: 15px;">
+                    <li><strong>Merge</strong> will only add missing history from the imported database into the current database.</li>
+                    <li><strong>Append</strong> will add all history from the imported database into the current database.
+                        <br>Note: History will be duplicated if it is present in both databases.</li>
+                    <li><strong>Overwrite</strong> will replace all history in the current database with the imported database.</li>
+                </ul>
+            </div>
+            <div class="checkbox">
+                <label>
+                    <input type="checkbox" name="import_backup_db" id="import_backup_db" value="1" checked> Backup Database
+                </label>
+                <p class="help-block">Automatically create a backup of the current database before importing.</p>
+            </div>
+            % else:
+            <div class="form-group">
+                <label for="import_table_name">Table Name</label>
+                <div class="row">
+                    <div class="col-xs-4">
+                        <select class="form-control" id="import_table_name" name="import_table_name">
+                            <option value="processed">Processed</option>
+                            <option value="grouped">Grouped</option>
+                        </select>
+                    </div>
+                </div>
+                <p class="help-block">Select the table name from which you wish to import. Only import one of these, importing both will result in duplicated data.</p>
             </div>
             <div class="form-group">
                 <label for="import_ignore_interval">Ignore Interval</label>
@@ -45,6 +74,7 @@
                 </div>
                 <p class="help-block">Enter the minimum duration (in seconds) an item must have been active for. Set to 0 to import all.</p>
             </div>
+            % endif
         </div>
         <div class="modal-footer">
             <div>
@@ -57,22 +87,26 @@
 <script>
     // Send database path to import script
     $("#import_db").click(function() {
-        var database_path = $("#db_location").val();
-        var table_name = $("#table_name").val();
+        var database_path = $("#import_database_path").val();
+        var import_method = $("#import_method").val();
+        var import_backup_db = $("#import_backup_db").is(':checked');
+        var import_table_name = $("#import_table_name").val();
         var import_ignore_interval = $("#import_ignore_interval").val();
         $.ajax({
             url: 'import_database',
             data: {
                 app: "${app}",
                 database_path: database_path,
-                table_name: table_name,
+                method: import_method,
+                backup: import_backup_db,
+                table_name: import_table_name,
                 import_ignore_interval: import_ignore_interval
             },
             cache: false,
             async: true,
             success: function(data) {
                 $("#status-message").html(data);
-                $("#db_location").val('')
+                $("#import_database_path").val('')
             }
         });
     });
diff --git a/data/interfaces/default/settings.html b/data/interfaces/default/settings.html
index 8426224a..89fa5721 100644
--- a/data/interfaces/default/settings.html
+++ b/data/interfaces/default/settings.html
@@ -1321,8 +1321,9 @@
                             <h3>Database Import</h3>
                         </div>
 
-                        <p class="help-block">Click a button below to import an existing database from another app.</p>
+                        <p class="help-block">Click a button below to import an existing database from the selected app.</p>
                         <div class="btn-group">
+                            <button class="btn btn-form toggle-app-import-modal" type="button" data-target="#app-import-modal" data-toggle="modal" data-app="tautulli">Tautulli</button>
                             <button class="btn btn-form toggle-app-import-modal" type="button" data-target="#app-import-modal" data-toggle="modal" data-app="plexwatch">PlexWatch</button>
                             <button class="btn btn-form toggle-app-import-modal" type="button" data-target="#app-import-modal" data-toggle="modal" data-app="plexivity">Plexivity</button>
                         </div>
diff --git a/plexpy/webserve.py b/plexpy/webserve.py
index 44c76ff2..49aa03a4 100644
--- a/plexpy/webserve.py
+++ b/plexpy/webserve.py
@@ -3740,17 +3740,23 @@ class WebInterface(object):
     @cherrypy.expose
     @requireAuth(member_of("admin"))
     @addtoapi()
-    def import_database(self, app=None, database_path=None, table_name=None, import_ignore_interval=0, **kwargs):
-        """ Import a PlexWatch or Plexivity database into Tautulli.
+    def import_database(self, app=None, database_path=None, method=None, backup=True,
+                        table_name=None, import_ignore_interval=0, **kwargs):
+        """ Import a Tautulli, PlexWatch, or Plexivity database into Tautulli.
 
             ```
             Required parameters:
-                app (str):                      "plexwatch" or "plexivity"
+                app (str):                      "tautulli" or "plexwatch" or "plexivity"
                 database_path (str):            The full path to the plexwatch database file
-                table_name (str):               "processed" or "grouped"
+                method (str):                   For Tautulli only, "merge" or "overwrite"
+                table_name (str):               For PlexWatch or Plexivity only, "processed" or "grouped"
+
 
             Optional parameters:
-                import_ignore_interval (int):   The minimum number of seconds for a stream to import
+                backup (bool):                  For Tautulli only, true or false whether to backup
+                                                the current database before importing
+                import_ignore_interval (int):   For PlexWatch or Plexivity only, the minimum number
+                                                of seconds for a stream to import
 
             Returns:
                 None
@@ -3759,7 +3765,18 @@ class WebInterface(object):
         if not app:
             return 'No app specified for import'
 
-        if app.lower() == 'plexwatch':
+        if app.lower() == 'tautulli':
+            db_check_msg = database.validate_database(database=database_path)
+            if db_check_msg == 'success':
+                threading.Thread(target=database.import_tautulli_db,
+                                 kwargs={'database': database_path,
+                                         'method': method,
+                                         'backup': helpers.bool_true(backup)}).start()
+                return 'Import has started. Check the Tautulli logs to monitor any problems.'
+            else:
+                return db_check_msg
+
+        elif app.lower() == 'plexwatch':
             db_check_msg = plexwatch_import.validate_database(database=database_path,
                                                               table_name=table_name)
             if db_check_msg == 'success':
@@ -3770,6 +3787,7 @@ class WebInterface(object):
                 return 'Import has started. Check the Tautulli logs to monitor any problems.'
             else:
                 return db_check_msg
+            
         elif app.lower() == 'plexivity':
             db_check_msg = plexivity_import.validate_database(database=database_path,
                                                               table_name=table_name)
@@ -3781,13 +3799,16 @@ class WebInterface(object):
                 return 'Import has started. Check the Tautulli logs to monitor any problems.'
             else:
                 return db_check_msg
+
         else:
             return 'App not recognized for import'
 
     @cherrypy.expose
     @requireAuth(member_of("admin"))
     def import_database_tool(self, app=None, **kwargs):
-        if app == 'plexwatch':
+        if app == 'tautulli':
+            return serve_template(templatename="app_import.html", title="Import Tautulli Database", app="Tautulli")
+        elif app == 'plexwatch':
             return serve_template(templatename="app_import.html", title="Import PlexWatch Database", app="PlexWatch")
         elif app == 'plexivity':
             return serve_template(templatename="app_import.html", title="Import Plexivity Database", app="Plexivity")