From d0ed04377cb9cab89b8c76db88b4eaacb89193d7 Mon Sep 17 00:00:00 2001 From: dandds Date: Thu, 13 Dec 2018 09:45:01 -0500 Subject: [PATCH] hard-code status in invitation status migration --- .../versions/e1081cf01780_adjust_invitation_status.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/alembic/versions/e1081cf01780_adjust_invitation_status.py b/alembic/versions/e1081cf01780_adjust_invitation_status.py index a2c552cc..368f39b8 100644 --- a/alembic/versions/e1081cf01780_adjust_invitation_status.py +++ b/alembic/versions/e1081cf01780_adjust_invitation_status.py @@ -7,10 +7,8 @@ Create Date: 2018-11-01 12:24:10.970963 """ from alembic import op import sqlalchemy as sa -from atst.models.invitation import Status from enum import Enum - # revision identifiers, used by Alembic. revision = 'e1081cf01780' down_revision = 'a9d8c6b6221c' @@ -18,6 +16,13 @@ branch_labels = None depends_on = None +class Status(Enum): + ACCEPTED = "accepted" + REVOKED = "revoked" + PENDING = "pending" + REJECTED_WRONG_USER = "rejected_wrong_user" + REJECTED_EXPIRED = "rejected_expired" + def upgrade(): conn = op.get_bind() constraints = ", ".join(["'{}'::character varying::text".format(s.name) for s in Status])