From 06a565c85edd7c9b830c4bccc02048b6957a731c Mon Sep 17 00:00:00 2001
From: Marijana Petojevic <marijana.petojevic@tuwien.ac.at>
Date: Wed, 9 Oct 2024 22:14:55 +0200
Subject: [PATCH] feat: alembic revision for auto release #222

---
 .../versions/e52c1a6c91fb_auto_release.py     | 58 +++++++++++++++++++
 grader_service/orm/assignment.py              |  2 +-
 2 files changed, 59 insertions(+), 1 deletion(-)
 create mode 100644 grader_service/migrate/versions/e52c1a6c91fb_auto_release.py

diff --git a/grader_service/migrate/versions/e52c1a6c91fb_auto_release.py b/grader_service/migrate/versions/e52c1a6c91fb_auto_release.py
new file mode 100644
index 00000000..e9eb2c1e
--- /dev/null
+++ b/grader_service/migrate/versions/e52c1a6c91fb_auto_release.py
@@ -0,0 +1,58 @@
+"""auto release
+
+Revision ID: e52c1a6c91fb
+Revises: a0718dae969d
+Create Date: 2024-10-09 13:21:54.080731
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+# revision identifiers, used by Alembic.
+revision = 'e52c1a6c91fb'
+down_revision = 'a0718dae969d'
+branch_labels = None
+depends_on = None
+
+new_enum = sa.Enum(
+    'created', 'pushed', 'released', 'fetching', 'fetched', 'complete', 'release_scheduled',
+    name='assignment_status'
+)
+
+
+def upgrade():
+    # Sqlite doesn't support ALTER table, so make backup of status column, drop original status column and make a new one wit new enum values
+    op.add_column('assignment', sa.Column('status_backup', sa.String(), nullable=True))
+    op.execute("UPDATE assignment SET status_backup = status")
+    op.drop_column('assignment', 'status')
+    new_enum.create(op.get_bind(), checkfirst=True)
+    op.add_column('assignment', sa.Column('status', new_enum, nullable=True))
+    op.execute("UPDATE assignment SET status = status_backup")
+    op.drop_column('assignment', 'status_backup')
+
+
+def downgrade():
+    # Remove 'release_scheduled' from assignment status options
+    op.add_column('assignment', sa.Column('status_backup', sa.String(), nullable=True))
+
+    op.execute(" UPDATE assignment SET status_backup = status")
+
+    op.drop_column('assignment', 'status')
+    old_enum = sa.Enum(
+        'created', 'pushed', 'released', 'fetching', 'fetched', 'complete',
+        name='assignment_status'
+    )
+    old_enum.create(op.get_bind(), checkfirst=True)
+    op.add_column('assignment', sa.Column('status', old_enum, nullable=True))
+
+    # If assignment status was 'release_scheduled' after downgrade it should be 'created' as the assignment is still not 'released'
+    op.execute("""
+            UPDATE assignment 
+            SET status = CASE 
+                WHEN status_backup = 'release_scheduled' THEN 'created'
+                ELSE status_backup 
+            END;
+        """)
+
+    op.drop_column('assignment', 'status_backup')
+    new_enum.drop(op.get_bind(), checkfirst=True)
diff --git a/grader_service/orm/assignment.py b/grader_service/orm/assignment.py
index ffe3a3be..d168c867 100644
--- a/grader_service/orm/assignment.py
+++ b/grader_service/orm/assignment.py
@@ -33,7 +33,7 @@ class Assignment(Base, Serializable):
     duedate = Column(DateTime, nullable=False)
     points = Column(DECIMAL(10, 3), nullable=True)
     status = Column(
-        Enum("created", "pushed", "released", "complete"),
+        Enum("created", "pushed", "released", "complete", "release_scheduled"),
         default="created",
     )
     automatic_grading = Column(Enum(AutoGradingBehaviour), nullable=False)