You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
8.7 KiB
124 lines
8.7 KiB
# -*- coding: utf8 -*- |
|
# Общественное достояние, 2023, Алексей Безбородов (Alexei Bezborodov) <AlexeiBv+mirocod_platform_bot@narod.ru> |
|
|
|
# Добавление элемента в БД |
|
|
|
from bot_sys import user_access, bot_bd, bot_messages |
|
from bot_modules import access_utils, groups_utils |
|
from template import simple_message, bd_item_select, bd_item |
|
|
|
cancel_message = ''' |
|
🚫 Добавление отменено |
|
''' |
|
error_photo_type_message = ''' |
|
🚫 Неверный формат изображений |
|
''' |
|
|
|
def StartAddBDItemTemplate(a_Bot, a_FSM, a_FSMStart, a_MessageFunc, a_ParentTableName, a_ParentKeyFieldName, a_Prefix, a_AccessFunc, a_ButtonFunc, a_FinishButtonFunc, access_mode = user_access.AccessMode.ADD): |
|
async def StartAddBDItem(a_CallbackQuery, state): |
|
user_id = str(a_CallbackQuery.from_user.id) |
|
user_groups = groups_utils.GetUserGroupData(a_Bot, user_id) |
|
parent_id = bd_item.GetKeyDataFromCallbackMessage(a_CallbackQuery, a_Prefix) |
|
res_of_work_func = None |
|
check = None |
|
|
|
await a_FSMStart.set() |
|
async with state.proxy() as item_data: |
|
if a_ParentKeyFieldName: |
|
item_data[a_ParentKeyFieldName] = parent_id |
|
|
|
if parent_id: |
|
check, res_of_work_func = await bd_item.CheckAccessBDItemTemplate(a_Bot, a_ParentTableName, a_ParentKeyFieldName, parent_id, a_MessageFunc, access_mode)(a_CallbackQuery) |
|
else: |
|
res_of_work_func = await a_MessageFunc(a_CallbackQuery, None, item_data) |
|
|
|
if not check is None: |
|
await state.finish() |
|
check.keyboard_func = a_FinishButtonFunc |
|
return check |
|
return res_of_work_func |
|
return simple_message.SimpleMessageTemplate(a_Bot, StartAddBDItem, a_ButtonFunc, None, a_AccessFunc, access_mode) |
|
|
|
def FinishAddBDItemTemplate(a_Bot, a_FSM, a_AddBDItemFunc, a_ParentTableName, a_ParentKeyFieldName, a_FieldName, a_MessageFunc, a_PostProcessFunc, a_AccessFunc, a_ButtonFunc, access_mode = user_access.AccessMode.ADD, field_type = bd_item.FieldType.text): |
|
return FinishOrNextAddBDItemTemplate(a_Bot, a_FSM, a_AddBDItemFunc, a_ParentTableName, a_ParentKeyFieldName, a_FieldName, a_MessageFunc, a_PostProcessFunc, a_AccessFunc, a_ButtonFunc, a_ButtonFunc, True, access_mode = access_mode, field_type = field_type) |
|
|
|
def NextAddBDItemTemplate(a_Bot, a_FSM, a_AddBDItemFunc, a_ParentTableName, a_ParentKeyFieldName, a_FieldName, a_MessageFunc, a_PostProcessFunc, a_AccessFunc, a_ButtonFunc, a_FinishButtonFunc, access_mode = user_access.AccessMode.ADD, field_type = bd_item.FieldType.text): |
|
return FinishOrNextAddBDItemTemplate(a_Bot, a_FSM, a_AddBDItemFunc, a_ParentTableName, a_ParentKeyFieldName, a_FieldName, a_MessageFunc, a_PostProcessFunc, a_AccessFunc, a_ButtonFunc, a_FinishButtonFunc, False, access_mode = access_mode, field_type = field_type) |
|
|
|
def FinishOrNextAddBDItemTemplate(a_Bot, a_FSM, a_AddBDItemFunc, a_ParentTableName, a_ParentKeyFieldName, a_FieldName, a_MessageFunc, a_PostProcessFunc, a_AccessFunc, a_ButtonFunc, a_FinishButtonFunc, a_Finish, access_mode = user_access.AccessMode.ADD, field_type = bd_item.FieldType.text): |
|
async def FinishAddBDItem(a_Message, state): |
|
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_utils.GetUserGroupData(a_Bot, 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: |
|
await state.finish() |
|
return simple_message.WorkFuncResult(bot_messages.MakeBotMessage(cancel_message), keyboard_func = a_FinishButtonFunc) |
|
|
|
parent_id = None |
|
if a_ParentKeyFieldName: |
|
parent_id = item_data[a_ParentKeyFieldName] |
|
if parent_id: |
|
check, res_of_work_func = await bd_item.CheckAccessBDItemTemplate(a_Bot, a_ParentTableName, a_ParentKeyFieldName, parent_id, a_MessageFunc, access_mode)(a_Message) |
|
else: |
|
res_of_work_func = await a_MessageFunc(a_Message, None, item_data) |
|
|
|
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: |
|
field_value = '0' |
|
else: |
|
if a_Message.photo == None or len(a_Message.photo) == 0: |
|
await state.finish() |
|
return simple_message.WorkFuncResult(bot_messages.MakeBotMessage(error_photo_type_message), keyboard_func = a_FinishButtonFunc) |
|
field_value = a_Message.photo[-1].file_id |
|
else: |
|
result = a_Message.text |
|
if a_PostProcessFunc: |
|
result = a_PostProcessFunc(result) |
|
field_value = result |
|
item_data[a_FieldName] = field_value |
|
if a_Finish: |
|
res, error = await a_AddBDItemFunc(item_data, user_id) |
|
await state_func() |
|
if error: |
|
return simple_message.WorkFuncResult(bot_messages.MakeBotMessage(error)) |
|
return res_of_work_func |
|
|
|
return simple_message.SimpleMessageTemplate(a_Bot, FinishAddBDItem, a_ButtonFunc, None, a_AccessFunc, access_mode) |
|
|
|
def AddBDItem3RegisterHandlers(a_Bot, a_StartCheckFunc, a_FSM, a_FSMName, a_FSMDesc, a_FSMPhoto, a_AddBDItemFunc, a_AddNameMessageFunc, a_AddDescMessageFunc, a_AddPhotoMessageFunc, a_FinishMessageFunc, a_ParentPrefix, a_ParentTableName : str, a_ParentKeyFieldName, a_NameField, a_DescField, a_PhotoField, a_GetButtonNameAndKeyValueAndAccessFunc, a_AccessFunc, a_ButtonFunc, access_mode = user_access.AccessMode.ADD): |
|
keyboard_cancel = bd_item.GetCancelKeyboardButtonsTemplate(a_Bot, a_AccessFunc, access_mode) |
|
keyboard_skip_and_cancel = bd_item.GetSkipAndCancelKeyboardButtonsTemplate(a_Bot, a_AccessFunc, access_mode) |
|
reg_func = a_Bot.RegisterMessageHandler |
|
if a_ParentTableName: |
|
reg_func = a_Bot.RegisterCallbackHandler |
|
reg_func(StartAddBDItemTemplate(a_Bot, a_FSM, a_FSMName, a_AddNameMessageFunc, a_ParentTableName, a_ParentKeyFieldName, a_ParentPrefix, a_AccessFunc, keyboard_cancel, a_ButtonFunc, access_mode), a_StartCheckFunc) |
|
|
|
# TODO: Сделать возможность не указывать все поля. пусть a_FSMName и a_NameField например могут быть пустыми. Возможно лучше вообще передавать список полей и их fsm |
|
a_Bot.RegisterMessageHandler(NextAddBDItemTemplate(a_Bot, a_FSM, None, a_ParentTableName, a_ParentKeyFieldName, a_NameField, a_AddDescMessageFunc, None, a_AccessFunc, keyboard_cancel, a_ButtonFunc, access_mode, field_type = bd_item.FieldType.text), state = a_FSMName) |
|
a_Bot.RegisterMessageHandler(NextAddBDItemTemplate(a_Bot, a_FSM, None, a_ParentTableName, a_ParentKeyFieldName, a_DescField, a_AddPhotoMessageFunc, None, a_AccessFunc, keyboard_skip_and_cancel, a_ButtonFunc, access_mode, field_type = bd_item.FieldType.text), state = a_FSMDesc) |
|
a_Bot.RegisterMessageHandler(FinishAddBDItemTemplate(a_Bot, a_FSM, a_AddBDItemFunc, a_ParentTableName, a_ParentKeyFieldName, a_PhotoField, a_FinishMessageFunc, None, a_AccessFunc, a_ButtonFunc, access_mode, field_type = bd_item.FieldType.photo), content_types = ['photo', 'text'], state = a_FSMPhoto) |
|
|
|
def AddBDItem1RegisterHandlers(a_Bot, a_StartCheckFunc, a_FSM, a_AddBDItemFunc, a_AddMessageFunc, a_FinishMessageFunc, a_ParentPrefix, a_ParentTableName : str, a_ParentKeyFieldName, a_FieldName, a_GetButtonNameAndKeyValueAndAccessFunc, a_AccessFunc, a_ButtonFunc, a_FieldType, access_mode = user_access.AccessMode.ADD): |
|
keyboard_cancel = bd_item.GetCancelKeyboardButtonsTemplate(a_Bot, a_AccessFunc, access_mode) |
|
reg_func = a_Bot.RegisterMessageHandler |
|
if a_ParentTableName: |
|
reg_func = a_Bot.RegisterCallbackHandler |
|
reg_func(StartAddBDItemTemplate(a_Bot, a_FSM, a_FSM.bd_item, a_AddMessageFunc, a_ParentTableName, a_ParentKeyFieldName, a_ParentPrefix, a_AccessFunc, keyboard_cancel, a_ButtonFunc, access_mode), a_StartCheckFunc) |
|
finish_handler = FinishAddBDItemTemplate(a_Bot, a_FSM, a_AddBDItemFunc, a_ParentTableName, a_ParentKeyFieldName, a_FieldName, a_FinishMessageFunc, None, a_AccessFunc, a_ButtonFunc, access_mode, field_type = a_FieldType) |
|
if a_FieldType == bd_item.FieldType.photo: |
|
a_Bot.RegisterMessageHandler(finish_handler, content_types = ['photo', 'text'], state = a_FSM.bd_item) |
|
else: |
|
a_Bot.RegisterMessageHandler(finish_handler, state = a_FSM.bd_item)
|
|
|