Merge pull request #122 from hteppl/fix-referrals

WIP: фикс уявзимости, где айди приглашенного реферала, совпадает с айди пригласившего
2024-12-16 23:41:29 +03:00
parent f5ec703994 18dde7e882
commit 725167f663
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(
            """