TPlatformBot/bot_modules/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
from template import bd_item_select, bd_item_view

from enum import Enum
from enum import auto

class OrderStatus(Enum):
    NEW = auto()
    PAY = auto()
    ADDRESS = auto()
    FINISH = auto()

# ---------------------------------------------------------
# БД
module_name = 'orders'

table_name = module_name
key_name = 'orderID'
name_field = 'orderName'
desc_field = 'orderDesc'
photo_field = 'orderPhoto'
status_field = 'orderStatus'
address_field = 'orderAddres'
access_field = 'orderAccess'
create_datetime_field = 'orderCreateDateTime'
parent_id_field = 'userID'

table = bd_table.Table(table_name, [
        bd_table.TableField(key_name, bd_table.TableFieldDestiny.KEY, bd_table.TableFieldType.INT),
        bd_table.TableField(name_field, bd_table.TableFieldDestiny.NAME, bd_table.TableFieldType.STR),
        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(status_field, bd_table.TableFieldDestiny.STATUS, bd_table.TableFieldType.STR),
        bd_table.TableField(address_field, bd_table.TableFieldDestiny.ADDRESS, 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),
        bd_table.TableField(parent_id_field, bd_table.TableFieldDestiny.PARENT_ID, bd_table.TableFieldType.INT),
        ])

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

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

class ButtonNames(Enum):
    LIST_CURRENT = auto() 

button_names = {
    mod_simple_message.ButtonNames.START: "‍🛒 Заказы",
    mod_table_operate.ButtonNames.LIST: "📃 Список всех заказов",
    ButtonNames.LIST_CURRENT: "📃 Список текущих заказов",
    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_ADDRESS: "𝌴 Изменить адрес в заказе",
    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}

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

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

<b>Время создания:</b> #{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_ADDRESS: 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: '''✅ Заказ успешно удалён!''',
}

def GetCurItemsTemplate(a_Bot, a_TableName, a_UserIDFieldName, a_StatusFieldName):
    def GetBDItem(a_KeyValue):
        return a_Bot.SQLRequest(f'SELECT * FROM {a_TableName} WHERE {a_UserIDFieldName} = ? AND {a_StatusFieldName} != ?', param = ([a_KeyValue, OrderStatus.FINISH]))
    return GetBDItem

def GetBDItemsForUserTemplate(a_GetItemsFunc):
    def GetBDItems(a_Message, a_UserGroups, a_ParentID):
        def GetBDItem(a_KeyValue):
            user_id = str(a_Message.from_user.id)
            return a_GetItemsFunc(a_Message, a_UserGroups, a_ParentID)(user_id)
        return GetBDItem

        return items
    return GetBDItems

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

    def GetItemsFunc(self):
        get_items_func = super().GetItemsFunc()
        if self.m_OnlyCurent:
            get_items_func = GetCurItemsTemplate(self.m_Bot, self.m_TableName, self.m_ParentIDFieldName, status_field)
        return GetBDItemsForUserTemplate(get_items_func)

class ModuleOrders(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, a_ParentModName, a_ChildModName, init_access, a_ChildModuleNameList, a_EditModuleNameList, a_Bot, a_ModuleAgregator, a_BotMessages, a_BotButtons, a_Log)

    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)

    def SelectSourceForCurrentTemplate(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_OnlyCurent = True)

    def AddBDItemFunc(self, a_ItemData, a_UserID):
        parent_id_field = self.m_Table.GetFieldNameByDestiny(bd_table.TableFieldDestiny.PARENT_ID)
        a_ItemData[parent_id_field] = a_UserID
        return super().AddBDItemFunc(a_ItemData, a_UserID)

    def GetStartKeyboardButtons(self, a_Message, a_UserGroups):
        parent_buttons = super().GetStartKeyboardButtons(a_Message, a_UserGroups)
        cur_buttons = [
                keyboard.ButtonWithAccess(self.GetButton(ButtonNames.LIST_CURRENT), user_access.AccessMode.VIEW, self.GetAccess()),
                ]
        return parent_buttons + keyboard.MakeButtons(self.m_Bot, cur_buttons, a_UserGroups)

    def RegisterHandlers(self):
        super().RegisterHandlers()
        table_name = self.m_Table.GetName()
        key_name = self.m_Table.GetFieldNameByDestiny(bd_table.TableFieldDestiny.KEY)

        def GetViewItemInlineKeyboardTemplate(a_ItemID):
            return self.GetViewItemInlineKeyboardTemplate(a_ItemID)

        GetButtonNameAndKeyValueAndAccess = self.m_GetButtonNameAndKeyValueAndAccessFunc
        GetAccess = self.m_GetAccessFunc

        default_keyboard_func = self.m_GetStartKeyboardButtonsFunc

        # Список текущих, открытых заказов
        a_ButtonName = self.GetButton(ButtonNames.LIST_CURRENT)
        if a_ButtonName:
            a_Prefix = self.RegisterSelect(a_ButtonName, user_access.AccessMode.VIEW, only_parent = True)
            a_Prefix = bd_item_select.SelectRegisterHandlers(self.m_Bot,\
                    self.SelectSourceForCurrentTemplate(a_Prefix, a_ButtonName), \
                    GetButtonNameAndKeyValueAndAccess,\
                    self.GetMessage(mod_table_operate.Messages.SELECT),\
                    GetAccess,\
                    access_mode = user_access.AccessMode.VIEW\
                    )
            bd_item_view.ShowBDItemRegisterHandlers(self.m_Bot,\
                    a_Prefix,\
                    table_name,\
                    key_name,\
                    self.ShowMessageTemplate(self.GetMessage(mod_table_operate.Messages.OPEN),GetViewItemInlineKeyboardTemplate),\
                    GetAccess,\
                    default_keyboard_func,\
                    access_mode = user_access.AccessMode.VIEW\
                    )