Merge pull request #122 from hteppl/fix-referrals

WIP: фикс уявзимости, где айди приглашенного реферала, совпадает с айди пригласившего
2024-12-16 23:41:29 +03:00
parent c8f9ebc443 0a0f943517
commit 710770d39d
1 changed files with 5 additions and 0 deletions
@@ -577,6 +577,11 @@ async def get_all_users(conn):

 async def add_referral(referred_tg_id: int, referrer_tg_id: int, session: Any):
    try:
+        # Если айди приглашенного совпадает с айди пригласившего
+        if referred_tg_id == referrer_tg_id:
+            logger.warning(f"Пользователь {referred_tg_id} попытался использовать свою собственную реферальную ссылку.")
+            return
+
        # Вставка записи о реферале в базу данных
        await session.execute(
            """