From 1ae19cd351aeda37588e382ea23d6db4f349b165 Mon Sep 17 00:00:00 2001 From: Capybara-z Date: Sun, 25 May 2025 23:35:29 +0300 Subject: [PATCH 1/2] 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 From e05f8f818354b4bce99ee08fd11043e5dadd1929 Mon Sep 17 00:00:00 2001 From: Capybara-z Date: Sun, 25 May 2025 23:38:31 +0300 Subject: [PATCH 2/2] add to accrual statistics MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - Добавлено логирование начислений купонов. --- handlers/coupons.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/handlers/coupons.py b/handlers/coupons.py index 17e1cd70..45c6ecad 100644 --- a/handlers/coupons.py +++ b/handlers/coupons.py @@ -22,6 +22,7 @@ from database import ( update_balance, update_coupon_usage_count, update_key_expiry, + add_payment, ) from handlers.buttons import MAIN_MENU from handlers.keys.key_utils import renew_key_in_cluster @@ -124,6 +125,12 @@ async def activate_coupon( await update_balance(session, user_id, coupon.amount) await update_coupon_usage_count(session, coupon.id) await create_coupon_usage(session, coupon.id, user_id) + await add_payment( + session, + tg_id=user_id, + amount=coupon.amount, + payment_system="coupon" + ) await message.answer( f"✅ Купон активирован, на баланс начислено {coupon.amount} рублей." )