# -*- coding:utf-8 -*-
import os
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
# 获取阀值字典;
'''
函数：获取指定option的阀值字典
参数：
optionName 节点名称;
返回：字典。
'''
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")
if opc_cfg.__len__() == 0:
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")
if opc_cfg.__len__() == 0:
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的图片配置
参数：
optionName 要获取的option。
返回：Boolean, {}
'''
def getOptionICONConfig(self, optionName):
outResutl, outData = self.__getICONConfig1(optionName, False)
if outResutl is False:
return self.__getICONConfig2(optionName, False)
return outResutl, outData
'''
函数：获取Value的图片配置
参数：
valueName 要获取的value名称。
返回：Boolean, {}
'''
def getValueICONConfig(self, valueName):
outResutl, outData = self.__getICONConfig1(valueName, True)
if outResutl is False:
return self.__getICONConfig2(valueName, True)
return outResutl, outData
'''
函数：获取Option的OCR配置
参数：
optionName 要获取的option名称
返回：字典
'''
def getOptionOCRConfig(self, optionName):
ocr_dict = []
# 默认的ocr配置;
def_orc = [{"lan": "ChinesePRC+English", "type": 10000},{"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("First", first_parent + '.ocr'):
ocr_dict = self.get_dict(self.get_value("First", 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()