From e82c392fc8db0ceb386943671e3bdce9c6c7f577 Mon Sep 17 00:00:00 2001
From: Sergey Elpashev <elpashevs@bk.ru>
Date: Fri, 17 Jan 2025 13:51:57 +0300
Subject: [PATCH] Hotfix: user deleting tg_id

---
 nwxraybot/handlers/hello.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/nwxraybot/handlers/hello.py b/nwxraybot/handlers/hello.py
index 4e41b0b..8acdc72 100644
--- a/nwxraybot/handlers/hello.py
+++ b/nwxraybot/handlers/hello.py
@@ -30,14 +30,20 @@ class HelloHandler(Handler):
             user: Optional[User] = None
             if len(data) == 2:
                 code = data[1]
-                query = User.update(telegram_id=None).where(
-                    User.telegram_id == message.from_user.id)
-                query.execute()
+                user = User.select().where(
+                    User.telegram_id == message.from_user.id).first()
+                if user is not None:
+                    await message.answer(f"Приветствуем в боте NwXray! Здесь вы сможете получить информацию о своем подключении к NwXray.\n\n{get_subscription_info(message.from_user.id)}",
+                                         reply_markup=self.__non_admin_main_menu(), parse_mode=ParseMode.MARKDOWN)
+                    return
                 user = User.select().where(
                     User.code == code).first()
                 if user is None:
                     await message.answer('Пользователь не найден, обратитесь к администратору за ссылкой!')
                     return
+                query = User.update(telegram_id=None).where(
+                    User.telegram_id == message.from_user.id)
+                query.execute()
                 user.telegram_id = message.from_user.id
                 user.code = ''
                 user.save()