bivashy/Blitz-Proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix Add User

Browse Source
This commit is contained in:
Whispering Wind
2024-08-12 23:48:25 +03:30
committed by GitHub
parent 7ebaaaeb9a
commit f386da6f93
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
core/scripts/telegrambot/tbot.py
View File

@ -52,14 +52,19 @@ def process_add_user_step1(message):
command = f"python3 {CLI_PATH} list-users" command = f"python3 {CLI_PATH} list-users"
result = run_cli_command(command) result = run_cli_command(command)
try: try:
users = json.loads(result) users = json.loads(result)
if username in users: if username in users:
bot.reply_to(message, f"Username '{username}' already exists. Please choose a different username.") bot.reply_to(message, f"Username '{username}' already exists. Please choose a different username.")
return return
except json.JSONDecodeError: except json.JSONDecodeError:
bot.reply_to(message, "Error retrieving user list. Please try again later.") if "No such file or directory" in result or result.strip() == "":
return bot.reply_to(message, "User list file does not exist. Adding the first user.")
else:
bot.reply_to(message, "Error retrieving user list. Please try again later.")
return
msg = bot.reply_to(message, "Enter traffic limit (GB):") msg = bot.reply_to(message, "Enter traffic limit (GB):")
bot.register_next_step_handler(msg, process_add_user_step2, username) bot.register_next_step_handler(msg, process_add_user_step2, username)