Function cancel_subscription

app/api/agentops/opsboard/views/orgs.py:1447–1513

Cancel a Stripe subscription for an organization immediately.

(
    *,
    request: Request,
    org_id: str,
    body: CancelSubscriptionBody,
    orm: Session = Depends(get_orm_session),
)

async def cancel_subscription(
    *,
    request: Request,
    org_id: str,
    body: CancelSubscriptionBody,
    orm: Session = Depends(get_orm_session),
) -> StatusResponse:
    """
    Cancel a Stripe subscription for an organization immediately.
    """
    stripe.api_key = STRIPE_SECRET_KEY

    if not STRIPE_SECRET_KEY:
        raise HTTPException(status_code=500, detail="Stripe secret key not configured.")

    user: Optional[UserModel] = UserModel.get_by_id(orm, request.state.session.user_id)
    if not user:  # Minimal check, primary auth is via endpoint protection
        raise HTTPException(status_code=401, detail="User not authenticated.")

    org: Optional[OrgModel] = OrgModel.get_by_id(orm, org_id)

    if not org:
        raise HTTPException(status_code=404, detail="Organization not found")

    if not org.is_user_admin_or_owner(request.state.session.user_id):
        raise HTTPException(
            status_code=403,
            detail="User does not have permission to manage this organization's subscription.",
        )

    if not org.subscription_id:
        raise HTTPException(status_code=400, detail="Organization does not have an active subscription.")

    if org.subscription_id != body.subscription_id:
        raise HTTPException(status_code=400, detail="Subscription ID mismatch.")

    try:
        # Add idempotency key to prevent duplicate cancellation requests
        idempotency_key = f"cancel_{org.id}_{body.subscription_id}_{int(time.time())}"
        stripe.Subscription.modify(
            body.subscription_id, cancel_at_period_end=True, idempotency_key=idempotency_key
        )

        logger.info(f"Subscription {body.subscription_id} set to cancel at period end for org {org_id}")

        return StatusResponse(
            message="Subscription will be cancelled at the end of the current billing period."
        )

    except stripe.error.StripeError as e:
        orm.rollback()
        logger.error(
            "Stripe error cancelling subscription %s for org %s: %s", body.subscription_id, org_id, str(e)
        )
        raise HTTPException(
            status_code=500,
            detail=(
                "Stripe error: Could not cancel subscription. "