Spec added + migrations (#12)

profcomff · Nov 8, 2022 · b536043 · b536043
1 parent 9198b7e
commit b536043
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/design_bot/models/db.py b/design_bot/models/db.py
@@ -18,6 +18,8 @@ class Year(str, Enum):
     FOURTH = "4"
     FIFTH = "1м"
     SIXTH = "2м"
+    FIFTH_SPEC = "5"
+    SIXTH_SPEC = "6"
 
 
 class Directions(str, Enum):

diff --git a/migrations/versions/c54ae0974f75_spec_add.py b/migrations/versions/c54ae0974f75_spec_add.py
@@ -0,0 +1,18 @@
+from alembic import op
+import sqlalchemy as sa
+
+
+revision = 'c54ae0974f75'
+down_revision = 'd124a172c3b3'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    op.execute('ALTER TABLE "user" ALTER COLUMN year TYPE varchar(10) USING year::varchar(6);')
+
+
+def downgrade():
+    op.execute('DELETE FROM "response" WHERE user_id IN (SELECT id FROM "user" WHERE LENGTH(year) > 6) RETURNING *')
+    op.execute('DELETE FROM "user" WHERE LENGTH(year) > 6')
+    op.execute('ALTER TABLE "user" ALTER COLUMN year TYPE varchar(6) USING year::varchar(10);')