From 1ae19cd351aeda37588e382ea23d6db4f349b165 Mon Sep 17 00:00:00 2001
From: Capybara-z <stas89091000@yandex.ru>
Date: Sun, 25 May 2025 23:35:29 +0300
Subject: [PATCH] add to accrual statistics
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Добавлено логирование начислений реферавлов.
---
 database/referrals.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/database/referrals.py b/database/referrals.py
index 3867e15c..1da621b7 100644
--- a/database/referrals.py
+++ b/database/referrals.py
@@ -5,6 +5,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
 from config import CHECK_REFERRAL_REWARD_ISSUED, REFERRAL_BONUS_PERCENTAGES
 from database.models import Referral
 from logger import logger
+from database import add_payment
 
 
 async def add_referral(session: AsyncSession, referred_tg_id: int, referrer_tg_id: int):
@@ -173,6 +174,14 @@ async def get_total_referral_bonus(
     )
     total_bonus_raw = result.scalar()
     total_bonus = round(float(total_bonus_raw or 0), 2)
+    if total_bonus > 0:
+        await add_payment(
+            session,
+            tg_id=referrer_tg_id,
+            amount=total_bonus,
+            payment_system="referral"
+        )
+
     logger.debug(f"Получена общая сумма бонусов от рефералов: {total_bonus}")
     return total_bonus