From b8f8d4580774b0de32772921754a3cc69277d870 Mon Sep 17 00:00:00 2001 From: JonnyWong16 <9099342+JonnyWong16@users.noreply.github.com> Date: Tue, 19 May 2020 22:40:53 -0700 Subject: [PATCH] Skip importing table if it doesn't exist --- plexpy/database.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/plexpy/database.py b/plexpy/database.py index 0a85f19c..0bfafa61 100644 --- a/plexpy/database.py +++ b/plexpy/database.py @@ -106,6 +106,11 @@ def import_tautulli_db(database=None, method=None, backup=False): # Skip temporary sessions table continue + current_table = db.select('PRAGMA main.table_info({table})'.format(table=table_name)) + if not current_table: + # Skip table does not exits + continue + logger.info("Tautulli Database :: Importing database table '%s'.", table_name) if method == 'overwrite': @@ -114,14 +119,13 @@ def import_tautulli_db(database=None, method=None, backup=False): db.action('DELETE FROM sqlite_sequence WHERE name = ?', [table_name]) # Get the list of columns to import - current_columns = db.select('PRAGMA table_info({table})'.format(table=table_name)) - current_columns = [c['name'] for c in current_columns] - import_columns = db.select('PRAGMA import_db.table_info({table})'.format(table=table_name)) + current_columns = [c['name'] for c in current_table] + import_table = db.select('PRAGMA import_db.table_info({table})'.format(table=table_name)) if method == 'merge': - import_columns = [c['name'] for c in import_columns if c['name'] in current_columns and not c['pk']] + import_columns = [c['name'] for c in import_table if c['name'] in current_columns and not c['pk']] else: - import_columns = [c['name'] for c in import_columns if c['name'] in current_columns] + import_columns = [c['name'] for c in import_table if c['name'] in current_columns] insert_columns = ', '.join(import_columns) table_columns[table_name] = insert_columns