From 79de9bd5555e9a4c8c1aa62a5cb6b719e1ad96c4 Mon Sep 17 00:00:00 2001
From: tino097 <konstantin.sivakov@gmail.com>
Date: Mon, 6 May 2024 19:37:49 +0200
Subject: [PATCH] Fix the migration

---
 .../01_e9c4a88839c8_upgrade_jobs_table.py       | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/ckanext/datapusher_plus/migration/datapusher_plus/versions/01_e9c4a88839c8_upgrade_jobs_table.py b/ckanext/datapusher_plus/migration/datapusher_plus/versions/01_e9c4a88839c8_upgrade_jobs_table.py
index 254a7f9..55c8958 100644
--- a/ckanext/datapusher_plus/migration/datapusher_plus/versions/01_e9c4a88839c8_upgrade_jobs_table.py
+++ b/ckanext/datapusher_plus/migration/datapusher_plus/versions/01_e9c4a88839c8_upgrade_jobs_table.py
@@ -19,18 +19,13 @@
 
 def upgrade():
     # upgrade jobs table if it not exists
-    if _check_column_exists("jobs", "aps_job_id"):
-        return
-    else:
+    if not _check_column_exists("jobs", "aps_job_id"):
         op.add_column(
             "jobs",
             sa.Column("aps_job_id", sa.UnicodeText),
         )
-
     # upgrade logs table
-    if _check_column_exists("logs", "id"):
-        return
-    else:
+    if not _check_column_exists("logs", "id"):
         op.add_column(
             "logs",
             sa.Column("id", sa.Integer, primary_key=True, autoincrement=True),
@@ -38,7 +33,13 @@ def upgrade():
 
 
 def downgrade():
-    pass
+    # downgrade jobs table
+    if _check_column_exists("jobs", "aps_job_id"):
+        op.drop_column("jobs", "aps_job_id")
+
+    # downgrade logs table
+    if _check_column_exists("logs", "id"):
+        op.drop_column("logs", "id")
 
 
 def _check_column_exists(table_name, column_name):