Alexei
2 years ago
9 changed files with 232 additions and 128 deletions
@ -0,0 +1,61 @@ |
|||||||
|
# -*- coding: utf8 -*- |
||||||
|
# Общественное достояние 2023, Алексей Безбородов (Alexei Bezborodov) <AlexeiBv+mirocod_platform_bot@narod.ru> |
||||||
|
|
||||||
|
# Работа с элементом в БД |
||||||
|
|
||||||
|
from bot_sys import user_access, bot_bd |
||||||
|
from bot_modules import groups |
||||||
|
|
||||||
|
from aiogram import types |
||||||
|
|
||||||
|
''' |
||||||
|
class TableListParam(): |
||||||
|
def __init__(self, a_TableName : str, a_KeyName : str, a_GetButtonNameAndKeyValueAndAccessFunc): |
||||||
|
self.table_name = a_TableName |
||||||
|
self.key_name = a_KeyName |
||||||
|
self.get_bname_and_key_value_func = a_GetButtonNameAndKeyValueAndAccessFunc |
||||||
|
''' |
||||||
|
|
||||||
|
item_not_found = 'Элемент {item_id} не найден в таблице {a_TableName}' |
||||||
|
|
||||||
|
def GetAllItemsTemplate(a_TableName): |
||||||
|
def GetAllItems(): |
||||||
|
return bot_bd.SelectBDTemplate(a_TableName)() |
||||||
|
return GetAllItems |
||||||
|
|
||||||
|
def GetBDItemsTemplate(a_TableName : str, a_KeyName : str): |
||||||
|
def GetBDItem(a_KeyValue): |
||||||
|
return bot_bd.SQLRequestToBD(f'SELECT * FROM {a_TableName} WHERE {a_KeyName} = ?', param = ([a_KeyValue])) |
||||||
|
return GetBDItem |
||||||
|
|
||||||
|
def DeleteBDItemInTableTemplate(a_TableName : str, a_KeyName : str): |
||||||
|
def DeleteBDItem(a_KeyValue): |
||||||
|
return bot_bd.SQLRequestToBD(f'DELETE FROM {a_TableName} WHERE {a_KeyName} = ?', commit = True, return_error = True, param = ([a_KeyValue])) |
||||||
|
return DeleteBDItem |
||||||
|
|
||||||
|
def EditBDItemInTableTemplate(a_TableName : str, a_KeyName : str, a_FieldName : str): |
||||||
|
def EditBDItemInTable(a_KeyValue, a_FieldValue): |
||||||
|
return bot_bd.SQLRequestToBD(f'UPDATE {a_TableName} SET {a_FieldName}=? WHERE {a_KeyName} = ?', commit = True, return_error = True, param = ([a_FieldValue, a_KeyValue])) |
||||||
|
return EditBDItemInTable |
||||||
|
|
||||||
|
def CheckAccessBDItemTemplate(a_TableName, a_KeyName, a_KeyValue, a_WorkFunc, a_AccessMode : user_access.AccessMode): |
||||||
|
async def CheckAccessBDItem(a_CallbackQuery : types.CallbackQuery, a_ResultWorkFunc): |
||||||
|
user_id = str(a_CallbackQuery.from_user.id) |
||||||
|
user_groups = groups.GetUserGroupData(user_id) |
||||||
|
item_id = a_KeyValue |
||||||
|
item = GetBDItemsTemplate(a_TableName, a_KeyName)(item_id) |
||||||
|
if len(item) < 1: |
||||||
|
msg = item_not_found.replace('{item_id}', str(item_id)).replace('{a_TableName}', a_TableName) |
||||||
|
log.Error(msg) |
||||||
|
return simple_message.WorkFuncResult(msg) |
||||||
|
|
||||||
|
a_ResultWorkFunc = await a_WorkFunc(a_CallbackQuery, item[0]) |
||||||
|
if a_ResultWorkFunc is None or a_ResultWorkFunc.string_message is None: |
||||||
|
print('a_ResultWorkFunc', a_ResultWorkFunc) |
||||||
|
return a_ResultWorkFunc |
||||||
|
|
||||||
|
if not a_ResultWorkFunc.item_access is None and not user_access.CheckAccessString(a_ResultWorkFunc.item_access, user_groups, a_AccessMode): |
||||||
|
return simple_message.WorkFuncResult(access.access_denied_message) |
||||||
|
|
||||||
|
return None |
||||||
|
return CheckAccessBDItem |
||||||
@ -0,0 +1,94 @@ |
|||||||
|
# -*- coding: utf8 -*- |
||||||
|
# Общественное достояние 2023, Алексей Безбородов (Alexei Bezborodov) <AlexeiBv+mirocod_platform_bot@narod.ru> |
||||||
|
|
||||||
|
# Редактирование элемента в БД |
||||||
|
|
||||||
|
from enum import Enum |
||||||
|
|
||||||
|
from bot_sys import user_access, bot_bd, keyboard |
||||||
|
from bot_modules import access, groups |
||||||
|
from template import simple_message, bd_item_select, bd_item |
||||||
|
|
||||||
|
from aiogram import types |
||||||
|
from aiogram.dispatcher import FSMContext |
||||||
|
from aiogram.dispatcher.filters.state import State, StatesGroup |
||||||
|
|
||||||
|
canсel_button_name = "🚫 Отменить" |
||||||
|
|
||||||
|
cancel_message = ''' |
||||||
|
🚫 Редактирование отменено |
||||||
|
''' |
||||||
|
|
||||||
|
error_photo_type_message = ''' |
||||||
|
🚫 Фотографий не найдено |
||||||
|
''' |
||||||
|
|
||||||
|
class FieldType(Enum): |
||||||
|
text = 0 |
||||||
|
photo = 1 |
||||||
|
|
||||||
|
def StartEditBDItemTemplate(a_FSM, a_MessageFunc, a_TableName, a_KeyName, a_FieldName, a_Prefix, a_AccessFunc, a_ButtonFunc, access_mode = user_access.AccessMode.EDIT): |
||||||
|
async def StartEditBDItem(a_CallbackQuery : types.CallbackQuery, state : FSMContext): |
||||||
|
user_id = str(a_CallbackQuery.from_user.id) |
||||||
|
user_groups = groups.GetUserGroupData(user_id) |
||||||
|
await a_FSM.item_id.set() |
||||||
|
item_id = str(a_CallbackQuery.data).replace(a_Prefix, '') |
||||||
|
res_of_work_func = None |
||||||
|
async with state.proxy() as item_data: |
||||||
|
item_data['item_id'] = item_id |
||||||
|
|
||||||
|
check_func = bd_item.CheckAccessBDItemTemplate(a_TableName, a_KeyName, item_id, a_MessageFunc, access_mode) |
||||||
|
print('check_func', check_func) |
||||||
|
print('res_of_work_func', res_of_work_func) |
||||||
|
check = await check_func(a_CallbackQuery, res_of_work_func) |
||||||
|
print('res_of_work_func', res_of_work_func) |
||||||
|
print('check', check) |
||||||
|
if not check is None: |
||||||
|
return check |
||||||
|
await a_FSM.next() |
||||||
|
print('res_of_work_func', res_of_work_func) |
||||||
|
return res_of_work_func |
||||||
|
|
||||||
|
return simple_message.SimpleMessageTemplate(StartEditBDItem, a_ButtonFunc, a_AccessFunc, access_mode) |
||||||
|
|
||||||
|
def FinishEditBDItemTemplate(a_FSM, a_TableName, a_KeyName, a_FieldName, a_MessageFunc, a_Prefix, a_AccessFunc, a_ButtonFunc, access_mode = user_access.AccessMode.EDIT, field_type = FieldType.text): |
||||||
|
async def FinishEditBDItem(a_CallbackQuery : types.CallbackQuery, state : FSMContext): |
||||||
|
user_id = str(a_Message.from_user.id) |
||||||
|
user_groups = groups.GetUserGroupData(user_id) |
||||||
|
error = None |
||||||
|
res_of_work_func = None |
||||||
|
async with state.proxy() as item_data: |
||||||
|
if a_Message.text == canсel_button_name: |
||||||
|
return simple_message.WorkFuncResult(cancel_message) |
||||||
|
|
||||||
|
item_id = item_data['item_id'] |
||||||
|
check = await bd_item.CheckAccessBDItemTemplate(a_TableName, a_KeyName, item_id, a_MessageFunc, access_mode)(a_CallbackQuery, res_of_work_func) |
||||||
|
|
||||||
|
if not check is None: |
||||||
|
return check |
||||||
|
|
||||||
|
field_value = '' |
||||||
|
if field_type == FieldType.photo: |
||||||
|
if a_Message.photo == None or len(a_Message.photo) == 0: |
||||||
|
return simple_message.WorkFuncResult(error_photo_type_message) |
||||||
|
field_value = a_Message.photo[0].file_id |
||||||
|
else: |
||||||
|
field_value = a_Message.text |
||||||
|
res, error = bot_item.EditBDItemInTableTemplate(a_TableName, a_KeyName, a_FieldName)(item_id, field_value) |
||||||
|
log.Success(f'Изменено поле в таблице {a_TableName} ключу {a_KeyName}={item_id}. Новое значение поля {a_FieldName}={field_value}. Пользователь {user_id}.') |
||||||
|
await state.finish() |
||||||
|
if error: |
||||||
|
return simple_message.WorkFuncResult(error) |
||||||
|
return res_of_work_func |
||||||
|
|
||||||
|
return simple_message.SimpleMessageTemplate(FinishEditBDItem, a_ButtonFunc, a_AccessFunc, access_mode) |
||||||
|
|
||||||
|
def EditBDItemRegisterHandlers(dp, a_ButtonName, a_StartMessage, a_EditMessageFunc, a_FinishMessageFunc, a_TableName : str, a_KeyName, a_FieldName, a_GetButtonNameAndKeyValueAndAccessFunc, a_AccessFunc, a_ButtonFunc, access_mode = user_access.AccessMode.EDIT, field_type = FieldType.text): |
||||||
|
class FSMEditItem(StatesGroup): |
||||||
|
item_id = State() |
||||||
|
item_field = State() |
||||||
|
a_Prefix = f'edit_{a_TableName}_{a_KeyName}_{a_FieldName}:' |
||||||
|
sel_handler = bd_item_select.SelectDBItemTemplate(a_TableName, a_GetButtonNameAndKeyValueAndAccessFunc, a_StartMessage, a_AccessFunc, a_Prefix, access_mode) |
||||||
|
dp.register_message_handler(sel_handler, text = a_ButtonName) |
||||||
|
dp.register_callback_query_handler(StartEditBDItemTemplate(FSMEditItem, a_EditMessageFunc, a_TableName, a_KeyName, a_FieldName, a_Prefix, a_AccessFunc, a_ButtonFunc, access_mode), lambda x: x.data.startswith(a_Prefix)) |
||||||
|
dp.register_message_handler(FinishEditBDItemTemplate(FSMEditItem, a_TableName, a_KeyName, a_FieldName, a_FinishMessageFunc, a_Prefix, a_AccessFunc, a_ButtonFunc, access_mode, field_type = field_type), state = FSMEditItem.item_field) |
||||||
Loading…
Reference in new issue