|
|
# -*- coding: utf8 -*- |
|
|
# Общественное достояние 2023, Алексей Безбородов (Alexei Bezborodov) <AlexeiBv+mirocod_platform_bot@narod.ru> |
|
|
|
|
|
# Группы пользователей |
|
|
|
|
|
from bot_sys import bot_bd, log, config, keyboard, user_access |
|
|
from bot_modules import start, access |
|
|
from aiogram import Bot, types |
|
|
from aiogram.dispatcher import FSMContext |
|
|
from aiogram.dispatcher.filters.state import State, StatesGroup |
|
|
|
|
|
import sqlite3 |
|
|
|
|
|
from aiogram.dispatcher import Dispatcher |
|
|
|
|
|
bot = Bot(token=config.GetTelegramBotApiToken(), parse_mode=types.ParseMode.HTML) |
|
|
|
|
|
class FSMRequestToBD(StatesGroup): |
|
|
sqlRequest = State() |
|
|
|
|
|
# --------------------------------------------------------- |
|
|
# БД |
|
|
module_name = 'groups' |
|
|
|
|
|
init_bd_cmds = ["""CREATE TABLE IF NOT EXISTS user_groups( |
|
|
group_id INTEGER PRIMARY KEY NOT NULL, |
|
|
groupName TEXT, |
|
|
UNIQUE(group_id) |
|
|
);""", |
|
|
"""CREATE TABLE IF NOT EXISTS user_in_groups( |
|
|
user_id INTEGER, |
|
|
group_id INTEGER, |
|
|
UNIQUE(user_id, group_id) |
|
|
);""", |
|
|
f"INSERT OR IGNORE INTO module_access (modName, modAccess) VALUES ('{module_name}', 'other=-');" |
|
|
] |
|
|
|
|
|
# --------------------------------------------------------- |
|
|
# Сообщения |
|
|
|
|
|
group_start_message = ''' |
|
|
<b>Группы пользователей находятся в стадии разработки</b> |
|
|
|
|
|
Пока можете воспользоваться хардкорным способом через запросы к БД |
|
|
''' |
|
|
|
|
|
request_start_message = ''' |
|
|
**Задайте запрос к БД** |
|
|
|
|
|
Можете воспользоваться следующими шаблонами: |
|
|
1. `SELECT * FROM users` - Все пользователи |
|
|
2. `SELECT group_id, groupName FROM user_groups` - Все группы пользоватлей |
|
|
3. `INSERT INTO user_groups(groupName) VALUES('GROUPNAME')` - добавление группы с именем GROUPNAME |
|
|
4. `SELECT group_id FROM user_in_groups WHERE user_id = USERID`- Все ID групп в которых состоит пользователь с USERID |
|
|
5. `SELECT groupName FROM user_groups WHERE group_id=(SELECT groupid FROM user_in_groups WHERE user_id = USERID)` - Все имена групп пользователя с USERID |
|
|
6. `INSERT INTO user_in_groups(user_id, group_id) VALUES(USERID, GROUPID)` - добавление пользователя USERID в группу с GROUPID |
|
|
''' |
|
|
|
|
|
help_message = ''' |
|
|
📄 Существует две БД для работы с группами |
|
|
`user_groups (group_id, groupName)` - содержит названия групп |
|
|
`user_in_groups(user_id, group_id)` - содержит соответсвия ID пользователей и групп |
|
|
''' |
|
|
|
|
|
request_cancel_message = ''' |
|
|
Запрос к БД отменён |
|
|
''' |
|
|
|
|
|
user_group_button_name = "️️▦ Группы пользователей" |
|
|
sql_request_button_name = "⛃ Запрос к БД для редактирования групп" |
|
|
help_button_name = "📄 Информация по группам" |
|
|
canсel_button_name = "🚫 Отменить" |
|
|
|
|
|
# --------------------------------------------------------- |
|
|
# Работа с кнопками |
|
|
|
|
|
def GetEditGroupKeyboardButtons(a_UserGroups): |
|
|
cur_buttons = [ |
|
|
keyboard.ButtonWithAccess(sql_request_button_name, user_access.AccessMode.EDIT, GetAccess()), |
|
|
keyboard.ButtonWithAccess(help_button_name, user_access.AccessMode.VIEW, GetAccess()) |
|
|
] |
|
|
mods = [start] |
|
|
return keyboard.MakeKeyboard(keyboard.GetButtons(mods) + cur_buttons, a_UserGroups) |
|
|
|
|
|
def GetCancelKeyboardButtons(a_UserGroups, a_AccessFunc): |
|
|
cur_buttons = [ |
|
|
keyboard.ButtonWithAccess(canсel_button_name, user_access.AccessMode.VIEW, a_AccessFunc()) |
|
|
] |
|
|
return keyboard.MakeKeyboard(cur_buttons, a_UserGroups) |
|
|
|
|
|
# --------------------------------------------------------- |
|
|
# Обработка сообщений |
|
|
|
|
|
# Приветствие |
|
|
async def GroupStart(a_Message): |
|
|
user_id = str(a_Message.from_user.id) |
|
|
user_groups = GetUserGroupData(user_id) |
|
|
if not user_access.CheckAccessString(GetAccess(), user_groups, user_access.AccessMode.VIEW): |
|
|
return await bot.send_message(user_id, access.access_denied_message, reply_markup = GetEditGroupKeyboardButtons(user_groups)) |
|
|
|
|
|
await bot.send_message(user_id, group_start_message, reply_markup = GetEditGroupKeyboardButtons(user_groups)) |
|
|
|
|
|
def RequestToBDCancelTemplate(a_GetButtonsFunc, a_AccessFunc): |
|
|
async def RequestToBDCancel(a_Message : types.message, state : FSMContext): |
|
|
user_id = str(a_Message.from_user.id) |
|
|
user_groups = GetUserGroupData(user_id) |
|
|
if not user_access.CheckAccessString(a_AccessFunc(), user_groups, user_access.AccessMode.VIEW): |
|
|
return await bot.send_message(user_id, access.access_denied_message, reply_markup = a_GetButtonsFunc(user_groups)) |
|
|
await state.finish() |
|
|
await a_Message.answer(request_cancel_message, reply_markup = a_GetButtonsFunc(user_groups)) |
|
|
return RequestToBDCancel |
|
|
|
|
|
def HelpTemplate(a_HelpMessage, a_GetButtonsFunc, a_AccessFunc): |
|
|
async def Help(a_Message : types.message): |
|
|
user_id = str(a_Message.from_user.id) |
|
|
user_groups = GetUserGroupData(user_id) |
|
|
if not user_access.CheckAccessString(a_AccessFunc(), user_groups, user_access.AccessMode.VIEW): |
|
|
return await bot.send_message(user_id, access.access_denied_message, reply_markup = a_GetButtonsFunc(user_groups)) |
|
|
await a_Message.answer(a_HelpMessage, reply_markup = a_GetButtonsFunc(user_groups)) #, parse_mode='Markdown') |
|
|
return Help |
|
|
|
|
|
def RequestToBDTemplate(a_StartMessage, a_AccessFunc, a_FSM): |
|
|
async def RequestToBDStart(a_Message : types.message): |
|
|
user_id = str(a_Message.from_user.id) |
|
|
user_groups = GetUserGroupData(user_id) |
|
|
if not user_access.CheckAccessString(a_AccessFunc(), user_groups, user_access.AccessMode.EDIT): |
|
|
return await bot.send_message(user_id, access.access_denied_message, reply_markup = GetCancelKeyboardButtons(user_groups, a_AccessFunc)) |
|
|
await a_FSM.sqlRequest.set() |
|
|
await a_Message.answer(a_StartMessage, reply_markup = GetCancelKeyboardButtons(user_groups, a_AccessFunc), parse_mode='Markdown') |
|
|
return RequestToBDStart |
|
|
|
|
|
def RequestToBDFinishTemplate(a_GetButtonsFunc, a_AccessFunc): |
|
|
async def RequestToBDFinish(a_Message : types.message, state : FSMContext): |
|
|
user_id = str(a_Message.from_user.id) |
|
|
user_groups = GetUserGroupData(user_id) |
|
|
if not user_access.CheckAccessString(a_AccessFunc(), user_groups, user_access.AccessMode.EDIT): |
|
|
return await bot.send_message(user_id, access.access_denied_message, reply_markup = GetEditGroupKeyboardButtons(user_groups)) |
|
|
result = '' |
|
|
async with state.proxy() as prjData: |
|
|
sql_request = a_Message.text |
|
|
log.Success(f'Сделан запрос [{sql_request}] пользователем {a_Message.from_user.id}.') |
|
|
result = bot_bd.SQLRequestToBDCommit(sql_request) |
|
|
log.Success(f'Результат запроса [{sql_request}] от пользователя {a_Message.from_user.id} следующий [{result}].') |
|
|
await state.finish() |
|
|
await a_Message.answer(str(result), reply_markup = a_GetButtonsFunc(user_groups)) |
|
|
return RequestToBDFinish |
|
|
|
|
|
# --------------------------------------------------------- |
|
|
# Работа с базой данных групп |
|
|
|
|
|
def GetGroupIDForUser(a_UserID): |
|
|
return bot_bd.SQLRequestToBD1('SELECT group_id FROM user_in_groups WHERE user_id = ?', [a_UserID]) |
|
|
|
|
|
def GetGroupNamesForUser(a_UserID): |
|
|
return bot_bd.SQLRequestToBD1('SELECT groupName FROM user_groups WHERE group_id=(SELECT group_id FROM user_in_groups WHERE user_id = ?)', [a_UserID]) |
|
|
|
|
|
def GetUserGroupData(a_UserID): |
|
|
r = GetGroupNamesForUser(a_UserID) |
|
|
groups = [] |
|
|
for i in r: |
|
|
groups += [i[0]] |
|
|
return user_access.UserGroups(a_UserID, groups) |
|
|
|
|
|
# --------------------------------------------------------- |
|
|
# API |
|
|
|
|
|
# Инициализация БД |
|
|
def GetInitBDCommands(): |
|
|
return init_bd_cmds |
|
|
|
|
|
def GetAccess(): |
|
|
return access.GetAccessForModule(module_name) |
|
|
|
|
|
# Доступные кнопки |
|
|
def GetModuleButtons(): |
|
|
return [keyboard.ButtonWithAccess(user_group_button_name, user_access.AccessMode.VIEW, GetAccess())] |
|
|
|
|
|
# Обработка кнопок |
|
|
def RegisterHandlers(dp : Dispatcher): |
|
|
dp.register_message_handler(GroupStart, text = user_group_button_name) |
|
|
dp.register_message_handler(HelpTemplate(help_message, GetEditGroupKeyboardButtons, GetAccess), text = help_button_name) |
|
|
|
|
|
dp.register_message_handler(RequestToBDTemplate(request_start_message, GetAccess, FSMRequestToBD), text = sql_request_button_name) |
|
|
dp.register_message_handler(RequestToBDCancelTemplate(GetEditGroupKeyboardButtons, GetAccess), text = canсel_button_name, state = FSMRequestToBD.sqlRequest) |
|
|
dp.register_message_handler(RequestToBDFinishTemplate(GetEditGroupKeyboardButtons, GetAccess), state = FSMRequestToBD.sqlRequest)
|
|
|
|
