TPlatformBot/bot_modules/messages.py

# -*- coding: utf8 -*-
# Общественное достояние, 2023, Алексей Безбородов (Alexei Bezborodov) <AlexeiBv+mirocod_platform_bot@narod.ru> 

# Сообщения

from bot_sys import bot_bd, keyboard, user_access, bd_table, bot_messages
from bot_modules import mod_table_operate, mod_simple_message, access_utils
from template import bd_item

# ---------------------------------------------------------
# БД
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.PHOTO),
        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}=-'

# ---------------------------------------------------------
# Сообщения и кнопки

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.EditButton(bd_table.TableFieldDestiny.PHOTO): "☐ Изменить изображение у сообщения",
    mod_table_operate.EditButton(bd_table.TableFieldDestiny.NAME): "≂  Изменить название сообщения",
    mod_table_operate.EditButton(bd_table.TableFieldDestiny.DESC): "𝌴  Изменить описание сообщения",
    mod_table_operate.EditButton(bd_table.TableFieldDestiny.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.CreateMessage(bd_table.TableFieldDestiny.NAME): f'''
Создание сообщения. Шаг №1

Вы можете добавить номер родителя ID в таком формате: ИмяСообщения{'<code>?' + str(bd_table.TableFieldDestiny.PARENT_ID) + '=</code>ID'}
, чтобы сообщение было только для него

Введите название сообщения:
''',
    mod_table_operate.CreateMessage(bd_table.TableFieldDestiny.DESC): '''
Создание сообщения. Шаг №2

Введите описание сообщения:
''',
    mod_table_operate.CreateMessage(bd_table.TableFieldDestiny.PHOTO): '''
Создание сообщения. Шаг №3

Загрузите обложку для сообщения (Фото):
Она будет отображаться в его описании.
''',
    mod_table_operate.Messages.SUCCESS_CREATE: '''✅ Сообщение успешно добавлено!''',
    mod_table_operate.Messages.START_EDIT: '''
Пожалуйста, выберите действие:
''',
    mod_table_operate.Messages.SELECT_TO_EDIT: '''
Выберите сообщение, который вы хотите отредактировать.
''',
    mod_table_operate.EditMessage(bd_table.TableFieldDestiny.PHOTO): '''
Загрузите новую обложку для сообщения (Фото):
Она будет отображаться в его описании.
''',
    mod_table_operate.EditMessage(bd_table.TableFieldDestiny.NAME): f'''
Текущее название сообщения:
#{name_field}

Введите новое название сообщения:
''',
    mod_table_operate.EditMessage(bd_table.TableFieldDestiny.DESC): f'''
Текущее описание сообщения:
#{desc_field}

Введите новое описание сообщения:
''',
    mod_table_operate.EditMessage(bd_table.TableFieldDestiny.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_BotSubscribes, a_Log):
        super().__init__(table, messages, button_names, a_ParentModName, a_ChildModName, init_access, init_access, a_ChildModuleNameList, a_EditModuleNameList, a_Bot, a_ModuleAgregator, a_BotMessages, a_BotButtons, a_BotSubscribes, a_Log)

    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 LoadAllMessages(self):
        table_name = self.m_Table.GetName()
        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)


    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)

        self.LoadAllMessages()

        self.m_BotMessages.UpdateSignal(self.m_Log.GetTimeNow())

    def OnChange(self):
        self.FlushMessages()

    async def PostDelete(self, a_CallbackQuery, a_ItemID):
        self.LoadAllMessages() # Сначала загружаем, т.к. одно сообщение удалилось
        await super().PostDelete(a_CallbackQuery, a_ItemID)