diff options
| author | Rémy HUBSCHER <hubscher.remy@gmail.com> | 2020-04-25 11:55:46 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-04-25 11:55:46 +0200 |
| commit | 6129191b26784b895e203fa3eafb89cee7d88b71 (patch) | |
| tree | 5ad02feb6ca0f9c918810dfcdb92ed819237d499 /ihatemoney/migrations/versions/cb038f79982e_sqlite_autoincrement.py | |
| parent | d6d084f26a9de543527486e3c1fbc4e9100fddbd (diff) | |
| download | ihatemoney-mirror-6129191b26784b895e203fa3eafb89cee7d88b71.zip ihatemoney-mirror-6129191b26784b895e203fa3eafb89cee7d88b71.tar.gz ihatemoney-mirror-6129191b26784b895e203fa3eafb89cee7d88b71.tar.bz2 | |
Fix sqlite only migration. (#579)
Diffstat (limited to 'ihatemoney/migrations/versions/cb038f79982e_sqlite_autoincrement.py')
| -rw-r--r-- | ihatemoney/migrations/versions/cb038f79982e_sqlite_autoincrement.py | 70 |
1 files changed, 40 insertions, 30 deletions
diff --git a/ihatemoney/migrations/versions/cb038f79982e_sqlite_autoincrement.py b/ihatemoney/migrations/versions/cb038f79982e_sqlite_autoincrement.py index ae5ab32..718aa75 100644 --- a/ihatemoney/migrations/versions/cb038f79982e_sqlite_autoincrement.py +++ b/ihatemoney/migrations/versions/cb038f79982e_sqlite_autoincrement.py @@ -15,36 +15,46 @@ import sqlalchemy as sa def upgrade(): - alter_table_batches = [ - op.batch_alter_table( - "person", recreate="always", table_kwargs={"sqlite_autoincrement": True} - ), - op.batch_alter_table( - "bill", recreate="always", table_kwargs={"sqlite_autoincrement": True} - ), - op.batch_alter_table( - "billowers", recreate="always", table_kwargs={"sqlite_autoincrement": True} - ), - ] - - for batch_op in alter_table_batches: - with batch_op: - pass + bind = op.get_bind() + if bind.engine.name == "sqlite": + alter_table_batches = [ + op.batch_alter_table( + "person", recreate="always", table_kwargs={"sqlite_autoincrement": True} + ), + op.batch_alter_table( + "bill", recreate="always", table_kwargs={"sqlite_autoincrement": True} + ), + op.batch_alter_table( + "billowers", + recreate="always", + table_kwargs={"sqlite_autoincrement": True}, + ), + ] + + for batch_op in alter_table_batches: + with batch_op: + pass def downgrade(): - alter_table_batches = [ - op.batch_alter_table( - "person", recreate="always", table_kwargs={"sqlite_autoincrement": False} - ), - op.batch_alter_table( - "bill", recreate="always", table_kwargs={"sqlite_autoincrement": False} - ), - op.batch_alter_table( - "billowers", recreate="always", table_kwargs={"sqlite_autoincrement": False} - ), - ] - - for batch_op in alter_table_batches: - with batch_op: - pass + bind = op.get_bind() + if bind.engine.name == "sqlite": + alter_table_batches = [ + op.batch_alter_table( + "person", + recreate="always", + table_kwargs={"sqlite_autoincrement": False}, + ), + op.batch_alter_table( + "bill", recreate="always", table_kwargs={"sqlite_autoincrement": False} + ), + op.batch_alter_table( + "billowers", + recreate="always", + table_kwargs={"sqlite_autoincrement": False}, + ), + ] + + for batch_op in alter_table_batches: + with batch_op: + pass |