def upgrade()

in annotation/alembic/versions/9c07a25ca06f_expand_file_model.py [0:0]

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

def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column(
        "files",
        sa.Column(
            "distributed_annotating_pages",
            postgresql.ARRAY(sa.INTEGER()),
            nullable=False,
            server_default="{}",
        ),
    )
    op.add_column(
        "files",
        sa.Column(
            "annotated_pages",
            postgresql.ARRAY(sa.INTEGER()),
            nullable=False,
            server_default="{}",
        ),
    )
    op.add_column(
        "files",
        sa.Column(
            "distributed_validating_pages",
            postgresql.ARRAY(sa.INTEGER()),
            nullable=False,
            server_default="{}",
        ),
    )
    op.add_column(
        "files",
        sa.Column(
            "validated_pages",
            postgresql.ARRAY(sa.INTEGER()),
            nullable=False,
            server_default="{}",
        ),
    )
    # ### end Alembic commands ###
    file_status = postgresql.ENUM(FileStatusEnumSchema, name="file_status")
    file_status.create(op.get_bind(), checkfirst=True)
    op.add_column(
        "files",
        sa.Column(
            "status",
            file_status,
            nullable=False,
            server_default=FileStatusEnumSchema.pending.name,
        ),
    )