From 0a0f9435179df20bdfbbbffd18c0b88f85188f9a Mon Sep 17 00:00:00 2001
From: hteppl <hteppl.dev@gmail.com>
Date: Mon, 16 Dec 2024 23:19:39 +0300
Subject: [PATCH] Fix self-referred bug

---
 database.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/database.py b/database.py
index d689070f..4a12915c 100644
--- a/database.py
+++ b/database.py
@@ -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(
             """