pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

hard-code status in invitation status migration

Browse Source
This commit is contained in:
dandds
2018-12-13 09:45:01 -05:00
parent 156ea02c90
commit d0ed04377c
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
alembic/versions/e1081cf01780_adjust_invitation_status.py
View File

@@ -7,10 +7,8 @@ Create Date: 2018-11-01 12:24:10.970963
""" """
from alembic import op from alembic import op
import sqlalchemy as sa import sqlalchemy as sa
from atst.models.invitation import Status
from enum import Enum from enum import Enum
# revision identifiers, used by Alembic. # revision identifiers, used by Alembic.
revision = 'e1081cf01780' revision = 'e1081cf01780'
down_revision = 'a9d8c6b6221c' down_revision = 'a9d8c6b6221c'
@@ -18,6 +16,13 @@ branch_labels = None
depends_on = 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(): def upgrade():
conn = op.get_bind() conn = op.get_bind()
constraints = ", ".join(["'{}'::character varying::text".format(s.name) for s in Status]) constraints = ", ".join(["'{}'::character varying::text".format(s.name) for s in Status])