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