Alexei
2 years ago
4 changed files with 186 additions and 65 deletions
@ -0,0 +1,112 @@
|
||||
# -*- coding: utf8 -*- |
||||
# Общественное достояние 2023, Алексей Безбородов (Alexei Bezborodov) <AlexeiBv+mirocod_platform_bot@narod.ru> |
||||
|
||||
# Добавление элемента в БД |
||||
|
||||
from bot_sys import user_access, bot_bd, log |
||||
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 |
||||
|
||||
|
||||
cancel_message = ''' |
||||
🚫 Добавление отменено |
||||
''' |
||||
|
||||
def StartAddBDItemTemplate(a_FSM, a_FSMStart, a_MessageFunc, a_TableName, a_KeyName, a_FieldName, a_Prefix, a_AccessFunc, a_ButtonFunc, access_mode = user_access.AccessMode.EDIT): |
||||
async def StartAddBDItem(a_CallbackQuery : types.CallbackQuery, state : FSMContext): |
||||
user_id = str(a_CallbackQuery.from_user.id) |
||||
user_groups = groups.GetUserGroupData(user_id) |
||||
await a_FSMStart.set() |
||||
key_item_id = None |
||||
if hasattr(a_CallbackQuery, 'data'): |
||||
key_item_id = str(a_CallbackQuery.data).replace(a_Prefix, '') |
||||
res_of_work_func = None |
||||
check = None |
||||
async with state.proxy() as item_data: |
||||
item_data['key_item_id'] = key_item_id |
||||
|
||||
if key_item_id: |
||||
check, res_of_work_func = await bd_item.CheckAccessBDItemTemplate(a_TableName, a_KeyName, key_item_id, a_MessageFunc, access_mode)(a_CallbackQuery) |
||||
else: |
||||
res_of_work_func = await a_MessageFunc(a_CallbackQuery) |
||||
|
||||
if not check is None: |
||||
await state.finish() |
||||
return check |
||||
if key_item_id: |
||||
await a_FSM.next() |
||||
return res_of_work_func |
||||
return simple_message.SimpleMessageTemplate(StartAddBDItem, a_ButtonFunc, a_AccessFunc, access_mode) |
||||
|
||||
def FinishAddBDItemTemplate(a_FSM, a_AddBDItemFunc, a_TableName, a_KeyName, a_FieldName, a_MessageFunc, a_AccessFunc, a_ButtonFunc, access_mode = user_access.AccessMode.ADD, field_type = bd_item.FieldType.text): |
||||
return FinishOrNextAddBDItemTemplate(a_FSM, a_AddBDItemFunc, a_TableName, a_KeyName, a_FieldName, a_MessageFunc, a_AccessFunc, a_ButtonFunc, True, access_mode = access_mode, field_type = field_type) |
||||
|
||||
def NextAddBDItemTemplate(a_FSM, a_AddBDItemFunc, a_TableName, a_KeyName, a_FieldName, a_MessageFunc, a_AccessFunc, a_ButtonFunc, access_mode = user_access.AccessMode.ADD, field_type = bd_item.FieldType.text): |
||||
return FinishOrNextAddBDItemTemplate(a_FSM, a_AddBDItemFunc, a_TableName, a_KeyName, a_FieldName, a_MessageFunc, a_AccessFunc, a_ButtonFunc, False, access_mode = access_mode, field_type = field_type) |
||||
|
||||
def FinishOrNextAddBDItemTemplate(a_FSM, a_AddBDItemFunc, a_TableName, a_KeyName, a_FieldName, a_MessageFunc, a_AccessFunc, a_ButtonFunc, a_Finish, access_mode = user_access.AccessMode.ADD, field_type = bd_item.FieldType.text): |
||||
async def FinishAddBDItem(a_Message : types.CallbackQuery, state : FSMContext): |
||||
state_func = None |
||||
if a_Finish: |
||||
state_func = state.finish |
||||
else: |
||||
state_func = a_FSM.next |
||||
user_id = str(a_Message.from_user.id) |
||||
user_groups = groups.GetUserGroupData(user_id) |
||||
error = None |
||||
res_of_work_func = None |
||||
check = None |
||||
async with state.proxy() as item_data: |
||||
if a_Message.text == bd_item.canсel_button_name: |
||||
print('canсel_button_name', bd_item.canсel_button_name) |
||||
await state.finish() |
||||
return simple_message.WorkFuncResult(cancel_message) |
||||
|
||||
key_item_id = item_data['key_item_id'] |
||||
if key_item_id: |
||||
check, res_of_work_func = await bd_item.CheckAccessBDItemTemplate(a_TableName, a_KeyName, key_item_id, a_MessageFunc, access_mode)(a_Message) |
||||
else: |
||||
print(a_MessageFunc) |
||||
res_of_work_func = await a_MessageFunc(a_Message) |
||||
|
||||
if not check is None: |
||||
await state_func() |
||||
return check |
||||
|
||||
field_value = '' |
||||
if field_type == bd_item.FieldType.photo: |
||||
if a_Message.text == bd_item.skip_button_name: |
||||
print('skip_button_name', bd_item.canсel_button_name) |
||||
field_value = '0' |
||||
else: |
||||
if a_Message.photo == None or len(a_Message.photo) == 0: |
||||
await state.finish() |
||||
return simple_message.WorkFuncResult(error_photo_type_message) |
||||
field_value = a_Message.photo[0].file_id |
||||
else: |
||||
field_value = a_Message.text |
||||
item_data[a_FieldName] = field_value |
||||
if a_Finish: |
||||
res, error = a_AddBDItemFunc(item_data) |
||||
await state_func() |
||||
if error: |
||||
return simple_message.WorkFuncResult(error) |
||||
return res_of_work_func |
||||
|
||||
return simple_message.SimpleMessageTemplate(FinishAddBDItem, a_ButtonFunc, a_AccessFunc, access_mode) |
||||
|
||||
def AddBDItem3RegisterHandlers(dp, a_FSM, a_FSMName, a_FSMDesc, a_FSMPhoto, a_ButtonName, a_AddBDItemFunc, a_AddNameMessageFunc, a_AddDescMessageFunc, a_AddPhotoMessageFunc, a_FinishMessageFunc, a_TableName : str, a_KeyName, a_NameField, a_DescField, a_PhotoField, a_GetButtonNameAndKeyValueAndAccessFunc, a_AccessFunc, a_ButtonFunc, access_mode = user_access.AccessMode.ADD): |
||||
keyboard_cancel = bd_item.GetCancelKeyboardButtonsTemplate(a_AccessFunc, access_mode) |
||||
keyboard_skip_and_cancel = bd_item.GetSkipAndCancelKeyboardButtonsTemplate(a_AccessFunc, access_mode) |
||||
a_Prefix = f'add_{a_TableName}_{a_KeyName}_{a_NameField}_{a_DescField}_{a_PhotoField}:' |
||||
# 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_message_handler(StartAddBDItemTemplate(a_FSM, a_FSMName, a_AddNameMessageFunc, a_TableName, a_KeyName, a_NameField, a_Prefix, a_AccessFunc, keyboard_cancel, access_mode), text = a_ButtonName) |
||||
dp.register_message_handler(NextAddBDItemTemplate(a_FSM, None, a_TableName, a_KeyName, a_NameField, a_AddDescMessageFunc, a_AccessFunc, keyboard_cancel, access_mode, field_type = bd_item.FieldType.text), state = a_FSMName) |
||||
dp.register_message_handler(NextAddBDItemTemplate(a_FSM, None, a_TableName, a_KeyName, a_DescField, a_AddPhotoMessageFunc, a_AccessFunc, keyboard_skip_and_cancel, access_mode, field_type = bd_item.FieldType.text), state = a_FSMDesc) |
||||
dp.register_message_handler(NextAddBDItemTemplate(a_FSM, a_AddBDItemFunc, a_TableName, a_KeyName, a_PhotoField, a_FinishMessageFunc, a_AccessFunc, a_ButtonFunc, access_mode, field_type = bd_item.FieldType.photo), content_types = ['photo'], state = a_FSMPhoto) |
||||
dp.register_message_handler(FinishAddBDItemTemplate(a_FSM, a_AddBDItemFunc, a_TableName, a_KeyName, a_PhotoField, a_FinishMessageFunc, a_AccessFunc, a_ButtonFunc, access_mode, field_type = bd_item.FieldType.photo), content_types = ['text'], state = a_FSMPhoto) |
||||
Loading…
Reference in new issue