def upgrade()

in annotation/alembic/versions/c053ae380212_expand_job_model.py [0:0]

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

def upgrade():
    op.rename_table("annotators", "users")
    op.create_table(
        "association_job_validator",
        sa.Column("user_id", postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column("job_id", sa.INTEGER(), nullable=False),
        sa.ForeignKeyConstraint(
            ["job_id"],
            ["jobs.job_id"],
        ),
        sa.ForeignKeyConstraint(
            ["user_id"],
            ["users.user_id"],
        ),
        sa.PrimaryKeyConstraint("user_id", "job_id"),
    )
    op.create_table(
        "association_job_owner",
        sa.Column("user_id", postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column("job_id", sa.INTEGER(), nullable=False),
        sa.ForeignKeyConstraint(
            ["job_id"],
            ["jobs.job_id"],
        ),
        sa.ForeignKeyConstraint(
            ["user_id"],
            ["users.user_id"],
        ),
        sa.PrimaryKeyConstraint("user_id", "job_id"),
    )

    validation_type = postgresql.ENUM(ValidationSchema, name="validation_type")
    validation_type.create(op.get_bind(), checkfirst=True)
    op.add_column(
        "jobs",
        sa.Column(
            "validation_type",
            validation_type,
            nullable=False,
            server_default=ValidationSchema.cross.name,
        ),
    )