Интеграция биллинг при покупке подписки

app/routes/subscription_routes.py

@@ -43,7 +43,7 @@ async def buy_subscription(
 
         logger.info(f"Результат buy_sub: {result}")
 
-        if result == "ERROR" or result is None:
+        if result == "ERROR":
             raise HTTPException(status_code=500, detail="Internal server error")
         elif result == "INSUFFICIENT_FUNDS":
             raise HTTPException(status_code=400, detail="INSUFFICIENT_FUNDS")
@@ -51,12 +51,24 @@ async def buy_subscription(
             raise HTTPException(status_code=400, detail="TARIFF_NOT_FOUND")
         elif result == "ACTIVE_SUBSCRIPTION_EXISTS":
             raise HTTPException(status_code=400, detail="ACTIVE_SUBSCRIPTION_EXISTS")
+        elif result == "USER_NOT_FOUND":
+            raise HTTPException(status_code=404, detail="USER_NOT_FOUND")
+        elif result == "SUBSCRIPTION_CREATION_FAILED":
+            raise HTTPException(status_code=500, detail="Failed to create subscription")
+        elif result == "PAYMENT_FAILED_AFTER_SUBSCRIPTION":
+            raise HTTPException(status_code=402, detail="SUBSCRIPTION_CREATED_BUT_PAYMENT_FAILED")
+        elif result == "SUBSCRIPTION_CREATED_BUT_PAYMENT_FAILED":
+            raise HTTPException(status_code=402, detail="SUBSCRIPTION_CREATED_BUT_PAYMENT_FAILED")
         
         # Если успешно, генерируем URI
         if isinstance(result, dict) and result.get('status') == 'OK':
             uri_result = await database_manager.generate_uri(request_data.telegram_id)
             logger.info(f"Результат генерации URI: {uri_result}")
-            return {"status": "success", "subscription_id": result.get('subscription_id'), "uri": uri_result[0]}
+            return {
+                "status": "success", 
+                "subscription_id": result.get('subscription_id'), 
+                "uri": uri_result[0] if uri_result and isinstance(uri_result, list) else uri_result
+            }
         else:
             return {"status": "success", "message": "Subscription created"}
             
@@ -68,6 +80,7 @@ async def buy_subscription(
         raise HTTPException(status_code=500, detail=f"Unexpected error: {str(e)}")
 
 
+
 # Эндпоинт для получения последней подписки
 @router.get("/subscription/{telegram_id}/last", response_model=SubscriptionResponse)
 async def last_subscription(telegram_id: int, database_manager: DatabaseManager = Depends(get_database_manager)):