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.
436 lines
20 KiB
436 lines
20 KiB
# -*- coding: utf8 -*- |
|
# Общественное достояние, 2023, Алексей Безбородов (Alexei Bezborodov) <AlexeiBv+mirocod_platform_bot@narod.ru> |
|
|
|
# Сообщения |
|
|
|
from bot_sys import bot_bd, keyboard, user_access, user_messages, bd_table, bot_messages |
|
from bot_modules import mod_table_operate, mod_simple_message, access_utils |
|
from template import bd_item |
|
|
|
from aiogram.dispatcher import FSMContext |
|
from aiogram.dispatcher.filters.state import State, StatesGroup |
|
|
|
class FSMCreateMessage(StatesGroup): |
|
name = State() |
|
desc = State() |
|
photo = State() |
|
|
|
class FSMEditMessagePhotoItem(StatesGroup): |
|
item_id = State() |
|
item_field = State() |
|
|
|
class FSMEditMessageNameItem(StatesGroup): |
|
item_id = State() |
|
item_field = State() |
|
|
|
class FSMEditMessageDescItem(StatesGroup): |
|
item_id = State() |
|
item_field = State() |
|
|
|
class FSMEditMessageAccessItem(StatesGroup): |
|
item_id = State() |
|
item_field = State() |
|
# --------------------------------------------------------- |
|
# БД |
|
module_name = 'messages' |
|
|
|
table_name = module_name |
|
key_name = 'messageID' |
|
name_field = 'messageName' |
|
desc_field = 'messageDesc' |
|
photo_field = 'messagePhoto' |
|
access_field = 'messageAccess' |
|
create_datetime_field = 'messageCreateDateTime' |
|
parent_id_field = 'languageID' |
|
|
|
table_name_field = bd_table.TableField(name_field, bd_table.TableFieldDestiny.NAME, bd_table.TableFieldType.STR) |
|
table_parent_id_field = bd_table.TableField(parent_id_field, bd_table.TableFieldDestiny.PARENT_ID, bd_table.TableFieldType.INT) |
|
|
|
table = bd_table.Table(table_name, [ |
|
bd_table.TableField(key_name, bd_table.TableFieldDestiny.KEY, bd_table.TableFieldType.INT), |
|
table_name_field, |
|
bd_table.TableField(desc_field, bd_table.TableFieldDestiny.DESC, bd_table.TableFieldType.STR), |
|
bd_table.TableField(photo_field, bd_table.TableFieldDestiny.PHOTO, bd_table.TableFieldType.STR), |
|
bd_table.TableField(access_field, bd_table.TableFieldDestiny.ACCESS, bd_table.TableFieldType.STR), |
|
bd_table.TableField(create_datetime_field, bd_table.TableFieldDestiny.CREATE_DATE, bd_table.TableFieldType.STR), |
|
table_parent_id_field, |
|
], |
|
[ |
|
[table_name_field, table_parent_id_field], |
|
] |
|
) |
|
|
|
init_access = f'{user_access.user_access_group_all}=-' |
|
|
|
fsm = { |
|
mod_table_operate.FSMs.CREATE: FSMCreateMessage, |
|
mod_table_operate.FSMs.EDIT_NAME: FSMEditMessageNameItem, |
|
mod_table_operate.FSMs.EDIT_DESC: FSMEditMessageDescItem, |
|
mod_table_operate.FSMs.EDIT_PHOTO: FSMEditMessagePhotoItem, |
|
mod_table_operate.FSMs.EDIT_ACCESS: FSMEditMessageAccessItem, |
|
} |
|
|
|
# --------------------------------------------------------- |
|
# Сообщения и кнопки |
|
|
|
button_names = { |
|
mod_simple_message.ButtonNames.START: "✉ Сообщения", |
|
mod_table_operate.ButtonNames.LIST: "📃 Список сообщений", |
|
mod_table_operate.ButtonNames.ADD: "☑ Добавить сообщение", |
|
mod_table_operate.ButtonNames.EDIT: "🛠 Редактировать сообщение", |
|
mod_table_operate.ButtonNames.EDIT_PHOTO: "☐ Изменить изображение у сообщения", |
|
mod_table_operate.ButtonNames.EDIT_NAME: "≂ Изменить название у сообщения", |
|
mod_table_operate.ButtonNames.EDIT_DESC: "𝌴 Изменить описание у сообщения", |
|
mod_table_operate.ButtonNames.EDIT_ACCESS: "✋ Изменить доступ к сообщению", |
|
mod_table_operate.ButtonNames.DEL: "❌ Удалить сообщение", |
|
} |
|
|
|
messages = { |
|
mod_simple_message.Messages.START: f''' |
|
<b>{button_names[mod_simple_message.ButtonNames.START]}</b> |
|
|
|
''', |
|
mod_table_operate.Messages.SELECT: ''' |
|
Пожалуйста, выберите сообщение: |
|
''', |
|
mod_table_operate.Messages.ERROR_FIND: ''' |
|
❌ Ошибка, сообщение не найден |
|
''', |
|
mod_table_operate.Messages.OPEN: f''' |
|
<b>Сообщение: #{name_field}</b> |
|
|
|
#{desc_field} |
|
|
|
Время создания: #{create_datetime_field} |
|
''', |
|
mod_table_operate.Messages.CREATE_NAME: ''' |
|
Создание сообщения. Шаг №1 |
|
|
|
Введите название сообщения: |
|
''', |
|
mod_table_operate.Messages.CREATE_DESC: ''' |
|
Создание сообщения. Шаг №2 |
|
|
|
Введите описание сообщения: |
|
''', |
|
mod_table_operate.Messages.CREATE_PHOTO: ''' |
|
Создание сообщения. Шаг №3 |
|
|
|
Загрузите обложку для сообщения (Фото): |
|
Она будет отображаться в его описании. |
|
''', |
|
mod_table_operate.Messages.SUCCESS_CREATE: '''✅ Сообщение успешно добавлено!''', |
|
mod_table_operate.Messages.START_EDIT: ''' |
|
Пожалуйста, выберите действие: |
|
''', |
|
mod_table_operate.Messages.SELECT_TO_EDIT: ''' |
|
Выберите сообщение, который вы хотите отредактировать. |
|
''', |
|
mod_table_operate.Messages.EDIT_PHOTO: ''' |
|
Загрузите новую обложку для сообщения (Фото): |
|
Она будет отображаться в его описании. |
|
''', |
|
mod_table_operate.Messages.EDIT_NAME: f''' |
|
Текущее название сообщения: |
|
#{name_field} |
|
|
|
Введите новое название сообщения: |
|
''', |
|
mod_table_operate.Messages.EDIT_DESC: f''' |
|
Текущее описание сообщения: |
|
#{desc_field} |
|
|
|
Введите новое описание сообщения: |
|
''', |
|
mod_table_operate.Messages.EDIT_ACCESS: f''' |
|
Текущий доступ к сообщениеу: |
|
#{access_field} |
|
|
|
{user_access.user_access_readme} |
|
|
|
Введите новую строку доступа: |
|
''', |
|
mod_table_operate.Messages.SUCCESS_EDIT: '''✅ Сообщение успешно отредактировано!''', |
|
mod_table_operate.Messages.SELECT_TO_DELETE: ''' |
|
Выберите сообщение, который вы хотите удалить. |
|
''', |
|
mod_table_operate.Messages.SUCCESS_DELETE: '''✅ Сообщение успешно удалено!''', |
|
} |
|
|
|
class ModuleMessages(mod_table_operate.TableOperateModule): |
|
def __init__(self, a_ParentModName, a_ChildModName, a_ChildModuleNameList, a_EditModuleNameList, a_Bot, a_ModuleAgregator, a_BotMessages, a_BotButtons, a_Log): |
|
super().__init__(table, messages, button_names, fsm, a_ParentModName, a_ChildModName, init_access, a_ChildModuleNameList, a_EditModuleNameList, a_Bot, a_ModuleAgregator, a_BotMessages, a_BotButtons, a_Log) |
|
self.m_LanguageIDs = {} |
|
|
|
def GetName(self): |
|
return module_name |
|
|
|
def AddOrIgnoreMessage(self, a_Message): |
|
table_name = self.m_Table.GetName() |
|
name_field = self.m_Table.GetFieldNameByDestiny(bd_table.TableFieldDestiny.NAME) |
|
photo_field = self.m_Table.GetFieldNameByDestiny(bd_table.TableFieldDestiny.PHOTO) |
|
desc_field = self.m_Table.GetFieldNameByDestiny(bd_table.TableFieldDestiny.DESC) |
|
access_field = self.m_Table.GetFieldNameByDestiny(bd_table.TableFieldDestiny.ACCESS) |
|
create_datetime_field = self.m_Table.GetFieldNameByDestiny(bd_table.TableFieldDestiny.CREATE_DATE) |
|
parent_id_field = self.m_Table.GetFieldNameByDestiny(bd_table.TableFieldDestiny.PARENT_ID) |
|
|
|
lang_id = self.GetModule(self.m_ParentModName).GetLangID(a_Message.m_Language) |
|
return self.m_Bot.SQLRequest(f'INSERT OR IGNORE INTO {table_name}({photo_field}, {name_field}, {desc_field}, {access_field}, {parent_id_field}, {create_datetime_field}) VALUES(?, ?, ?, ?, ?, {bot_bd.GetBDDateTimeNow()})', |
|
commit = True, return_error = True, param = (a_Message.m_PhotoID, a_Message.m_MessageName, a_Message.m_MessageDesc, access_utils.GetItemDefaultAccessForModule(self.m_Bot, module_name), lang_id)) |
|
|
|
def FlushMessages(self): |
|
msg = self.m_BotMessages.GetMessages() |
|
for lang, msg_dict in msg.items(): |
|
for msg_name, message in msg_dict.items(): |
|
self.AddOrIgnoreMessage(message) |
|
|
|
msgs_bd = bd_item.GetAllItemsTemplate(self.m_Bot, table_name)() |
|
if msgs_bd: |
|
for m in msgs_bd: |
|
name = m[1] |
|
desc = m[2] |
|
photo_id = m[3] |
|
lang_id = m[6] |
|
lang_name = self.GetModule(self.m_ParentModName).GetLangName(lang_id) |
|
self.m_BotMessages.CreateMessage(name, desc, self.m_Log.GetTimeNow(), a_MessagePhotoID = photo_id, a_MessageLang = lang_name) |
|
|
|
self.m_BotMessages.UpdateSignal(self.m_Log.GetTimeNow()) |
|
|
|
def OnChange(self): |
|
self.FlushMessages() |
|
|
|
|
|
''' |
|
# --------------------------------------------------------- |
|
# Работа с кнопками |
|
|
|
def GetEditMessageKeyboardButtons(a_Message, a_UserGroups): |
|
cur_buttons = GetModuleButtons() + [ |
|
keyboard.ButtonWithAccess(edit_message_photo_button_name, user_access.AccessMode.EDIT, GetAccess()), |
|
keyboard.ButtonWithAccess(edit_message_name_button_name, user_access.AccessMode.EDIT, GetAccess()), |
|
keyboard.ButtonWithAccess(edit_message_desc_button_name, user_access.AccessMode.EDIT, GetAccess()), |
|
keyboard.ButtonWithAccess(edit_message_access_button_name, user_access.AccessMode.ACCEES_EDIT, GetAccess()), |
|
] |
|
mods = [start] |
|
return keyboard.MakeKeyboard(keyboard.GetButtons(mods) + cur_buttons, a_UserGroups) |
|
|
|
def GetStartMessageKeyboardButtons(a_Message, a_UserGroups): |
|
cur_buttons = [ |
|
keyboard.ButtonWithAccess(list_message_button_name, user_access.AccessMode.VIEW, GetAccess()), |
|
keyboard.ButtonWithAccess(add_message_button_name, user_access.AccessMode.ADD, GetAccess()), |
|
keyboard.ButtonWithAccess(del_message_button_name, user_access.AccessMode.DELETE, GetAccess()), |
|
keyboard.ButtonWithAccess(edit_message_button_name, user_access.AccessMode.EDIT, GetAccess()) |
|
] |
|
mods = [start, languages] |
|
return keyboard.MakeKeyboard(keyboard.GetButtons(mods) + cur_buttons, a_UserGroups) |
|
|
|
def GetViewItemInlineKeyboardTemplate(a_ItemID): |
|
def GetViewItemInlineKeyboard(a_Message, a_UserGroups): |
|
cur_buttons = [ |
|
#keyboard.InlineButtonWithAccess(needs.list_need_button_name, needs.select_needs_prefix, a_ItemID, GetAccess(), user_access.AccessMode.VIEW), |
|
] |
|
return keyboard.MakeInlineKeyboard(cur_buttons, a_UserGroups) |
|
return GetViewItemInlineKeyboard |
|
|
|
# --------------------------------------------------------- |
|
# Обработка сообщений |
|
|
|
# стартовое сообщение |
|
async def MessagesOpen(a_Message : types.message, state = None): |
|
return simple_message.WorkFuncResult(base_message_message) |
|
|
|
def GetButtonNameAndKeyValueAndAccess(a_Item): |
|
# messageName messageID messageAccess |
|
return a_Item[1], a_Item[0], a_Item[4] |
|
|
|
def ShowMessageTemplate(a_StringMessage, keyboard_template_func = None): |
|
async def ShowMessage(a_CallbackQuery : types.CallbackQuery, a_Item): |
|
if (len(a_Item) < 6): |
|
return simple_message.WorkFuncResult(error_find_proj_message) |
|
|
|
if message_success_edit_message == a_StringMessage: |
|
FlushMessages() |
|
# TODO FlushMessages происходит рано. Нужно после изменений |
|
msg = a_StringMessage.\ |
|
replace(f'#{name_field}', a_Item[1]).\ |
|
replace(f'#{desc_field}', a_Item[2]).\ |
|
replace(f'#{create_datetime_field}', a_Item[5]).\ |
|
replace(f'#{access_field}', a_Item[4]) |
|
keyboard_func = None |
|
if keyboard_template_func: |
|
keyboard_func = keyboard_template_func(a_Item[0]) |
|
return simple_message.WorkFuncResult(msg, photo_id = a_Item[3], item_access = a_Item[4], keyboard_func = keyboard_func) |
|
return ShowMessage |
|
|
|
def SimpleMessageTemplateLegacy(a_StringMessage): |
|
async def ShowMessage(a_CallbackQuery : types.CallbackQuery, a_Item): |
|
return simple_message.WorkFuncResult(a_StringMessage) |
|
return ShowMessage |
|
|
|
# Удаление сообщения |
|
|
|
async def MessagePreDelete(a_CallbackQuery : types.CallbackQuery, a_Item): |
|
if (len(a_Item) < 6): |
|
return simple_message.WorkFuncResult(error_find_proj_message) |
|
access = a_Item[4] |
|
return simple_message.WorkFuncResult('', None, item_access = access) |
|
|
|
async def MessagePostDelete(a_CallbackQuery : types.CallbackQuery, a_ItemID): |
|
log.Success(f'Сообщение №{a_ItemID} была удалена пользователем {a_CallbackQuery.from_user.id}.') |
|
#TODO: удалить вложенные |
|
FlushMessages() |
|
return simple_message.WorkFuncResult(message_success_delete_message) |
|
|
|
# --------------------------------------------------------- |
|
# Работа с базой данных сообщений |
|
|
|
def AddBDItemFunc(a_ItemData, a_UserID): |
|
res, error = bot_bd.SQLRequestToBD(f'INSERT INTO {table_name}({photo_field}, {name_field}, {desc_field}, {access_field}, {parent_id_field}, {create_datetime_field}) VALUES(?, ?, ?, ?, ?, {bot_bd.GetBDDateTimeNow()})', |
|
commit = True, return_error = True, param = (a_ItemData[photo_field], a_ItemData[name_field], a_ItemData[desc_field], access.GetItemDefaultAccessForModule(module_name) + f";{a_UserID}=+", a_ItemData[parent_id_field])) |
|
|
|
if error: |
|
log.Error(f'Пользоватлель {a_UserID}. Ошибка добавления записи в таблицу {table_name} ({a_ItemData[photo_field]}, {a_ItemData[name_field]}, {a_ItemData[desc_field]}, {access.GetItemDefaultAccessForModule(module_name)}).') |
|
else: |
|
log.Success(f'Пользоватлель {a_UserID}. Добавлена запись в таблицу {table_name} ({a_ItemData[photo_field]}, {a_ItemData[name_field]}, {a_ItemData[desc_field]}, {access.GetItemDefaultAccessForModule(module_name)}).') |
|
|
|
FlushMessages() |
|
return res, error |
|
|
|
# --------------------------------------------------------- |
|
# API |
|
|
|
def AddOrIgnoreMessage(a_Message : user_messages.Message): |
|
return bot_bd.SQLRequestToBD(f'INSERT OR IGNORE INTO {table_name}({photo_field}, {name_field}, {desc_field}, {access_field}, {parent_id_field}, {create_datetime_field}) VALUES(?, ?, ?, ?, ?, {bot_bd.GetBDDateTimeNow()})', |
|
commit = True, return_error = True, param = (a_Message.m_PhotoID, a_Message.m_MessageName, a_Message.m_MessageDesc, access.GetItemDefaultAccessForModule(module_name), languages.GetLangID(a_Message.m_Language))) |
|
|
|
def FlushMessages(): |
|
msg = user_messages.GetMessages() |
|
for lang, msg_dict in msg.items(): |
|
for msg_name, message in msg_dict.items(): |
|
AddOrIgnoreMessage(message) |
|
|
|
msgs_bd = bd_item.GetAllItemsTemplate(table_name)() |
|
if msgs_bd: |
|
for m in msgs_bd: |
|
name = m[1] |
|
lang_id = m[6] |
|
lang_name = languages.GetLangName(lang_id) |
|
new_msg = user_messages.Message(name, m[2], lang_name, m[3], log.GetTimeNow()) |
|
if not msg.get(lang_name, None): |
|
msg[lang_name] = {} |
|
msg[lang_name][name] = new_msg |
|
user_messages.UpdateSignal(log.GetTimeNow()) |
|
|
|
# Инициализация БД |
|
def GetInitBDCommands(): |
|
return init_bd_cmds |
|
|
|
def GetAccess(): |
|
return access.GetAccessForModule(module_name) |
|
|
|
# Доступные кнопки |
|
def GetModuleButtons(): |
|
return [keyboard.ButtonWithAccess(messages_button_name, user_access.AccessMode.VIEW, GetAccess())] |
|
|
|
# Обработка кнопок |
|
def RegisterHandlers(dp : Dispatcher): |
|
defaul_keyboard_func = GetStartMessageKeyboardButtons |
|
def RegisterSelectParent(a_ButtonName, access_mode): |
|
a_PrefixBase = a_ButtonName |
|
return bd_item_select.FirstSelectBDItemRegisterHandlers(dp, \ |
|
a_PrefixBase, \ |
|
a_ButtonName, \ |
|
languages.table_name, \ |
|
languages.key_name, \ |
|
languages.GetButtonNameAndKeyValueAndAccess, \ |
|
languages.select_language_message, \ |
|
languages.GetAccess, access_mode = access_mode\ |
|
) |
|
|
|
# Стартовое сообщение |
|
dp.register_message_handler(simple_message.SimpleMessageTemplateLegacy(MessagesOpen, defaul_keyboard_func, GetAccess), text = messages_button_name) |
|
|
|
# Список сообщений |
|
a_Prefix = RegisterSelectParent(list_message_button_name, user_access.AccessMode.VIEW) |
|
bd_item_view.LastSelectAndShowBDItemRegisterHandlers(dp, \ |
|
a_Prefix, parent_id_field, \ |
|
table_name, key_name, \ |
|
ShowMessageTemplate(message_open_message), \ |
|
GetButtonNameAndKeyValueAndAccess, \ |
|
select_message_message, \ |
|
GetAccess, \ |
|
defaul_keyboard_func, \ |
|
access_mode = user_access.AccessMode.VIEW\ |
|
) |
|
global select_messages_prefix |
|
select_messages_prefix = a_Prefix |
|
|
|
# Удаление сообщения |
|
a_Prefix = RegisterSelectParent(del_message_button_name, user_access.AccessMode.DELETE) |
|
bd_item_delete.DeleteBDItemRegisterHandlers(dp, \ |
|
a_Prefix, \ |
|
bd_item.GetCheckForPrefixFunc(a_Prefix), \ |
|
table_name, \ |
|
key_name, \ |
|
parent_id_field, \ |
|
MessagePreDelete, \ |
|
MessagePostDelete, \ |
|
GetButtonNameAndKeyValueAndAccess, \ |
|
select_message_message, \ |
|
GetAccess, \ |
|
defaul_keyboard_func\ |
|
) |
|
|
|
# Добавление сообщения |
|
a_Prefix = RegisterSelectParent(add_message_button_name, user_access.AccessMode.ADD) |
|
bd_item_add.AddBDItem3RegisterHandlers(dp, \ |
|
bd_item.GetCheckForPrefixFunc(a_Prefix), \ |
|
FSMCreateMessage, \ |
|
FSMCreateMessage.name,\ |
|
FSMCreateMessage.desc, \ |
|
FSMCreateMessage.photo,\ |
|
AddBDItemFunc, \ |
|
SimpleMessageTemplateLegacy(message_create_name_message), \ |
|
SimpleMessageTemplateLegacy(message_create_desc_message), \ |
|
SimpleMessageTemplateLegacy(message_create_photo_message), \ |
|
SimpleMessageTemplateLegacy(message_success_create_message), \ |
|
a_Prefix,\ |
|
languages.table_name, \ |
|
languages.key_name, \ |
|
name_field, \ |
|
desc_field, \ |
|
photo_field, \ |
|
GetButtonNameAndKeyValueAndAccess, \ |
|
GetAccess, \ |
|
GetStartMessageKeyboardButtons\ |
|
) |
|
|
|
def RegisterEdit(a_ButtonName, a_FSM, a_EditMessage, a_FieldName, a_FieldType, a_AccessMode = user_access.AccessMode.EDIT): |
|
a_Prefix = RegisterSelectParent(a_ButtonName, a_AccessMode) |
|
bd_item_edit.EditBDItemRegisterHandlers(dp, \ |
|
a_Prefix, \ |
|
a_FSM, \ |
|
bd_item.GetCheckForPrefixFunc(a_Prefix), \ |
|
message_select_to_edit_message, \ |
|
ShowMessageTemplate(a_EditMessage), \ |
|
ShowMessageTemplate(message_success_edit_message), \ |
|
table_name, \ |
|
key_name, \ |
|
parent_id_field, \ |
|
a_FieldName, \ |
|
GetButtonNameAndKeyValueAndAccess, \ |
|
GetAccess, \ |
|
edit_keyboard_func, \ |
|
access_mode = a_AccessMode, \ |
|
field_type = a_FieldType\ |
|
) |
|
|
|
# Редактирование сообщения |
|
edit_keyboard_func = GetEditMessageKeyboardButtons |
|
dp.register_message_handler(simple_message.InfoMessageTemplateLegacy(message_start_edit_message, edit_keyboard_func, GetAccess, access_mode = user_access.AccessMode.EDIT), text = edit_message_button_name) |
|
|
|
RegisterEdit(edit_message_photo_button_name, FSMEditMessagePhotoItem, message_edit_photo_message, photo_field, bd_item.FieldType.photo) |
|
RegisterEdit(edit_message_name_button_name, FSMEditMessageNameItem, message_edit_name_message, name_field, bd_item.FieldType.text) |
|
RegisterEdit(edit_message_desc_button_name, FSMEditMessageDescItem, message_edit_desc_message, desc_field, bd_item.FieldType.text) |
|
RegisterEdit(edit_message_access_button_name, FSMEditMessageAccessItem, message_edit_access_message, access_field, bd_item.FieldType.text) |
|
'''
|
|
|