Fix migration

nens · Jan 23, 2024 · f90ee1b · f90ee1b
1 parent f8005fe
commit f90ee1b
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/threedi_schema/migrations/versions/0219_nullable_cross_section_location_friction_value.py b/threedi_schema/migrations/versions/0219_nullable_cross_section_location_friction_value.py
@@ -1,8 +1,8 @@
 """rename vegetation columns
 
-Revision ID: 0218
-Revises: 0217
-Create Date: 2023-12-22 08:33
+Revision ID: 0219
+Revises: 0218
+Create Date: 2024-01-22 15:00
 
 """
 
@@ -15,7 +15,18 @@
 branch_labels = None
 depends_on = None
 
+MIGRATION_QUERIES = """
+SELECT RecoverGeometryColumn('v2_cross_section_location', 'the_geom', 4326, 'POINT', 'XY')
+"""
+
 
 def upgrade():
     with op.batch_alter_table("v2_cross_section_location") as batch_op:
-        batch_op.alter_column("friction_value", nullable=True, type_=sa.TEXT)
+        batch_op.alter_column("friction_value", nullable=True, type_=sa.Float)
+    for q in MIGRATION_QUERIES.split(";"):
+        op.execute(sa.text(q))
+
+
+def downgrade():
+    with op.batch_alter_table("v2_cross_section_location") as batch_op:
+        batch_op.alter_column("friction_value", nullable=False, type_=sa.Float)