Добавлен новый файл call_validation_promo_api.py для валидации промокодов через API. Реализована логика получения JWT токена и проверки валидности промокода. Обновлены модели в integration_models.py для поддержки запросов и ответов на валидацию промокодов. Добавлен новый эндпоинт /validationPromo в integration_api.py для обработки запросов на валидацию промокодов.

integration_api.py

@@ -8,7 +8,7 @@ from random import choices
 import string
 
 from sql_models import Company, IntegrationToken, Ref, Sale, AgentTransaction, PartnerTransaction, AgentBalance, TgAgent, CompanyBalance, SaleCategory
-from integration_models import Token, SaleCreateRequest, SaleCreateResponse, TransactionStatus, WithdrawRequest, WithdrawResponse
+from integration_models import Token, SaleCreateRequest, SaleCreateResponse, TransactionStatus, WithdrawRequest, WithdrawResponse, PromoValidationRequest, PromoValidationResponse
 from bff_models import RegisterResponse, TgAuthResponse
 from tg_models import RefAddRequest, RefResponse, RefAddResponse, RefStatResponse, RegisterRequest, StatResponse
 from helpers_bff import AUTH_DB_ENGINE, get_integration_db, create_integration_jwt_token, get_current_company_from_jwt, get_tg_agent_by_tg_id, get_current_tg_agent
@@ -289,6 +289,25 @@ async def create_sale(
         "crediting": new_sale.crediting
     }
 
+@app.post("/validationPromo", tags=["integration"], response_model=PromoValidationResponse)
+async def validate_promocode(
+    req: PromoValidationRequest,
+    company: Company = Depends(get_current_company_from_jwt),
+    db: Session = Depends(get_integration_db)
+):
+    """
+    Проверяет валидность промокода для текущей компании.
+    """
+    referral = db.exec(select(Ref).where(Ref.promocode == req.promocode)).first()
+    if not referral:
+        return {"validation": False}
+
+    tg_agent = db.exec(select(TgAgent).where(TgAgent.id == referral.tg_agent_id, TgAgent.company_id == company.id)).first()
+    if not tg_agent:
+        return {"validation": False}
+
+    return {"validation": True}
+
 @app.post("/register", tags=["agent-tg"], response_model=RegisterResponse)
 def register(req: RegisterRequest, db: Session = Depends(get_integration_db)):
     """