feat(user-create): add random username button (#182)

Author: erfjab

app/keys/__init__.py

@@ -1,5 +1,5 @@
 from ._callbacks import PageCB, SelectCB
-from ._enums import Pages, Actions, YesOrNot, SelectAll, JsonHandler
+from ._enums import Pages, Actions, YesOrNot, SelectAll, JsonHandler, RandomHandler
 from .manager import _KeyboardsManager
 
 BotKeys = _KeyboardsManager()
@@ -13,4 +13,5 @@
     "YesOrNot",
     "SelectAll",
     "JsonHandler",
+    "RandomHandler",
 ]

app/keys/_enums.py

@@ -36,3 +36,7 @@ class SelectAll(str, Enum):
 
 class JsonHandler(str, Enum):
     USER = "user"
+
+
+class RandomHandler(str, Enum):
+    USERNAME = "username"

app/routers/users/create.py

@@ -1,13 +1,23 @@
 import asyncio
 import json
+from secrets import token_hex
 
 from aiogram import Router, F
 from aiogram.types import CallbackQuery, Message, BufferedInputFile
 from aiogram.filters import StateFilter
 from aiogram.fsm.state import State, StatesGroup
 from aiogram.fsm.context import FSMContext
 
-from app.keys import BotKeys, PageCB, Pages, Actions, SelectCB, SelectAll, JsonHandler
+from app.keys import (
+    BotKeys,
+    PageCB,
+    Pages,
+    Actions,
+    SelectCB,
+    SelectAll,
+    JsonHandler,
+    RandomHandler,
+)
 from app.db import crud
 from app.settings.language import MessageTexts
 from app.api import ClinetManager
@@ -75,10 +85,14 @@ async def adminselect(
     return await callback.message.edit_text(
         text=MessageTexts.ASK_REMARK,
         reply_markup=BotKeys.selector(
-            data=[(MessageTexts.CREATE_WITH_JSON, JsonHandler.USER)],
+            data=[
+                (MessageTexts.RANDOM_USERNAME, RandomHandler.USERNAME),
+                (MessageTexts.CREATE_WITH_JSON, JsonHandler.USER),
+            ],
             types=Pages.USERS,
             action=Actions.JSON,
             panel=callback_data.panel,
+            width=1,
         ),
     )
 
@@ -90,9 +104,15 @@ async def adminselect(
 async def json_start(
     callback: CallbackQuery, callback_data: SelectCB, state: FSMContext
 ):
-    await state.set_state(UserCreateForm.JSON)
+    if callback_data.select == JsonHandler.USER:
+        await state.set_state(UserCreateForm.JSON)
+        text = MessageTexts.ASK_JSON
+    else:
+        await state.update_data(username=str(token_hex(3)))
+        await state.set_state(UserCreateForm.USERCOUNT)
+        text = MessageTexts.ASK_COUNT
     return await callback.message.edit_text(
-        text=MessageTexts.ASK_JSON,
+        text=text,
         reply_markup=BotKeys.cancel(),
     )
 

app/settings/language/_message.py

@@ -44,6 +44,7 @@ class _MessageSettings(BaseSettings):
     ASK_DATE_LIMIT: str = "Enter DateLimit: [0-9]"
     ASK_CONFIGS: str = "Select Configs:"
     FAILED_USERNAME: str = "❌ Failed to create {username}."
+    RANDOM_USERNAME: str = "Random Username"
     USER_INFO: str = (
         "• <b>Username:</b> <code>{username}</code>\n"
         "• <b>Data Limit:</b> <code>{data_limit}</code>\n"