def upgrade()

in lemur/migrations/versions/3adfdd6598df_.py [0:0]

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

def upgrade():
    # create provider table
    print("Creating dns_providers table")
    op.create_table(
        "dns_providers",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("name", sa.String(length=256), nullable=True),
        sa.Column("description", sa.String(length=1024), nullable=True),
        sa.Column("provider_type", sa.String(length=256), nullable=True),
        sa.Column("credentials", Vault(), nullable=True),
        sa.Column("api_endpoint", sa.String(length=256), nullable=True),
        sa.Column(
            "date_created", ArrowType(), server_default=sa.text("now()"), nullable=False
        ),
        sa.Column("status", sa.String(length=128), nullable=True),
        sa.Column("options", JSON),
        sa.Column("domains", sa.JSON(), nullable=True),
        sa.PrimaryKeyConstraint("id"),
        sa.UniqueConstraint("name"),
    )
    print("Adding dns_provider_id column to certificates")
    op.add_column(
        "certificates", sa.Column("dns_provider_id", sa.Integer(), nullable=True)
    )
    print("Adding dns_provider_id column to pending_certs")
    op.add_column(
        "pending_certs", sa.Column("dns_provider_id", sa.Integer(), nullable=True)
    )
    print("Adding options column to pending_certs")
    op.add_column("pending_certs", sa.Column("options", JSON))

    print("Creating pending_dns_authorizations table")
    op.create_table(
        "pending_dns_authorizations",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("account_number", sa.String(length=128), nullable=True),
        sa.Column("domains", JSON, nullable=True),
        sa.Column("dns_provider_type", sa.String(length=128), nullable=True),
        sa.Column("options", JSON, nullable=True),
    )

    print("Creating certificates_dns_providers_fk foreign key")
    op.create_foreign_key(
        "certificates_dns_providers_fk",
        "certificates",
        "dns_providers",
        ["dns_provider_id"],
        ["id"],
        ondelete="cascade",
    )

    print("Altering column types in the api_keys table")
    op.alter_column("api_keys", "issued_at", existing_type=sa.BIGINT(), nullable=True)
    op.alter_column("api_keys", "revoked", existing_type=sa.BOOLEAN(), nullable=True)
    op.alter_column("api_keys", "ttl", existing_type=sa.BIGINT(), nullable=True)
    op.alter_column("api_keys", "user_id", existing_type=sa.INTEGER(), nullable=True)

    print("Creating dns_providers_id foreign key on pending_certs table")
    op.create_foreign_key(
        None,
        "pending_certs",
        "dns_providers",
        ["dns_provider_id"],
        ["id"],
        ondelete="CASCADE",
    )