buu_command_dict.py

import buu_config, buu_database
import utils
import itchat
import buu_step6_menu

class class_command_dict(object):

    def step_0_tips(self, msg, userId):
        return '欢迎使用打印小助手，\n' \
                '6 - 自助打印\n\n' \
                'PS: 您在进行任何操作时都可以用 #返回 来返回到上一级操作~'

    def step_mig(self, msg, userId):
        return '功能已迁移。'

    def step_process(self, msg, userId, msg_opt):
        try:
            is_back = False
            if msg.text == '#返回':
                self.class_database_op.step_back(userId)
                is_back = True

            cur_step = self.class_database_op.get_user_current_step(userId)

            is_head = False
            if not cur_step or cur_step == '0' and not is_back:
                cur_step = str(msg.text)
                next_step = int(msg.text)
                is_head = True

            steps = cur_step.split('-')
            cur_step_action = self.command_dict
            for step in steps:
                next_step = int(step)
                cur_step_action = (cur_step_action[0])[int(step)]

            if type(cur_step_action[0]) is not type(self.step_process):
                if not is_head and not is_back:
                    next_step = int(msg.text)

                    if cur_step == '0':
                        cur_step = str(next_step)
                    else:
                        cur_step = cur_step + "-" + str(next_step)

                    cur_step_action = (cur_step_action[0])[next_step]

                self.class_database_op.set_user_current_step(userId, cur_step)

            if type(cur_step_action[0]) == type([]):
                # 菜单处理
                self.class_database_op.delete_redis_kv(userId, 'inputing')
                return cur_step_action[1](msg, userId)
            else:
                #动作处理
                if msg_opt:
                    return cur_step_action[0](msg, userId, msg_opt)
                else:
                    return cur_step_action[0](msg, userId)

        except Exception:
            import traceback
            traceback.print_exc()
            self.class_database_op.set_user_current_step(userId, '0')
            for single_key in self.config.fail_clean:
                self.class_database_op.delete_redis_kv(userId, single_key)

            itchat.send_msg("似乎。。。出现了一个错误。。。\n这一般是因为：\n[1]您的输入有误\n[2]系统在处理时发生了一个错误\n\n综上，已返回到首级菜单。", msg['FromUserName'])
            return self.step_0_tips(msg, userId)

    def under_constuct_tips(self, msg, userId):
        self.class_database_op.step_back(userId)
        return "功能开发中。。。"

    def __init__(self):
        self.buu_step6_menu = buu_step6_menu.class_step()

        self.command_dict = [  \
                                (self.step_0_tips, None), \
                                (self.step_mig, None), \
                                (self.step_mig, None), \
                                (self.step_mig, None), \
                                (self.step_mig, None), \
                                (self.under_constuct_tips, None), \
                                (self.buu_step6_menu.step_dict, self.buu_step6_menu.step_tips) \
                            ]

        self.command_dict = (self.command_dict, self.command_dict)

        self.config = buu_config.config
        self.class_database_op = buu_database.class_database_op()