Check Usernames
This commit is contained in:
15
core/tbot.py
15
core/tbot.py
@ -45,6 +45,21 @@ def add_user(message):
|
|||||||
|
|
||||||
def process_add_user_step1(message):
|
def process_add_user_step1(message):
|
||||||
username = message.text.strip()
|
username = message.text.strip()
|
||||||
|
if username == "":
|
||||||
|
bot.reply_to(message, "Username cannot be empty. Please enter a valid username.")
|
||||||
|
return
|
||||||
|
|
||||||
|
command = f"python3 {CLI_PATH} list-users"
|
||||||
|
result = run_cli_command(command)
|
||||||
|
try:
|
||||||
|
users = json.loads(result)
|
||||||
|
if username in users:
|
||||||
|
bot.reply_to(message, f"Username '{username}' already exists. Please choose a different username.")
|
||||||
|
return
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
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)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user