def upgrade()

in src/Backend/migrations/versions/2023_05_30_173335-09bcd5ecbf1d_moving_grade_to_enum.py [0:0]

• 26 lines of code
• 1 McCabe index (conditional complexity)

def upgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    grades_enum = psql.ENUM(
        "FIRST_FUND",
        "SECOND_FUND",
        "THIRD_FUND",
        "FOURTH_FUND",
        "FIFTH_FUND",
        "SIXTH_FUND",
        "SEVENTH_FUND",
        "EIGHTH_FUND",
        "NINETH_FUND",
        "FIRST_HS",
        "SECOND_HS",
        "THIRD_HS",
        name="grades",
        create_type=False,
    )
    grades_enum.create(op.get_bind(), checkfirst=False)
    op.add_column(
        "exams", sa.Column("start_date", sa.DateTime(timezone=True), nullable=False)
    )
    op.add_column(
        "exams", sa.Column("end_date", sa.DateTime(timezone=True), nullable=False)
    )
    op.drop_column("groups", "grade")
    op.add_column("groups", sa.Column("grade", grades_enum, nullable=False))