основной текст в отдельном файле

This commit is contained in:
Vladless
2024-10-22 01:53:50 +03:00
parent ba3352aa46
commit 9829e7cd3e
15 changed files with 135 additions and 129 deletions
+1 -7
View File
@@ -126,7 +126,6 @@ async def update_balance(tg_id: int, amount: float):
WHERE tg_id = $2
''', amount, tg_id)
# Проверка и начисление реферального бонуса
await handle_referral_on_balance_update(tg_id, amount)
await conn.close()
@@ -157,7 +156,6 @@ async def add_referral(referred_tg_id: int, referrer_tg_id: int):
async def handle_referral_on_balance_update(tg_id: int, amount: float):
conn = await asyncpg.connect(DATABASE_URL)
# Проверяем, есть ли реферал в таблице
referral = await conn.fetchrow('''
SELECT referrer_tg_id FROM referrals WHERE referred_tg_id = $1
''', tg_id)
@@ -165,17 +163,13 @@ async def handle_referral_on_balance_update(tg_id: int, amount: float):
if referral:
referrer_tg_id = referral['referrer_tg_id']
# Начисляем бонус (25% от платежа) пригласившему
bonus = amount * 0.25 # 25% от платежа
bonus = amount * 0.25
# Проверяем, чтобы бонус не был отрицательным
if bonus < 0:
bonus = 0
# Обновляем баланс пригласившего
await update_balance(referrer_tg_id, bonus)
# Обновляем статус бонуса для пригласившего
await conn.execute('''
UPDATE referrals SET reward_issued = TRUE
WHERE referrer_tg_id = $1 AND referred_tg_id = $2