github/plexpy
1
0
Fork 0
mirror of https://github.com/Tautulli/Tautulli.git synced 2025-07-12 08:16:06 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Check for existing column names when importing

Browse source
This commit is contained in:
JonnyWong16 2020-05-19 21:57:04 -07:00
parent 09f6eb8e19
commit b01fefc235
No known key found for this signature in database
GPG key ID: B1F1F9807184697A
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
plexpy/database.py
View file

@ -114,11 +114,14 @@ 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
columns = db.select('PRAGMA import_db.table_info({table})'.format(table=table_name))
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))
if method == 'merge':
import_columns = [c['name'] for c in columns if not c['pk']]
import_columns = [c['name'] for c in import_columns if c['name'] in current_columns and not c['pk']]
else:
import_columns = [c['name'] for c in columns]
import_columns = [c['name'] for c in import_columns if c['name'] in current_columns]
insert_columns = ', '.join(import_columns)
table_columns[table_name] = insert_columns