Alexei
2 years ago
6 changed files with 268 additions and 29 deletions
@ -0,0 +1,49 @@ |
|||||||
|
# -*- coding: utf8 -*- |
||||||
|
# Общественное достояние 2023, Алексей Безбородов (Alexei Bezborodov) <AlexeiBv+mirocod_platform_bot@narod.ru> |
||||||
|
|
||||||
|
# Профиль пользователя |
||||||
|
|
||||||
|
from bot_sys import bot_bd, log, config, keyboard |
||||||
|
from bot_modules import start |
||||||
|
from aiogram import Bot, types |
||||||
|
|
||||||
|
import sqlite3 |
||||||
|
|
||||||
|
from aiogram.dispatcher import Dispatcher |
||||||
|
|
||||||
|
bot = Bot(token=config.GetTelegramBotApiToken(), parse_mode=types.ParseMode.HTML) |
||||||
|
|
||||||
|
# --------------------------------------------------------- |
||||||
|
# БД |
||||||
|
init_bd_cmds = [] |
||||||
|
|
||||||
|
# --------------------------------------------------------- |
||||||
|
# Сообщения |
||||||
|
|
||||||
|
# --------------------------------------------------------- |
||||||
|
# Работа с кнопками |
||||||
|
|
||||||
|
# --------------------------------------------------------- |
||||||
|
# Обработка сообщений |
||||||
|
|
||||||
|
# --------------------------------------------------------- |
||||||
|
# Работа с базой данных |
||||||
|
|
||||||
|
|
||||||
|
# --------------------------------------------------------- |
||||||
|
# API |
||||||
|
|
||||||
|
def GetUserAccess(a_UserID): |
||||||
|
return None |
||||||
|
|
||||||
|
# Инициализация БД |
||||||
|
def GetInitBDCommands(): |
||||||
|
return init_bd_cmds |
||||||
|
|
||||||
|
# Доступные кнопки |
||||||
|
def GetButtonNames(a_UserAccess): |
||||||
|
return [] |
||||||
|
|
||||||
|
# Обработка кнопок |
||||||
|
def RegisterHandlers(dp : Dispatcher): |
||||||
|
return |
||||||
@ -0,0 +1,193 @@ |
|||||||
|
# -*- coding: utf8 -*- |
||||||
|
# Общественное достояние 2023, Алексей Безбородов (Alexei Bezborodov) <AlexeiBv+mirocod_platform_bot@narod.ru> |
||||||
|
|
||||||
|
# Группы пользователей |
||||||
|
|
||||||
|
from bot_sys import bot_bd, log, config, keyboard |
||||||
|
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() |
||||||
|
|
||||||
|
# --------------------------------------------------------- |
||||||
|
# БД |
||||||
|
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) |
||||||
|
);""", |
||||||
|
] |
||||||
|
|
||||||
|
# --------------------------------------------------------- |
||||||
|
# Сообщения |
||||||
|
|
||||||
|
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 |
||||||
|
''' |
||||||
|
|
||||||
|
request_cancel_message = ''' |
||||||
|
Запрос к БД отменён |
||||||
|
''' |
||||||
|
|
||||||
|
user_group_button_name = "📰 Группы пользователей" |
||||||
|
sql_request_button_name = "📰 Запрос к БД" |
||||||
|
canсel_button_name = "📰 Отменить" |
||||||
|
|
||||||
|
# --------------------------------------------------------- |
||||||
|
# Работа с кнопками |
||||||
|
|
||||||
|
def GetEditGroupKeyboardButtons(a_UserAccess): |
||||||
|
cur_buttons = [sql_request_button_name] |
||||||
|
mods = [start] |
||||||
|
return keyboard.MakeKeyboard(keyboard.GetButtons(mods, a_UserAccess) + cur_buttons) |
||||||
|
|
||||||
|
def GetCancelKeyboardButtons(a_UserAccess): |
||||||
|
return keyboard.MakeKeyboard([canсel_button_name]) |
||||||
|
|
||||||
|
# --------------------------------------------------------- |
||||||
|
# Обработка сообщений |
||||||
|
|
||||||
|
# Приветствие |
||||||
|
async def GroupStart(a_Message): |
||||||
|
user_id = str(a_Message.from_user.id) |
||||||
|
user_access = access.GetUserAccess(a_Message.from_user.id) |
||||||
|
await bot.send_message(user_id, group_start_message, reply_markup = GetEditGroupKeyboardButtons(user_access)) |
||||||
|
|
||||||
|
async def RequestToBDCancel(a_Message : types.message, state : FSMContext): |
||||||
|
user_access = access.GetUserAccess(a_Message.from_user.id) |
||||||
|
await state.finish() |
||||||
|
await a_Message.answer(request_cancel_message, reply_markup = GetEditGroupKeyboardButtons(user_access)) |
||||||
|
|
||||||
|
async def RequestToBDStart(a_Message : types.message): |
||||||
|
user_access = access.GetUserAccess(a_Message.from_user.id) |
||||||
|
await FSMRequestToBD.sqlRequest.set() |
||||||
|
await a_Message.answer(request_start_message, reply_markup = GetCancelKeyboardButtons(user_access), parse_mode='Markdown') |
||||||
|
|
||||||
|
async def RequestToBD(a_Message : types.message, state : FSMContext): |
||||||
|
user_access = access.GetUserAccess(a_Message.from_user.id) |
||||||
|
result = '' |
||||||
|
async with state.proxy() as prjData: |
||||||
|
sql_request = a_Message.text |
||||||
|
log.Success(f'Сделан запрос [{sql_request}] пользователем {a_Message.from_user.id}.') |
||||||
|
result = SQLRequestToBD(sql_request) |
||||||
|
log.Success(f'Результат запроса [{sql_request}] от пользователя {a_Message.from_user.id} следующий [{result}].') |
||||||
|
await state.finish() |
||||||
|
await a_Message.answer(str(result), reply_markup = GetEditGroupKeyboardButtons(user_access)) |
||||||
|
|
||||||
|
# --------------------------------------------------------- |
||||||
|
# Работа с базой данных групп |
||||||
|
|
||||||
|
def SQLRequestToBD(a_Request : str): |
||||||
|
db = sqlite3.connect(bot_bd.GetBDFileName()) |
||||||
|
cursor = db.cursor() |
||||||
|
cursor.execute(a_Request) |
||||||
|
result = cursor.fetchall() |
||||||
|
db.commit() |
||||||
|
cursor.close() |
||||||
|
db.close() |
||||||
|
return result |
||||||
|
|
||||||
|
def GetGroupIDForUser(a_UserID): |
||||||
|
db = sqlite3.connect(bot_bd.GetBDFileName()) |
||||||
|
cursor = db.cursor() |
||||||
|
groups = cursor.execute('SELECT group_id FROM user_in_groups WHERE user_id = ?', (a_UserID)).fetchall() |
||||||
|
cursor.close() |
||||||
|
db.close() |
||||||
|
return groups |
||||||
|
|
||||||
|
def GetGroupNamesForUser(a_UserID): |
||||||
|
db = sqlite3.connect(bot_bd.GetBDFileName()) |
||||||
|
cursor = db.cursor() |
||||||
|
groups = cursor.execute('SELECT groupName FROM user_groups WHERE group_id=(SELECT group_id FROM user_in_groups WHERE user_id = ?)', (a_UserID)).fetchall() |
||||||
|
cursor.close() |
||||||
|
db.close() |
||||||
|
print(groups) |
||||||
|
return groups |
||||||
|
|
||||||
|
def GetGroupList(): |
||||||
|
db = sqlite3.connect(bot_bd.GetBDFileName()) |
||||||
|
cursor = db.cursor() |
||||||
|
groups = cursor.execute('SELECT * FROM user_groups').fetchall() |
||||||
|
cursor.close() |
||||||
|
db.close() |
||||||
|
return groups |
||||||
|
|
||||||
|
def GetGroup(a_GroupID): |
||||||
|
db = sqlite3.connect(bot_bd.GetBDFileName()) |
||||||
|
cursor = db.cursor() |
||||||
|
group = cursor.execute('SELECT * FROM user_groups WHERE groupID = ?', ([a_GroupID])).fetchall() |
||||||
|
cursor.close() |
||||||
|
db.close() |
||||||
|
return group |
||||||
|
|
||||||
|
def AddGroup(a_Name): |
||||||
|
db = sqlite3.connect(bot_bd.GetBDFileName()) |
||||||
|
cursor = db.cursor() |
||||||
|
cursor.execute('INSERT INTO user_groups(groupName) VALUES(?)', (a_prjName)) |
||||||
|
db.commit() |
||||||
|
cursor.close() |
||||||
|
db.close() |
||||||
|
return |
||||||
|
|
||||||
|
def EditGroup(a_GroupID, a_Name): |
||||||
|
db = sqlite3.connect(bot_bd.GetBDFileName()) |
||||||
|
cursor = db.cursor() |
||||||
|
cursor.execute('UPDATE user_groups SET groupName = ? WHERE groupID = ?', (a_prjPhoto, a_Name)) |
||||||
|
db.commit() |
||||||
|
cursor.close() |
||||||
|
db.close() |
||||||
|
return |
||||||
|
|
||||||
|
def DelGroup(a_GroupID): |
||||||
|
db = sqlite3.connect(bot_bd.GetBDFileName()) |
||||||
|
cursor = db.cursor() |
||||||
|
cursor.execute('DELETE FROM user_groups WHERE groupID = ?', ([a_GroupID])) |
||||||
|
db.commit() |
||||||
|
db.close() |
||||||
|
return |
||||||
|
|
||||||
|
# --------------------------------------------------------- |
||||||
|
# API |
||||||
|
|
||||||
|
# Инициализация БД |
||||||
|
def GetInitBDCommands(): |
||||||
|
return init_bd_cmds |
||||||
|
|
||||||
|
# Доступные кнопки |
||||||
|
def GetButtonNames(a_UserAccess): |
||||||
|
return [user_group_button_name] |
||||||
|
|
||||||
|
# Обработка кнопок |
||||||
|
def RegisterHandlers(dp : Dispatcher): |
||||||
|
dp.register_message_handler(GroupStart, text = user_group_button_name) |
||||||
|
dp.register_message_handler(RequestToBDStart, text = sql_request_button_name) |
||||||
|
dp.register_message_handler(RequestToBDCancel, text = canсel_button_name, state = FSMRequestToBD.sqlRequest) |
||||||
|
dp.register_message_handler(RequestToBD, state = FSMRequestToBD.sqlRequest) |
||||||
Loading…
Reference in new issue