scbc_repos/ssat_sdk/MenuTree3/OptionConfig.py

# -*- coding:utf-8 -*-
import os
from UIT_PathManage import UITPathManage
from BaseLog import CBaseLog
from TConfig import TConfig
from ExtraData import CExtraData
from OptionExcel import COptionExcel


g_level = ['First', 'Second', 'Third', 'Fourth', 'Fifth', 'Sixth',
           'Seventh', 'Eighth', 'Ninth', 'Tenth', 'Eleventh', 'Twelfth']


# 注意：所有不对外暴露的变量和函数需要私有化，以明确哪些接口和参数是对外的。
# 这样便于后期维护时，根据对外的变量和函数来做处理。
class COptionConfig(TConfig, CBaseLog):
    def __init__(self, exData, optionExcel):
        CBaseLog.__init__(self)
        self.__exData = exData
        self.__optionExcel = optionExcel
        if self.__optionExcel is None:
            self.error(u"路径参数错误：None")
        # 状态：False表示路径值失败;
        self.status = False
        # menuTree目录;
        self.__uiTreeDir = self.__exData.menuTreeDir
        # 判断是否为目录;
        if not os.path.isdir(self.__uiTreeDir):
            self.error(u"%s 不是menuTree目录" % self.__uiTreeDir)
        # 判断目录是否存在;
        if not os.path.exists(self.__uiTreeDir):
            self.error(u"menuTree（%s）目录不存在" % self.__uiTreeDir)
        # 判断配置文件是否存在;
        self.__configPath = os.path.join(self.__uiTreeDir, "menutree.ini")
        if os.path.exists(self.__configPath):
            self.status = True
        else:
            self.error(u"menutree配置文件不存在：%s" % self.__configPath)
        # 初始化父类;
        TConfig.__init__(self, self.__configPath)

    # 获取超级密码;
    def getSuperPassword(self):
        return self.get_value("Password", "super")

    # 获取普通密码;
    def getOrdinaryPassword(self):
        return self.get_value("Password", "ordinary")

    # 获取父节点等待进入子节点时间;
    def getParentWaitTime(self, parent):
        optionName = parent + "WaitTime"
        try:
            waitTime = float(self.get_value("waitTime", optionName))
        except Exception:
            waitTime = 1.0
            self.error(u"获取%s界面的等待界面时间失败，使用默认值1.0s" % str(parent))
        return waitTime

    # 获取阀值字典;
    def getThresholdDict(self, optionName):
        section = "Threshold"
        thresholdDict = {}
        value = self.get_value(section, optionName)
        self.info("value:%s,%s" % (value, type(value)))
        if not value:
            return thresholdDict
        value_str = str(value)
        self.info("value_str:%s,%s" % (value_str, type(value_str)))
        value_dict = eval(value_str)
        self.info("value_dict:%s,%s" % (value_dict, type(value_dict)))
        return value_dict

    '''
    函数：获取Option的图片配置，方式1
    参数：
    返回：Boolean, 字典
    示例：True, {"icon_path": "", "dcfg": {}, "dir_path": ""}
    
    说明：
        查找顺序：
            [当前父节点][当前层级][当前节点] ->[当前父节点][当前层级] ->[根节点][当前层级][当前节点]
            [curParent][curLevel][curOption] ->[curParent][curLevel] ->[firstParent][curLevel][curOption]
    '''
    def __getICONConfig1(self, optionName, is_value_sheet=False):
        # 默认值;
        def_cfg = {"icon_path": "", "dcfg": {}, "dir_path": ""}
        paths = self.__optionExcel.getOptionPaths(optionName)
        # 判断路径节点是否空;
        if paths.__len__() == 0:
            self.error(u"当前【%s】的路径节点空，使用默认的配置值%s" % (optionName, def_cfg))
            return False, def_cfg

        # 获取first parent;
        first_parent = paths[g_level[0]]['parent']
        # 获取当前option的level;
        cur_level = g_level[paths.__len__() - 1]
        # 当前option的父节点名称;
        cur_parent = paths[cur_level]['parent']
        if is_value_sheet is True:
            cur_level = "value"

        # option聚焦的首配图片路径;
        icon_path = os.path.join(self.__uiTreeDir, "icon\\", cur_parent + "." + cur_level + "_" + optionName + ".png")
        # option聚焦的dir首配图片路径;
        icon_dir_path = os.path.join(self.__uiTreeDir, "icon\\" + cur_parent + "." + cur_level + "_" + optionName + ".dir.png")
        # 图片聚焦时的定位参数;
        opc_cfg = self.get_value_dict(cur_level, cur_parent + '.' + optionName)

        # 首配图片判断是否存在,不存在取用次配图片路径;
        if not os.path.exists(icon_path):
            # 使用次配图片，父级配图;
            self.warn(u"Option(%s)首配图片不存在：%s" % (optionName, icon_path))
            icon_path = os.path.join(self.__uiTreeDir, "icon\\" + cur_parent + "." + cur_level + ".png")
            icon_dir_path = os.path.join(self.__uiTreeDir, "icon\\" + cur_parent + "." + cur_level + ".dir.png")
            opc_cfg = self.get_value_dict(cur_level, cur_parent)
            self.warn(u"Option(%s)首配图片不存在，尝试使用次配图(%s)：%s，配置%s" % (optionName, cur_parent, icon_path, opc_cfg))
            # 如果次配都不存在，使用顶层配图;
            if not os.path.exists(icon_path):
                self.warn(u"Option(%s)次配图片不存在：%s" % (optionName, icon_path))
                # 使用顶层配图（first parent）
                icon_path = os.path.join(self.__uiTreeDir, "icon\\" + first_parent + "." + cur_level + "_" + optionName + ".png")
                icon_dir_path = os.path.join(self.__uiTreeDir, "icon\\" + first_parent + "." + cur_level + "_" + optionName + ".dir.png")
                opc_cfg = self.get_value_dict(cur_level, first_parent)
                self.warn(u"Option(%s)次配图片不存在，尝试使用顶层配图(%s)：%s，配置%s" % (optionName, first_parent, icon_path, opc_cfg))
                # 如果顶层配图不存在，退出;
                if not os.path.exists(icon_path):
                    self.error(u"%s对应的顶层菜单配图不存在%s，使用默认配置%s" % (optionName, icon_path, def_cfg))
                    return False, def_cfg
                # endif
            # endif
        # endif

        if opc_cfg.__len__() == 0:
            opc_cfg = {"offset": 20, "minPeri": 0, "maxPeri": 0, "minArea": 0, "maxArea": 0, "morphology": []}

        return True, {"icon_path": icon_path, "dcfg": opc_cfg, "dir_path": icon_dir_path}

    '''
    函数：获取Option的图片配置，方式2
    参数：
    返回：Boolean, 字典
    示例：True, {"icon_path": "", "dcfg": {}, "dir_path": ""}

    说明：
        查找顺序：
            [根节点][当前层级][当前节点] ->[根节点][当前层级] ->[根节点][根层级]
            [firstParent][curLevel][curOption] ->[firstParent][curLevel] ->[firstParent][firstLevel]
    '''
    def __getICONConfig2(self, optionName, is_value_sheet=False):
        # 默认值;
        def_cfg = {"icon_path": "", "dcfg": {}, "dir_path": ""}
        paths = self.__optionExcel.getOptionPaths(optionName)
        # 判断路径节点是否空;
        if paths.__len__() == 0:
            self.error(u"当前【%s】的路径节点空，使用默认的配置值%s" % (optionName, def_cfg))
            return False, def_cfg

        # 获取first parent;
        first_parent = paths[g_level[0]]['parent']
        # 获取当前option的level;
        cur_level = g_level[paths.__len__() - 1]
        # 当前option的父节点名称;
        cur_parent = paths[cur_level]['parent']
        if is_value_sheet is True:
            cur_level = "value"

        # option聚焦的首配图片路径;
        icon_path = os.path.join(self.__uiTreeDir, "icon\\",
                                 first_parent + "." + cur_level + "_" + optionName + ".png")
        # option聚焦的dir首配图片路径;
        icon_dir_path = os.path.join(self.__uiTreeDir,
                                     "icon\\" + first_parent + "." + cur_level + "_" + optionName + ".dir.png")
        # 图片聚焦时的定位参数;
        opc_cfg = self.get_value_dict(cur_level, first_parent)

        # 首配图片判断是否存在,不存在取用次配图片路径;
        if not os.path.exists(icon_path):
            # 使用次配图片，父级配图;
            self.warn(u"Option(%s)首配图片不存在：%s" % (optionName, icon_path))
            icon_path = os.path.join(self.__uiTreeDir, "icon\\" + first_parent + "." + cur_level + ".png")
            icon_dir_path = os.path.join(self.__uiTreeDir, "icon\\" + first_parent + "." + cur_level + ".dir.png")
            self.warn(u"Option(%s)首配图片不存在，尝试使用次配图(%s)：%s，配置%s" % (optionName, cur_parent, icon_path, opc_cfg))
            # 如果次配都不存在，使用顶层配图;
            if not os.path.exists(icon_path):
                self.warn(u"Option(%s)次配图片不存在：%s" % (optionName, icon_path))
                # 使用顶层配图（first parent）
                icon_path = os.path.join(self.__uiTreeDir, "icon\\" + first_parent + "." + "First.png")
                icon_dir_path = os.path.join(self.__uiTreeDir, "icon\\" + first_parent + "." + "First.dir.png")
                self.warn(
                    u"Option(%s)次配图片不存在，尝试使用顶层配图(%s)：%s，配置%s" % (optionName, first_parent, icon_path, opc_cfg))
                # 如果顶层配图不存在，退出;
                if not os.path.exists(icon_path):
                    self.error(u"%s对应的顶层菜单配图不存在%s，使用默认配置%s" % (optionName, icon_path, def_cfg))
                    return False, def_cfg
                # endif
            # endif
        # endif

        if opc_cfg.__len__() == 0:
            opc_cfg = {"offset": 20, "minPeri": 0, "maxPeri": 0, "minArea": 0, "maxArea": 0, "morphology": []}

        return True, {"icon_path": icon_path, "dcfg": opc_cfg, "dir_path": icon_dir_path}

    # 获取Option的图片配置;
    def getOptionICONConfig(self, optionName):
        outResutl, outData = self.__getICONConfig1(optionName, False)
        if outResutl is False:
            return self.__getICONConfig2(optionName, False)

        return outResutl, outData

    # 获取Value的图片配置;
    def getValueICONConfig(self, valueName):
        outResutl, outData = self.__getICONConfig1(valueName, True)
        if outResutl is False:
            return self.__getICONConfig2(valueName, True)

        return outResutl, outData

    # 获取Option的OCR配置;
    def getOptionOCRConfig(self, optionName):
        ocr_dict = []
        # 默认的ocr配置;
        def_orc = [{"lan": "ChinesePRC+English", "type": 4}, {"lan": "ChinesePRC+English", "type": 253}, {"lan": "ChinesePRC+English", "type": 10001}]
        paths = self.__optionExcel.getOptionPaths(optionName)
        # 判断路径节点是否空;
        if paths.__len__() == 0:
            self.error(u"当前【%s】的路径节点空，将使用默认的ocr配置%s" % (optionName, def_orc))
        else:
            # 如果状态False,退出;
            if self.status is False:
                self.warn(u"配置文件(%s)不存在,%s使用默认的ocr配置%s" % (self.__configPath, optionName, def_orc))
            else:
                # 获取当前option的level;
                cur_level = g_level[paths.__len__() - 1]
                # 获取first parent;
                first_parent = paths[g_level[0]]['parent']
                # 当前option的父节点名称;
                cur_parent = paths[cur_level]['parent']

                # 是否有当前option的ocr配置;
                if self.has_option(cur_level, optionName + '.ocr'):
                    ocr_dict = self.get_dict(self.get_value(cur_level, optionName + '.ocr'))
                    self.warn(u"%s使用自身的ocr配置%s" % (optionName, ocr_dict))
                else:
                    # 如果option本身没有配置,获取其父节点的配置;
                    if self.has_option(cur_level, cur_parent + '.ocr'):
                        ocr_dict = self.get_dict(self.get_value(cur_level, cur_parent + '.ocr'))
                        self.warn(u"%s使用父节点%s的ocr配置%s" % (optionName, cur_parent, ocr_dict))
                    else:
                        # 如果option父级没有配置,则获取顶层配置(first parent);
                        if self.has_option(cur_level, first_parent + '.ocr'):
                            ocr_dict = self.get_dict(self.get_value(cur_level, first_parent + '.ocr'))
                            self.warn(u"%s使用顶层节点%s的ocr配置%s" % (optionName, first_parent, ocr_dict))
                    #endif
                #endif
            # end-if
        # end-if

        if ocr_dict.__len__() == 0:
            ocr_dict = def_orc
            self.warn(u"无有效的ocr配置，将使用默认的ocr配置%s" % def_orc)

        self.info(u"%s使用的ocr配置=%s" % (optionName, ocr_dict))

        return ocr_dict

    # 获取ICON源屏幕的分辨率配置值;
    # 该值用途：由于有些icon是在某些分辨率下截图的，但是同机芯项目可能使用同一套UI，只是分辨率不一样而已。
    # 所以，根据当初的icon分辨率，可以计算出在别的分辨率下的icon区域大小;
    def getICONResolutionConfig(self):
        return self.get_value_dict('Screen', 'shape') if self.has_option('Screen', 'shape') else [1920, 1080]


if __name__ == "__main__":
    exData = CExtraData()
    optionExcel = COptionExcel(exData)
    optionConfig = COptionConfig(exData, optionExcel)
    # print "getSuperPassword", optionConfig.getSuperPassword()
    # print "getOrdinaryPassword", optionConfig.getOrdinaryPassword()
    # print "getParentWaitTime", optionConfig.getParentWaitTime('setting')
    # print "getThresholdDict", optionConfig.getThresholdDict('setting')
    # print "getOptionICONConfig", optionConfig.getOptionICONConfig('picture')
    print "getOptionICONConfig", optionConfig.getOptionICONConfig('picture_preset')
    # print "getValueICONConfig", optionConfig.getValueICONConfig('picture')
    # print "getOptionOCRConfig", optionConfig.getOptionOCRConfig('picture')
    # print "getICONResolutionConfig", optionConfig.getICONResolutionConfig()