voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Use same transactional try-except-else as YCDL.

Browse source
This commit is contained in:
voussoir 2020-08-10 18:50:54 -07:00
parent 7609f20dd0
commit 03ae8325c9
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
utilities/database_upgrader/database_upgrader.py
View file

@ -284,9 +284,17 @@ def upgrade_all(data_directory):
print('Upgrading from %d to %d.' % (current_version, version_number))
upgrade_function = 'upgrade_%d_to_%d' % (current_version, version_number)
upgrade_function = eval(upgrade_function)
upgrade_function(photodb)
photodb.sql.cursor().execute('PRAGMA user_version = %d' % version_number)
photodb.commit()
try:
photodb.sql.execute('BEGIN')
upgrade_function(photodb)
except Exception as exc:
photodb.rollback()
raise
else:
photodb.sql.cursor().execute('PRAGMA user_version = %d' % version_number)
photodb.commit()
current_version = version_number
print('Upgrades finished.')