Fix sqlite only migration. (#579)

spiral-project · Apr 25, 2020 · 6129191 · 6129191
1 parent d6d084f
commit 6129191
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 33 deletions.
diff --git a/.gitignore b/.gitignore
@@ -10,5 +10,5 @@ dist
 build
 .vscode
 .env
-.pytest_cache
-
+.pytest_cache
+ihatemoney/budget.db
diff --git a/docs/contributing.rst b/docs/contributing.rst
@@ -220,7 +220,7 @@ In order to prepare a new release, we are following the following steps:
     make compress-assets
 
 - Build the translations::
- 
+
     make update-translations
     make build-translations
 

diff --git a/ihatemoney/budget.db b/ihatemoney/budget.db
diff --git a/ihatemoney/migrations/versions/cb038f79982e_sqlite_autoincrement.py b/ihatemoney/migrations/versions/cb038f79982e_sqlite_autoincrement.py
@@ -15,36 +15,46 @@
 
 
 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