Feat: fixed start command

docker-compose.yml

@@ -22,7 +22,7 @@ services:
       POSTGRES_USER: ${POSTGRES_USER:-postgres}
       POSTGRES_DB: ${POSTGRES_DB:-postgres}
     ports:
-      - 5432
+      - "127.0.0.1:${POSTGRES_PORT}:5432"
     healthcheck:
       test: ["CMD", "pg_isready"]
       interval: 10s

nwxraybot/handlers/admin.py

@@ -19,7 +19,7 @@ class AdminHandler(Handler):
 
         @self.router.message(Command('adduser'))
         async def add_user(message: Message):
-            mask = r"^(?P<name>[a-zA-Z0-9]+)\s(?P<url>[^\s]+)($|\s(?P<date>[0-9]{2}\.[0-9]{2}\.[0-9]{4})\s(?P<time>[0-9]{2}\:[0-9]{2})$)"
+            mask = r"^(?P<name>[a-zA-Z0-9]+)\s(?P<url>vless://[^\s]+)($|\s(?P<date>[0-9]{2}\.[0-9]{2}\.[0-9]{4})\s(?P<time>[0-9]{2}\:[0-9]{2})$)"
             text = message.text.replace('/adduser ', '')
             match = re.match(mask, text)
             if match is None:

nwxraybot/handlers/hello.py

@@ -27,18 +27,24 @@ class HelloHandler(Handler):
         @self.router.message(Command("start"))
         async def hello(message: types.Message):
             data = message.text.split()
+            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: Optional[User] = User.select().where(
+                user = User.select().where(
                     User.code == code).first()
                 if user is None:
                     await message.answer('Пользователь не найден, обратитесь к администратору за ссылкой!')
                     return
                 user.telegram_id = message.from_user.id
+                user.code = ''
                 user.save()
+            user = User.select().where(User.telegram_id == message.from_user.id).first()
+            if user is None:
+                await message.answer('Пользователь не найден, обратитесь к администратору за ссылкой!')
+                return
             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)