TPlatformBot/bot_modules/all_orders.py

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

# Заказы

from bot_sys import bot_bd, keyboard, user_access, bd_table
from bot_modules import mod_table_operate, mod_simple_message, orders, access_utils
from bot_modules import users, user_message
from template import bd_item_select, bd_item_view, bd_item, bd_item_edit

from enum import Enum
from enum import auto

# ---------------------------------------------------------
# БД
module_name = 'all_orders'

table = orders.table

init_access = f'{user_access.user_access_group_new}=-'

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

class ButtonNames(Enum):
    SHOW_USER_INFO = auto() 
    SEND_USER_MESSAGE = auto() 
    ADD_TO_DESC = auto() 

button_names = {}
button_names.update(orders.button_names)
button_names.pop(mod_table_operate.ButtonNames.ADD)

cur_button_names = {
    mod_simple_message.ButtonNames.START: "‍🛒 Все заказы",
    mod_table_operate.ButtonNames.SHOW: "☐  Открыть заказ",
    mod_table_operate.ButtonNames.LIST: "📃 Список текущих заказов",
    orders.ButtonNames.LIST_ALL: "📃 Список всех заказов",
    ButtonNames.SHOW_USER_INFO: "👤 Информация о пользователе",
    ButtonNames.SEND_USER_MESSAGE: "📨 Отправить сообщение пользователю",
    ButtonNames.ADD_TO_DESC: "𝌴 Добавить к описанию",
    mod_table_operate.ButtonNames.EDIT: "🛠 Редактировать заказ",
    mod_table_operate.EditButton(bd_table.TableFieldDestiny.PHOTO_PAY): "☐ Загрузить чек по оплате заказа",
    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.ADDRESS): "𝌴 Изменить адрес в заказе",
    mod_table_operate.EditButton(bd_table.TableFieldDestiny.ACCESS): "✋ Изменить доступ к заказу",
    mod_table_operate.EditButton(bd_table.TableFieldDestiny.STATUS): "𝌴 Изменить статус в заказе",
    mod_table_operate.EnumButton(orders.OrderStatus.NEW): "Заказ ожидает модерации",
    mod_table_operate.EnumButton(orders.OrderStatus.PAY): "Заказ ожидает оплаты",
    mod_table_operate.EnumButton(orders.OrderStatus.ADDRESS): "Заказ ожидает уточнения адреса",
    mod_table_operate.EnumButton(orders.OrderStatus.FINISH): "Заказ выполнен",
    mod_table_operate.ButtonNames.DEL: "❌ Удалить заказ",}
button_names.update(cur_button_names)

messages = {}
messages.update(orders.messages)

cur_messages = {
    mod_simple_message.Messages.START: f'''
<b>{button_names[mod_simple_message.ButtonNames.START]}</b>

''',
    mod_table_operate.EditMessage(bd_table.TableFieldDestiny.STATUS): f'''
Текущий статус заказа:
#{orders.status_field}

Введите новый статус заказа:
''',
    mod_table_operate.Messages.OPEN: f'''
<b>Номер заказа: #{orders.key_name} </b>

<b>Заказ: #{orders.name_field}</b>

<b>Описание и состав заказа:</b> #{orders.desc_field}

<b>Статус:</b> #{orders.status_field}

<b>Пользователь:</b> #{orders.user_id_field}

<b>Адрес доставки:</b> #{orders.address_field}

<b>Время создания:</b> #{orders.create_datetime_field}
''',
    mod_table_operate.InlineMessage(ButtonNames.ADD_TO_DESC): f'''
Текущее описание заказа:
<code>#{orders.desc_field}</code>

Введите дополение к описанию заказа:
''',
}

messages_notification = {
    mod_table_operate.NotificationMessage(orders.OrderStatus.NEW): f'''Статус заказа "#{orders.name_field}" изменён на - ожидает модерации''',
    mod_table_operate.NotificationMessage(orders.OrderStatus.PAY): f'''Статус заказа "#{orders.name_field}" изменён на - Заказ ожидает оплаты.

<b>Описание заказа:</b>
#{orders.desc_field}

Оплатите заказ и прикрепите чек об оплате к заказу. Для этого проследуйте по пути "Главное меню"->"Заказы"->"Загрузить чек по оплате моего заказа"
Или воспользуйтесь кнопкой:''',
    mod_table_operate.NotificationMessage(orders.OrderStatus.ADDRESS): f'''Статус заказа "#{orders.name_field}" изменён на - Заказ ожидает указания адреса доставки.

<b>Описание заказа:</b>
#{orders.desc_field}

Для этого проследуйте по пути "Главное меню"->"Заказы"->"Редактировать мой заказ"->"Изменить адрес в моём заказе"''',
    mod_table_operate.NotificationMessage(orders.OrderStatus.FINISH): f'''Статус заказа "#{orders.name_field}" изменён на - Заказ выполнен''',
}

messages.update(orders.messages_order_status)
messages.update(messages_notification)
messages.update(cur_messages)

def GetCurItemsTemplate(a_Bot, a_TableName, a_StatusFieldName, a_ParentIDFieldName):
    def GetBDItems(a_Message, a_UserGroups, a_ParentID):
        request = f'SELECT * FROM {a_TableName} WHERE {a_ParentIDFieldName} = ? AND {a_StatusFieldName} != ?'
        return a_Bot.SQLRequest(request, param = ([a_ParentID, str(orders.OrderStatus.FINISH)]))
    return GetBDItems

def GetBDItemsForUserTemplate(a_Bot, a_TableName, a_ParentIDFieldName):
    def GetBDItems(a_Message, a_UserGroups, a_ParentID):
        request = f'SELECT * FROM {a_TableName} WHERE {a_ParentIDFieldName} = ?'
        return a_Bot.SQLRequest(request, param = ([a_ParentID]))
    return GetBDItems

class DBItemForUserSelectSource(bd_item_select.DBItemSelectSource):
    def __init__(self, a_Bot, a_TableName, a_ParentIDFieldName, a_PrevPrefix, a_ButtonName, a_OnlyCurrent = False):
        super().__init__(a_Bot, a_TableName, a_ParentIDFieldName, a_PrevPrefix, a_ButtonName)
        self.m_OnlyCurrent = a_OnlyCurrent

    def GetItemsFunc(self):
        if self.m_OnlyCurrent:
            return GetCurItemsTemplate(self.m_Bot, self.m_TableName, orders.status_field, self.m_ParentIDFieldName)
        return GetBDItemsForUserTemplate(self.m_Bot, self.m_TableName, self.m_ParentIDFieldName)

class ModuleAllOrders(orders.ModuleOrders):
    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 GetInitBDCommands(self):
        # уже сделано в ModuleUserOrders
        return [
                access_utils.GetAccessForModuleRequest(self.GetName(), self.m_InitAccess, self.m_DefInitAccess),
                ]

    def GetName(self):
        return module_name

    def SelectSourceTemplate(self, a_PrevPrefix, a_ButtonName):
        parent_id_field = self.m_Table.GetFieldNameByDestiny(bd_table.TableFieldDestiny.PARENT_ID)
        return DBItemForUserSelectSource(self.m_Bot, self.m_Table.GetName(), parent_id_field, a_PrevPrefix, a_ButtonName, a_OnlyCurrent = True)

    def SelectSourceForAllTemplate(self, a_PrevPrefix, a_ButtonName):
        parent_id_field = self.m_Table.GetFieldNameByDestiny(bd_table.TableFieldDestiny.PARENT_ID)
        return DBItemForUserSelectSource(self.m_Bot, self.m_Table.GetName(), parent_id_field, a_PrevPrefix, a_ButtonName)

    def GetButtonNameAndKeyValueAndAccess(self, a_Item):
        user_field_id = self.m_Table.GetFieldIDByDestiny(bd_table.TableFieldDestiny.USER_ID)
        n, k, a = super().GetButtonNameAndKeyValueAndAccess(a_Item)
        return n + "," + str(a_Item[user_field_id]), k, a

    def GetPayInlineKeyboardFunc(self, a_ItemID, a_Destiny):
        def PayInlineKeyboard(a_Message, a_UserGroups):
            #print('PayInlineKeyboard', a_Message, a_UserGroups, a_ItemID)
            cur_buttons = []

            orders_mod = self.GetModule(orders.module_name)
            dst = a_Destiny
            if orders_mod:
                prefix = orders_mod.m_EditPrefix.get(dst, None)
                if prefix:
                    access = orders_mod.GetAccessForEditKeyboardButtons(orders_mod.m_Table.GetFieldByDestiny(dst))
                    cur_buttons += [
                        keyboard.InlineButtonWithAccess(orders_mod.GetButton(mod_table_operate.EditButton(dst)), prefix, a_ItemID, orders_mod.GetAccess(), access),
                    ]

            #print('cur_buttons', cur_buttons, a_Message, a_UserGroups, a_ItemID)
            return keyboard.MakeInlineKeyboardButtons(self.m_Bot, cur_buttons, a_UserGroups)
        return PayInlineKeyboard


    async def OnChangeField(self, a_Field, a_ItemID, a_ItemData, a_EditUserID):
        await super().OnChangeField(a_Field, a_ItemID, a_ItemData, a_EditUserID)
        if a_Field.m_Destiny == bd_table.TableFieldDestiny.STATUS:
            key_field = self.m_Table.GetFieldNameByDestiny(bd_table.TableFieldDestiny.KEY)
            item = bd_item.GetBDItemsTemplate(self.m_Bot, self.m_Table.GetName(), key_field)(a_ItemID)
            if len(item) < 1:
                self.m_Log.Error(f'Не удалось оповестить по заказу №{a_ItemID}.')
                return

            item = item[0]
            user_field_id = self.m_Table.GetFieldIDByDestiny(bd_table.TableFieldDestiny.USER_ID)
            parent_field_id = self.m_Table.GetFieldIDByDestiny(bd_table.TableFieldDestiny.PARENT_ID)
            status_field_id = self.m_Table.GetFieldIDByDestiny(bd_table.TableFieldDestiny.STATUS)
            #print('OnChangeField', item, user_field_id, status_field_id)
            owner_id = item[user_field_id]
            new_status = item[status_field_id]
            parent_id = item[parent_field_id]
            msg = self.GetMessage(mod_table_operate.NotificationMessage(new_status))
            parent_id_msg = None
            if msg:
                name = self.GetMessageNameWithTableFieldDestinyAndValue(mod_table_operate.NotificationMessage(new_status), bd_table.TableFieldDestiny.PARENT_ID, parent_id)
                #print('name', name, msg.GetLanguage())
                parent_id_msg = msg.FindMessageForLang(name, msg.GetLanguage())
            if parent_id_msg:
                msg = parent_id_msg
            if not msg:
                self.m_Log.Error(f'Не удалось оповестить по заказу №{a_ItemID}. Пустое сообщение для нового статуса {new_status}')
                return
            msg = self.UpdateMessage(msg, None, item)
            inline_keyboard = None
            #print('new_status', str(orders.OrderStatus.PAY), new_status)
            if new_status == str(orders.OrderStatus.PAY):
                inline_keyboard = self.GetPayInlineKeyboardFunc(a_ItemID, bd_table.TableFieldDestiny.PHOTO_PAY)
            elif new_status == str(orders.OrderStatus.ADDRESS):
                inline_keyboard = self.GetPayInlineKeyboardFunc(a_ItemID, bd_table.TableFieldDestiny.ADDRESS)

            await self.SendMessageToUser(msg, owner_id, a_GetInlineButtonsFunc = inline_keyboard)

    def GetStartButtons(self, a_Message, a_UserGroups):
        return [
            [mod_table_operate.ButtonNames.LIST, user_access.AccessMode.VIEW],
            [mod_table_operate.ButtonNames.EDIT, user_access.AccessMode.EDIT],
            [mod_table_operate.ButtonNames.ADD, user_access.AccessMode.ADD],
            [mod_table_operate.ButtonNames.DEL, user_access.AccessMode.DELETE],
        ]

    def GetViewItemInlineKeyboard(self, a_Message, a_UserGroups, a_ItemID):
        table_name = self.m_Table.GetName()
        key_name = self.GetKeyFieldName()
        cur_item = mod_table_operate.GetCurItem(self.m_Bot, table_name, key_name, a_ItemID)

        user_id = None
        user_id_field_index = self.m_Table.GetFieldIDByDestiny(bd_table.TableFieldDestiny.USER_ID)
        key_field_index = self.m_Table.GetFieldIDByDestiny(bd_table.TableFieldDestiny.KEY)
        if user_id_field_index and cur_item and cur_item[user_id_field_index]:
            user_id = cur_item[user_id_field_index]

        if not user_id:
            return keyboard.MakeInlineKeyboardButtons(self.m_Bot, [], a_UserGroups)

        cur_buttons = []

        for dst in bd_table.TableFieldDestiny.DESC, bd_table.TableFieldDestiny.STATUS:
            access = self.GetAccessForEditKeyboardButtons(self.m_Table.GetFieldByDestiny(dst))
            btn = mod_table_operate.EditButton(dst)
            prefix = self.m_EditPrefix.get(dst, None)
            if prefix:
                cur_buttons += [
                    keyboard.InlineButtonWithAccess(self.GetButton(btn), prefix, cur_item[key_field_index], self.GetAccess(), access),
                    ]

        access = self.GetAccessForEditKeyboardButtons(self.m_Table.GetFieldByDestiny(bd_table.TableFieldDestiny.DESC))
        a_Button = ButtonNames.ADD_TO_DESC
        prefix = self.m_EditPrefix.get(a_Button, None)
        cur_buttons += [
            keyboard.InlineButtonWithAccess(self.GetButton(a_Button), prefix, cur_item[key_field_index], self.GetAccess(), access),
            ]

        user_message_mod = self.GetModule(user_message.module_name)
        if user_message_mod:
            cur_buttons += [
                keyboard.InlineButtonWithAccess(self.GetButton(ButtonNames.SEND_USER_MESSAGE), user_message_mod.GetAddPrefix(), user_id, user_message_mod.GetAccess(), user_access.AccessMode.ADD),
                ]

        users_mod = self.GetModule(users.module_name)
        if users_mod:
            cur_buttons += [
                keyboard.InlineButtonWithAccess(self.GetButton(ButtonNames.SHOW_USER_INFO), users_mod.GetShowPrefix(), user_id, users_mod.GetAccess(), user_access.AccessMode.VIEW),
                ]

        return keyboard.MakeInlineKeyboardButtons(self.m_Bot, cur_buttons, a_UserGroups)

    def RegisterAddToDesc(self):
            a_AccessMode = user_access.AccessMode.EDIT
            a_Button = ButtonNames.ADD_TO_DESC
            a_ButtonName = self.GetButton(a_Button)
            a_EditMessage = self.GetMessage(mod_table_operate.InlineMessage(a_Button))
            a_Field = self.m_Table.GetFieldByDestiny(bd_table.TableFieldDestiny.DESC)
            a_FieldName = a_Field.m_Name

            a_FieldType = bd_item.FieldType.text
            if a_Field.m_Type == bd_table.TableFieldType.PHOTO:
                a_FieldType = bd_item.FieldType.photo

            if not a_ButtonName or not a_EditMessage:
                return

            async def OnChange(a_ItemID, a_ItemData, a_EditUserID):
                await self.OnChangeField(a_Field, a_ItemID, a_ItemData, a_EditUserID)
                return self.OnChange()

            table_name = self.m_Table.GetName()
            key_name = self.GetKeyFieldName()
            edit_keyboard_func = self.m_GetEditKeyboardButtonsFunc
            GetButtonNameAndKeyValueAndAccess = self.m_GetButtonNameAndKeyValueAndAccessFunc
            GetAccess = self.m_GetAccessFunc

            async def EditBDItemFunc(a_ItemData, a_UserID):
                a_KeyName = key_name
                a_TableName = table_name
                item_id = a_ItemData[a_KeyName]
                field_value = a_ItemData[a_FieldName]

                cur_item = mod_table_operate.GetCurItem(self.m_Bot, table_name, key_name, item_id)

                desc = ""
                desc_field_index = self.m_Table.GetFieldIDByDestiny(bd_table.TableFieldDestiny.DESC)
                if desc_field_index and cur_item and cur_item[desc_field_index]:
                    desc = cur_item[desc_field_index]
                
                new_value = desc + "\n" + field_value
                res, error = bd_item.EditBDItemInTableTemplate(self.m_Bot, a_TableName, a_KeyName, a_FieldName)(item_id, new_value)
                if error:
                    self.m_Bot.GetLog().Error(f'Пользователю {a_UserID} не удалось изменить поле в таблице {a_TableName} ключу {a_KeyName}={item_id}. Новое значение поля {a_FieldName}={field_value}. Ошибка: {error}')
                else:
                    self.m_Bot.GetLog().Success(f'Пользователь {a_UserID} добавил в поле в таблице {a_TableName} ключу {a_KeyName}={item_id}. Новое значение поля {a_FieldName}={new_value}. Старое {desc}.')

                await OnChange(item_id, a_ItemData, a_UserID)
                return res, error

            a_Prefix = self.RegisterSelect(a_ButtonName, user_access.AccessMode.VIEW, only_parent = True)

            a_Prefix = bd_item_edit.CustomEditBDItemRegisterHandlers(self.m_Bot, \
                EditBDItemFunc,
                self.SelectSourceTemplate(a_Prefix, a_ButtonName), \
                mod_table_operate.MakeFSMForAddAndEdit(self.GetName(), a_FieldName), \
                self.GetMessage(mod_table_operate.Messages.SELECT_TO_EDIT), \
                self.ShowMessageTemplate(a_EditMessage), \
                self.ShowMessageTemplate(self.GetMessage(mod_table_operate.Messages.SUCCESS_EDIT)), \
                table_name, \
                key_name, \
                a_FieldName, \
                self.PostProccessingForFieldForEditTemplate(a_Field),\
                GetButtonNameAndKeyValueAndAccess, \
                GetAccess, \
                self.AdditionalKeyboardForEditTemplate(a_Field),\
                edit_keyboard_func, \
                access_mode = a_AccessMode, \
                field_type = a_FieldType\
                )

            self.m_EditPrefix.update({a_Button: a_Prefix})


    def RegisterHandlers(self):
        super().RegisterHandlers()
        self.RegisterAddToDesc()