From 42c6340c06ceedf480a888283378a7c78d9dd8b0 Mon Sep 17 00:00:00 2001
From: JonnyWong16 <9099342+JonnyWong16@users.noreply.github.com>
Date: Sat, 23 May 2020 16:02:45 -0700
Subject: [PATCH] Delete uploaded file if invalid database

---
 plexpy/webserve.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/plexpy/webserve.py b/plexpy/webserve.py
index 4de3a273..dfe66f4f 100644
--- a/plexpy/webserve.py
+++ b/plexpy/webserve.py
@@ -3797,6 +3797,8 @@ class WebInterface(object):
                 return {'result': 'success',
                         'message': 'Import has started. Check the logs to monitor any problems.'}
             else:
+                if database_file:
+                    helpers.delete_file(database_path)
                 return {'result': 'error', 'message': db_check_msg}
 
         elif app.lower() == 'plexwatch':
@@ -3810,6 +3812,8 @@ class WebInterface(object):
                 return {'result': 'success',
                         'message': 'Import has started. Check the logs to monitor any problems.'}
             else:
+                if database_file:
+                    helpers.delete_file(database_path)
                 return {'result': 'error', 'message': db_check_msg}
 
         elif app.lower() == 'plexivity':
@@ -3823,6 +3827,8 @@ class WebInterface(object):
                 return {'result': 'success',
                         'message': 'Import has started. Check the logs to monitor any problems.'}
             else:
+                if database_file:
+                    helpers.delete_file(database_path)
                 return {'result': 'error', 'message': db_check_msg}
 
         else: