Добавлено новое поле promocode в модель StatReferralsItem и модель Ref. Обновлены функции в fill_db.py для генерации уникальных промокодов при создании реферальных ссылок. Обновлены эндпоинты в integration_api.py для возврата промокодов вместе с реферальными ссылками. Удалены устаревшие функции работы с промокодами из кода. Обновлены SQL-скрипты и модели для учета новых полей.

integration_api.py

@@ -7,10 +7,10 @@ import uuid
 from random import choices
 import string
 
-from sql_models import Company, IntegrationToken, Ref, Sale, AgentTransaction, PartnerTransaction, AgentBalance, TgAgent, CompanyBalance, PromoCode
+from sql_models import Company, IntegrationToken, Ref, Sale, AgentTransaction, PartnerTransaction, AgentBalance, TgAgent, CompanyBalance
 from integration_models import Token, SaleCreateRequest, SaleCreateResponse, TransactionStatus, WithdrawRequest, WithdrawResponse
 from bff_models import RegisterResponse, TgAuthResponse
-from tg_models import RefAddRequest, RefResponse, RefAddResponse, RefStatResponse, RegisterRequest, StatResponse, PromoCodeAddRequest, PromoCodeResponse
+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
 
 app = FastAPI()
@@ -61,22 +61,28 @@ def get_refs(current_tg_agent: TgAgent = Depends(get_current_tg_agent), db: Sess
     Возвращает список реферальных ссылок текущего Telegram-агента.
     """
     refs = db.exec(select(Ref).where(Ref.tg_agent_id == current_tg_agent.id)).all()
-    return [RefResponse(ref=r.ref, description=r.description or "") for r in refs]
+    return [RefResponse(ref=r.ref, description=r.description or "", promocode=r.promocode) for r in refs]
 
 @app.post("/ref/add", tags=["agent-tg"], response_model=RefAddResponse)
 def add_ref(req: RefAddRequest, current_tg_agent: TgAgent = Depends(get_current_tg_agent), db: Session = Depends(get_integration_db)):
     """
     Добавляет новую реферальную ссылку для текущего Telegram-агента.
     """
+    # Генерация промокода (логика как была для промокодов)
+    alphabet = string.ascii_letters + string.digits + "!@#$%^&*"
+    promocode = ''.join(choices(alphabet, k=8))
+    while db.exec(select(Ref).where(Ref.promocode == promocode)).first():
+        promocode = ''.join(choices(alphabet, k=8))
     new_ref = Ref(
         tg_agent_id=current_tg_agent.id,
         ref=str(uuid.uuid4()),
-        description=req.description
+        description=req.description,
+        promocode=promocode
     )
     db.add(new_ref)
     db.commit()
     db.refresh(new_ref)
-    return {"ref": new_ref.ref}
+    return {"ref": new_ref.ref, "promocode": new_ref.promocode, "description": new_ref.description}
 
 @app.get("/ref/stat", tags=["agent-tg"], response_model=RefStatResponse)
 def get_ref_stat(current_tg_agent: TgAgent = Depends(get_current_tg_agent), db: Session = Depends(get_integration_db)):
@@ -306,32 +312,3 @@ def register(req: RegisterRequest, db: Session = Depends(get_integration_db)):
     db.commit()
     db.refresh(new_tg_agent)
     return {"msg": "TgAgent registered successfully"}
-
-@app.post("/promocode/add", tags=["agent-tg"], response_model=PromoCodeResponse)
-def add_promocode(req: PromoCodeAddRequest, current_tg_agent: TgAgent = Depends(get_current_tg_agent), db: Session = Depends(get_integration_db)):
-    """
-    Создает новый промокод для текущего Telegram-агента.
-    """
-    description = req.description
-    alphabet = string.ascii_letters + string.digits + "!@#$%^&*"
-    promocode = ''.join(choices(alphabet, k=8))
-    while db.exec(select(PromoCode).where(PromoCode.promocode == promocode)).first():
-        promocode = ''.join(choices(alphabet, k=8))
-    new_code = PromoCode(
-        promocode=promocode,
-        perc=10.0,
-        agent_id=current_tg_agent.id,
-        description=description
-    )
-    db.add(new_code)
-    db.commit()
-    db.refresh(new_code)
-    return PromoCodeResponse(promocode=new_code.promocode, perc=new_code.perc, description=new_code.description)
-
-@app.get("/promocode", tags=["agent-tg"], response_model=List[PromoCodeResponse])
-def get_promocodes(current_tg_agent: TgAgent = Depends(get_current_tg_agent), db: Session = Depends(get_integration_db)):
-    """
-    Получает все промокоды текущего Telegram-агента.
-    """
-    codes = db.exec(select(PromoCode).where(PromoCode.agent_id == current_tg_agent.id)).all()
-    return [PromoCodeResponse(promocode=c.promocode, perc=c.perc, description=c.description) for c in codes]