Remove automatic BEGIN because it breaks pragma foreign_keys.

It turns out you can't adjust foreign_keys while a transaction is active,
so the upgraders that need to disable and re-enable them were not
working right and encountering foreign key violations.
master
voussoir 2020-09-15 14:54:21 -07:00
parent 50ecc4c1c1
commit 2d5fb19ce7
1 changed files with 19 additions and 9 deletions

View File

@ -10,6 +10,7 @@ def upgrade_1_to_2(photodb):
In this version, a column `tagged_at` was added to the Photos table, to keep In this version, a column `tagged_at` was added to the Photos table, to keep
track of the last time the photo's tags were edited (added or removed). track of the last time the photo's tags were edited (added or removed).
''' '''
photodb.sql_execute('BEGIN')
cur = photodb.sql.cursor() cur = photodb.sql.cursor()
cur.execute('ALTER TABLE photos ADD COLUMN tagged_at INT') cur.execute('ALTER TABLE photos ADD COLUMN tagged_at INT')
@ -19,6 +20,7 @@ def upgrade_2_to_3(photodb):
with id, username, password hash, and a timestamp. with id, username, password hash, and a timestamp.
Plus some indices. Plus some indices.
''' '''
photodb.sql_execute('BEGIN')
cur = photodb.sql.cursor() cur = photodb.sql.cursor()
cur.execute(''' cur.execute('''
CREATE TABLE IF NOT EXISTS users( CREATE TABLE IF NOT EXISTS users(
@ -35,6 +37,7 @@ def upgrade_3_to_4(photodb):
''' '''
Add an `author_id` column to Photos. Add an `author_id` column to Photos.
''' '''
photodb.sql_execute('BEGIN')
cur = photodb.sql.cursor() cur = photodb.sql.cursor()
cur.execute('ALTER TABLE photos ADD COLUMN author_id TEXT') cur.execute('ALTER TABLE photos ADD COLUMN author_id TEXT')
cur.execute('CREATE INDEX IF NOT EXISTS index_photo_author ON photos(author_id)') cur.execute('CREATE INDEX IF NOT EXISTS index_photo_author ON photos(author_id)')
@ -43,6 +46,7 @@ def upgrade_4_to_5(photodb):
''' '''
Add table `bookmarks` and its indices. Add table `bookmarks` and its indices.
''' '''
photodb.sql_execute('BEGIN')
cur = photodb.sql.cursor() cur = photodb.sql.cursor()
cur.execute(''' cur.execute('''
CREATE TABLE IF NOT EXISTS bookmarks( CREATE TABLE IF NOT EXISTS bookmarks(
@ -67,6 +71,7 @@ def upgrade_5_to_6(photodb):
into `album_group_rel` into `album_group_rel`
- Gives albums their own last_id value, starting with the current tag value. - Gives albums their own last_id value, starting with the current tag value.
''' '''
photodb.sql_execute('BEGIN')
# 1. Start the id_numbers.albums value at the tags value so that the number # 1. Start the id_numbers.albums value at the tags value so that the number
# can continue to increment safely and separately, instead of starting at # can continue to increment safely and separately, instead of starting at
# zero and bumping into existing albums. # zero and bumping into existing albums.
@ -112,6 +117,7 @@ def upgrade_6_to_7(photodb):
separate table `album_associated_directories`, so that we can have albums separate table `album_associated_directories`, so that we can have albums
which load from multiple directories. which load from multiple directories.
''' '''
photodb.sql_execute('BEGIN')
cur = photodb.sql.cursor() cur = photodb.sql.cursor()
cur.execute('SELECT name FROM sqlite_master WHERE type == "index"') cur.execute('SELECT name FROM sqlite_master WHERE type == "index"')
indices = [x[0] for x in cur.fetchall()] indices = [x[0] for x in cur.fetchall()]
@ -143,6 +149,7 @@ def upgrade_7_to_8(photodb):
''' '''
Give the Tags table a description field. Give the Tags table a description field.
''' '''
photodb.sql_execute('BEGIN')
cur = photodb.sql.cursor() cur = photodb.sql.cursor()
cur.execute('ALTER TABLE tags ADD COLUMN description TEXT') cur.execute('ALTER TABLE tags ADD COLUMN description TEXT')
@ -150,6 +157,7 @@ def upgrade_8_to_9(photodb):
''' '''
Give the Photos table a searchhidden field. Give the Photos table a searchhidden field.
''' '''
photodb.sql_execute('BEGIN')
cur = photodb.sql.cursor() cur = photodb.sql.cursor()
cur.execute('ALTER TABLE photos ADD COLUMN searchhidden INT') cur.execute('ALTER TABLE photos ADD COLUMN searchhidden INT')
cur.execute('UPDATE photos SET searchhidden = 0') cur.execute('UPDATE photos SET searchhidden = 0')
@ -162,6 +170,7 @@ def upgrade_9_to_10(photodb):
Previously, the stored path was unnecessarily high and contained the PDB's Previously, the stored path was unnecessarily high and contained the PDB's
data_directory, reducing portability. data_directory, reducing portability.
''' '''
photodb.sql_execute('BEGIN')
cur = photodb.sql.cursor() cur = photodb.sql.cursor()
photos = list(photodb.search(has_thumbnail=True, is_searchhidden=None)) photos = list(photodb.search(has_thumbnail=True, is_searchhidden=None))
@ -179,6 +188,9 @@ def upgrade_10_to_11(photodb):
Added Primary keys, Foreign keys, and NOT NULL constraints. Added Primary keys, Foreign keys, and NOT NULL constraints.
Added author_id column to Album and Tag tables. Added author_id column to Album and Tag tables.
''' '''
photodb.sql_execute('PRAGMA foreign_keys = OFF')
photodb.sql_execute('BEGIN')
tables_to_copy = { tables_to_copy = {
'users': '*', 'users': '*',
'albums': '*, NULL', 'albums': '*, NULL',
@ -193,9 +205,6 @@ def upgrade_10_to_11(photodb):
'tag_group_rel': '*', 'tag_group_rel': '*',
'tag_synonyms': '*', 'tag_synonyms': '*',
} }
cur = photodb.sql.cursor()
cur.execute('PRAGMA foreign_keys = OFF')
print('Renaming existing tables.') print('Renaming existing tables.')
for table in tables_to_copy: for table in tables_to_copy:
statement = 'ALTER TABLE %s RENAME TO %s_old' % (table, table) statement = 'ALTER TABLE %s RENAME TO %s_old' % (table, table)
@ -223,7 +232,6 @@ def upgrade_10_to_11(photodb):
for statement in create_indices: for statement in create_indices:
cur.execute(statement) cur.execute(statement)
cur.execute('PRAGMA foreign_keys = ON')
def upgrade_11_to_12(photodb): def upgrade_11_to_12(photodb):
''' '''
@ -231,6 +239,7 @@ def upgrade_11_to_12(photodb):
improve the speed of individual relation searching, important for the improve the speed of individual relation searching, important for the
new intersection-based search. new intersection-based search.
''' '''
photodb.sql_execute('BEGIN')
query = ''' query = '''
CREATE INDEX IF NOT EXISTS index_photo_tag_rel_photoid_tagid on photo_tag_rel(photoid, tagid) CREATE INDEX IF NOT EXISTS index_photo_tag_rel_photoid_tagid on photo_tag_rel(photoid, tagid)
''' '''
@ -241,9 +250,10 @@ def upgrade_12_to_13(photodb):
Added display_name column to the User table. Added display_name column to the User table.
''' '''
cur = photodb.sql.cursor() cur = photodb.sql.cursor()
cur.execute('PRAGMA foreign_keys = OFF') photodb.sql_execute('PRAGMA foreign_keys = OFF')
cur.execute('ALTER TABLE users RENAME TO users_old') photodb.sql_execute('BEGIN')
cur.execute(''' photodb.sql_execute('ALTER TABLE users RENAME TO users_old')
photodb.sql_execute('''
CREATE TABLE users( CREATE TABLE users(
id TEXT PRIMARY KEY NOT NULL, id TEXT PRIMARY KEY NOT NULL,
username TEXT NOT NULL COLLATE NOCASE, username TEXT NOT NULL COLLATE NOCASE,
@ -253,12 +263,12 @@ def upgrade_12_to_13(photodb):
)''') )''')
cur.execute('INSERT INTO users SELECT id, username, password, NULL, created FROM users_old') cur.execute('INSERT INTO users SELECT id, username, password, NULL, created FROM users_old')
cur.execute('DROP TABLE users_old') cur.execute('DROP TABLE users_old')
cur.execute('PRAGMA foreign_keys = ON')
def upgrade_13_to_14(photodb): def upgrade_13_to_14(photodb):
''' '''
Rename user.min_length to min_username_length. Rename user.min_length to min_username_length.
''' '''
photodb.sql_execute('BEGIN')
photodb.config['user']['min_username_length'] = photodb.config['user'].pop('min_length') photodb.config['user']['min_username_length'] = photodb.config['user'].pop('min_length')
photodb.config['user']['max_username_length'] = photodb.config['user'].pop('max_length') photodb.config['user']['max_username_length'] = photodb.config['user'].pop('max_length')
photodb.save_config() photodb.save_config()
@ -286,7 +296,7 @@ def upgrade_all(data_directory):
upgrade_function = eval(upgrade_function) upgrade_function = eval(upgrade_function)
try: try:
photodb.sql.execute('BEGIN') photodb.sql_execute('PRAGMA foreign_keys = ON')
upgrade_function(photodb) upgrade_function(photodb)
except Exception as exc: except Exception as exc:
photodb.rollback() photodb.rollback()