def upgrade()

in annotation/alembic/versions/6fb3e0d231ff_create_document_links_table.py [0:0]
42 lines of code
1 McCabe index (conditional complexity)

def upgrade():
    op.create_table(
        "document_links",
        sa.Column("original_revision", sa.VARCHAR),
        sa.Column("original_file_id", sa.INTEGER),
        sa.Column("original_job_id", sa.INTEGER),
        sa.Column("similar_revision", sa.VARCHAR),
        sa.Column("similar_file_id", sa.INTEGER),
        sa.Column("similar_job_id", sa.INTEGER),
        sa.Column(
            "label",
            sa.VARCHAR,
            sa.ForeignKey("categories.id", ondelete="SET NULL"),
            nullable=False,
        ),
        sa.ForeignKeyConstraint(
            ("original_revision", "original_file_id", "original_job_id"),
            (
                "annotated_docs.revision",
                "annotated_docs.file_id",
                "annotated_docs.job_id",
            ),
            ondelete="cascade",
        ),
        sa.ForeignKeyConstraint(
            ("similar_revision", "similar_file_id", "similar_job_id"),
            (
                "annotated_docs.revision",
                "annotated_docs.file_id",
                "annotated_docs.job_id",
            ),
            ondelete="cascade",
        ),
        sa.PrimaryKeyConstraint(
            "original_revision",
            "original_file_id",
            "original_job_id",
            "similar_revision",
            "similar_file_id",
            "similar_job_id",
        ),
    )