From 98fca6403eb7ae3c9532ef4075cabc82d9a13ac1 Mon Sep 17 00:00:00 2001
From: Andrea Callia D'Iddio <ac4014@ic.ac.uk>
Date: Wed, 24 Aug 2022 12:18:05 +0200
Subject: [PATCH] Feat: add staff_ranking column to the shortlisting DB table.

---
 app/models/shortlist.py                       |  7 ++++
 .../ce0f2fd42857_add_staff_ranking_column.py  | 41 +++++++++++++++++++
 2 files changed, 48 insertions(+)
 create mode 100644 migrations/versions/ce0f2fd42857_add_staff_ranking_column.py

diff --git a/app/models/shortlist.py b/app/models/shortlist.py
index 96930d1..dcd007a 100644
--- a/app/models/shortlist.py
+++ b/app/models/shortlist.py
@@ -3,6 +3,7 @@ from sqlalchemy import Sequence
 from ..database import db
 
 RANKING_SEQUENCE = Sequence("shortlisting_ranking_seq")
+STAFF_RANKING_SEQUENCE = Sequence("shortlisting_staff_ranking_seq")
 
 
 class Shortlisting(db.Model):
@@ -15,3 +16,9 @@ class Shortlisting(db.Model):
         server_default=RANKING_SEQUENCE.next_value(),
         nullable=False,
     )
+    staff_ranking = db.Column(
+        db.Integer,
+        STAFF_RANKING_SEQUENCE,
+        server_default=STAFF_RANKING_SEQUENCE.next_value(),
+        nullable=False,
+    )
diff --git a/migrations/versions/ce0f2fd42857_add_staff_ranking_column.py b/migrations/versions/ce0f2fd42857_add_staff_ranking_column.py
new file mode 100644
index 0000000..2a203cf
--- /dev/null
+++ b/migrations/versions/ce0f2fd42857_add_staff_ranking_column.py
@@ -0,0 +1,41 @@
+"""Add staff_ranking column
+
+Revision ID: ce0f2fd42857
+Revises: d009c2b07eed
+Create Date: 2022-08-24 10:13:42.707243
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+from sqlalchemy import Sequence
+from sqlalchemy.sql.ddl import DropSequence, CreateSequence
+
+revision = "ce0f2fd42857"
+down_revision = "d009c2b07eed"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.execute(CreateSequence(Sequence("shortlisting_staff_ranking_seq")))
+    op.add_column(
+        "shortlisting",
+        sa.Column(
+            "staff_ranking",
+            sa.Integer(),
+            server_default=sa.text("nextval('shortlisting_staff_ranking_seq')"),
+            nullable=False,
+        ),
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column("shortlisting", "staff_ranking")
+    op.execute(DropSequence(Sequence("shortlisting_staff_ranking_seq")))
+    # ### end Alembic commands ###
-- 
GitLab