admin: improve add user validation

parent 030064b2
...@@ -7,7 +7,7 @@ from data.keyboards import admin_keyboards ...@@ -7,7 +7,7 @@ from data.keyboards import admin_keyboards
from database.func import DB from database.func import DB
from rules import BotAdmin from rules import BotAdmin
from config import tg_api, config from config import tg_api, config, DEFAUIL_BRANCHES
dp = Dispatch() dp = Dispatch()
dp.message.auto_rules = BotAdmin() dp.message.auto_rules = BotAdmin()
...@@ -71,14 +71,19 @@ async def alrtrepo_users( ...@@ -71,14 +71,19 @@ async def alrtrepo_users(
if not branch: if not branch:
branch = "sisyphus" branch = "sisyphus"
if branch not in DEFAUIL_BRANCHES:
await m.answer("Репозиторий не найден")
return
maintainer_obj = DB.maintainer.get(maintainer) maintainer_obj = DB.maintainer.get(maintainer.lower())
if not maintainer_obj: if not maintainer_obj:
await m.answer("Сопровождающий не найден") await m.answer("Сопровождающий не найден")
return return
if DB.user.add(user_id, maintainer_obj, branch): if DB.user.add(user_id, maintainer_obj, branch):
await m.answer("Пользователь добавлен") await m.answer("Пользователь добавлен")
else:
await m.answer("Пользователь уже существует")
@dp.message(Command("altrepo_remove_user", Argument("user_id"))) @dp.message(Command("altrepo_remove_user", Argument("user_id")))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment