Merge pull request #172 from dathere/migration-added

Migration added
dathere · Oct 30, 2024 · 8c86c1e · 8c86c1e
2 parents 452afec + a854720
commit 8c86c1e
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/ckanext/datapusher_plus/migration/datapusher_plus/versions/03_23d45fa26447_add_job_key.py b/ckanext/datapusher_plus/migration/datapusher_plus/versions/03_23d45fa26447_add_job_key.py
@@ -17,8 +17,16 @@
 
 
 def upgrade():
-    op.add_column('jobs', sa.Column('job_key', sa.UnicodeText(), nullable=True))
+    if not _check_column_exists('jobs', 'job_key'):
+       op.add_column('jobs', sa.Column('job_key', sa.UnicodeText(), nullable=True))
 
 
 def downgrade():
     op.drop_column('jobs', 'job_key')
+
+
+def _check_column_exists(table_name, column_name):
+    bind = op.get_bind()
+    insp = sa.engine.reflection.Inspector.from_engine(bind)
+    columns = insp.get_columns(table_name)
+    return column_name in [column["name"] for column in columns]