github/plexpy
1
0
Fork 0
mirror of https://github.com/Tautulli/Tautulli.git synced 2025-07-10 23:42:37 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Skip importing table if it doesn't exist

Browse source
This commit is contained in:
JonnyWong16 2020-05-19 22:40:53 -07:00
parent b01fefc235
commit b8f8d45807
No known key found for this signature in database
GPG key ID: B1F1F9807184697A
1 changed files with 9 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

14
plexpy/database.py
View file

@ -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