12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196 |
- # -*- coding:utf-8 -*-
- import os
- from enum import Enum
- from BaseLog import CBaseLog
- from ExtraData import CExtraData
- from OptionExcel import COptionExcel
- from OptionConfig import COptionConfig
- from OptionFocus import COptionFocus
- from OptionOCR import COptionOCR
- from ssat_sdk.tv_detect import *
- from ssat_sdk.device_manage.capturecard_manager import CCardManager
- from ssat_sdk.utils.string_util import strcmp
- g_level = ['First', 'Second', 'Third', 'Fourth', 'Fifth', 'Sixth',
- 'Seventh', 'Eighth', 'Ninth', 'Tenth', 'Eleventh', 'Twelfth']
- def strSplit(text):
- ret = []
- str_int = ''
- str_ch = ''
- ch_last = ' '
- for ch in text:
- if 47 < ord(ch) < 58:
- str_int += ch
- if str_ch.__len__():
- ret.append(str_ch)
- str_ch = ''
- else:
- if 47 < ord(ch_last) < 58 and ch == '.':
- str_int += ch
- if str_ch.__len__():
- ret.append(str_ch)
- str_ch = ''
- else:
- str_ch += ch
- if str_int.__len__():
- ret.append(str_int)
- str_int = ''
- ch_last = ch
- if str_ch.__len__():
- ret.append(str_ch)
- if str_int.__len__():
- ret.append(str_int)
- return ret
- # 枚举:value类型
- class VType(Enum):
- SelectType = 0
- InputType = 1
- RangeType = 2
- # 注意:所有不对外暴露的变量和函数需要私有化,以明确哪些接口和参数是对外的。
- # 这样便于后期维护时,根据对外的变量和函数来做处理。
- class COptionAction(CBaseLog):
- # ==============设备对象============== #
- # 红老鼠遥控对象;
- __redRat3 = TvOperator()
- # 创建视频采集对象
- __captureCard = CCardManager()
- # 图片切割对象
- __imgCMP = ImageCMP()
- def __init__(self, optionName, optionValue, optionConfig, optionExcel):
- CBaseLog.__init__(self)
- # 层级位置;
- self.__pos = 0
- # 目标option;
- self.__optionName = optionName
- # __optionValue可空;
- self.__optionValue = optionValue
- self.__optionExcel = optionExcel
- self.__optionConfig = optionConfig
- # 焦点定位及文字识别;
- self.__optionFocus = COptionFocus(optionConfig)
- self.__optionOCR = COptionOCR(optionConfig, optionExcel)
- if self.__optionExcel is None:
- self.error(u"表格对象空")
- # ==============常用对象数据==============;
- self.__optionPaths = self.__optionExcel.getOptionPaths(self.__optionName)
- # 如果__optionValue空则不取value表任何内容;
- if self.__optionValue != "" or self.__optionValue is not None:
- self.__optionValueInfo = self.__optionExcel.getOptionValueInfo(self.__optionName, self.__optionValue)
- self.__optionInfo = self.__optionExcel.getOptionInfo(self.__optionName)
- # 当前状态下的变量,与__pos对应;
- self.__curOptionName = ''
- self.__curOptionInfo = None
- self.__prevOptionName = ''
- self.__prevOptionInfo = None
- # 到达value节点后,记录value值(一般只用于range() 数值记录,其他值无意义);
- self.__optionValueText = ""
- # 获取一次当前层级信息;
- self.getCurOptionInfo()
- # 相关返回操作;
- self.__isEnterKeyInValue = False # 节点在value表时,是否发送enter key.
- self.__valueType = VType.SelectType # 节点在value表时,value值的类型(0:选择型, 1:输入型,2:range()数值型。)
- @property
- def pos(self):
- return self.__pos
- @property
- def optionName(self):
- return self.__optionName
- @property
- def optionValue(self):
- return self.__optionValue
- @property
- def curOptionName(self):
- return self.__curOptionName
- @property
- def curOptionInfo(self):
- return self.__curOptionInfo
- @property
- def optionValueText(self):
- return self.__optionValueText
- @property
- def optionValueInfo(self):
- return self.__optionValueInfo
- @property
- def optionPaths(self):
- return self.__optionPaths
- '''
- 函数:截图并返回截图路径,如果当前option/optionValue含有layout坐标参数,则返回坐标截图;
- 参数:无
- 返回:截图路径;
- '''
- def takePicture(self):
- img = os.path.join(getSATTmpDIR(), "menutree_runpath.png")
- COptionAction.__captureCard.takePicture(img)
- if not os.path.exists(img):
- self.error(u"截图失败:%s" % img)
- retImg = img
- # 判断当前处于option层还是value层
- if self.__pos >= self.__optionPaths.__len__():
- curInfo = self.__optionValueInfo
- else:
- curInfo = self.__curOptionInfo
- # 判断当前层是否存在layout参数
- if "layout" in curInfo and 'bounds' in curInfo['layout']:
- if curInfo['layout']['bounds'] is "":
- self.info("当前option未配置layout参数")
- elif curInfo['layout']['bounds'].__len__() != 4:
- self.info("当前option的layout坐标配置异常:%s" % str(curInfo['layout']['bounds']))
- else:
- self.info("当前option的layout坐标为:%s" % str(curInfo['layout']['bounds']))
- retImg = os.path.join(getSATTmpDIR(), "menutree_runpath_layout.png")
- COptionAction.__imgCMP.saveCropPic(img, retImg, curInfo['layout']['bounds'])
- return retImg
- '''
- 函数:调用根节点快捷键(中间节点不需要快捷键;);
- 参数:无
- 返回:无
- '''
- def callFirstOptionShortCutKey(self):
- # 获取父节点等待时间(如果没找到,getParentWaitTime默认返回1)。
- waitTime = self.__optionConfig.getParentWaitTime(self.__optionPaths['First']['parent'])
- # 是否有另配置快捷按键代替根节点按键;
- if 'shortcut_key' in self.__optionPaths['First']:
- self.sendKey(self.__optionPaths['First']['shortcut_key'], 1, waitTime)
- else:
- self.sendKey(self.__optionPaths['First']['parent'], 1, waitTime)
- self.warn(u"表格没有shortcut_key字段,执行默认的parent按键:%s" % self.__optionPaths['First']['parent'])
- '''
- 函数:调用当前结点的toparent_key(叫back_key会简单点)
- 参数:
- curOptionName 当前层级的目标节点.
- 返回:无
- '''
- def callCurOptionBackKey(self, curOptionName):
- curOptionInfo = self.__optionExcel.getOptionInfo(curOptionName)
- if 'toparent_key' in self.__optionPaths[curOptionInfo['level']]:
- self.sendKey(self.__optionPaths[curOptionInfo['level']]['toparent_key'])
- else:
- self.error(u"表格没有toparent_key字段,执行默认按键return")
- self.sendKey('return')
- '''
- 函数:
- 参数:
- 返回:
- '''
- def inputUnlock(self, stdText, password=''):
- # 如果锁住,按ok会弹出输入密码框;
- self.sendKey("ok")
- # 获取密码;
- if password.__len__() == 0:
- password = self.__optionConfig.getSuperPassword("Password", "super")
- # 发送按键;
- for key in list(password):
- self.sendKey(key, 1, 0.2)
- time.sleep(1)
- # 发送ok键;
- self.sendKey('ok')
- # 判断是否成功输入密码;
- textPic = self.takePicture()
- # 遍历ocr类型;
- found = False
- ocrDict = [{"lan": "ChinesePRC+English", "type": 4}, {"lan": "ChinesePRC+English", "type": 253},
- {"lan": "ChinesePRC+English", "type": 10001}]
- for item in ocrDict:
- # 识别ocr;
- ocrText = self.__optionOCR.getImageText(textPic, item, {})
- ocrText = unicode(ocrText).lower()
- self.info("lan=%s,type=%d,ocr=%s" % (item["lan"], item["type"], ocrText))
- if ocrText in stdText or stdText == ocrText:
- found = True
- break
- return not found
- '''
- 函数:处理弹出密码框.
- 参数:
- 返回:Boolean, Boolean。
- 如何没有密码框处理,返回False,False。
- 如果有,且成功输入密码后密码框消失,返回True, True
-
- 注意:对应以前的dealOthers函数。
- '''
- def dealPasswordBox(self):
- parentOption = ''
- if self.isOnValueSheet():
- if self.__optionValueInfo is None or self.__optionValueInfo.__len__() == 0:
- self.error(u"当前value[%s]信息空" % str(self.__optionValue))
- return False, False
- others = self.__optionValueInfo['others']
- if others is None or others.__len__() == 0:
- self.info(u"[%s]others字段空" % str(self.__curOptionName))
- return False, False
- # 获取父节点
- parentOption = self.__optionValueInfo['option']
- else:
- getStatus, nextOptionInfo = self.getNextOptionInfo()
- if self.__curOptionInfo is None or self.__curOptionInfo.__len__() == 0:
- self.error(u"当前option[%s]信息空" % str(self.__curOptionName))
- return False, False
- others = self.__curOptionInfo['others']
- if others is None or others.__len__() == 0:
- self.info(u"[%s]others字段空" % str(self.__curOptionName))
- return False, False
- # 获取父节点;
- parentOption = self.__curOptionInfo['parent']
- # 转换为字典;
- others = json.loads(others)
- if "password" not in others:
- self.info(u"[%s]others没有密码框处理" % str(self.__curOptionName))
- return False, False
- password = self.__optionConfig.get_value("Password", others["password"])
- # 发送密码前,停2秒(因为像6586机芯响应很慢,密码框还没弹出就完成了密码输入的操作);
- time.sleep(2)
- # 发送按键;
- for key in list(password):
- self.sendKey(key, 1, 0.2)
- time.sleep(1)
- # 发送ok键;
- if not strcmp(others["enter_key"], "default"):
- self.sendKey(others["enter_key"])
- # 判断是否成功输入密码;
- current_uiPic = self.takePicture()
- # 此处findRectByIcon参数3传递的不是first_parent,而是当前option的parent;
- found, contourRect = self.__optionFocus.findFocusByIcon(current_uiPic, parentOption)
- return True, not found
- '''
- 函数:是否在父节点菜单上。一般在执行了callFirstOptionShortCutKey后调用;
- 参数:无
- 返回:Boolean, 数组(坐标)。 如:True, [0,0,1920,1080]
- 注意:由于所有父节点上的子项都共用一个图片定位参数,所以只要随意一个父节点的子项option即可获取定位参数;
- 测试:。
- '''
- def isOnFirstOption(self):
- pic = self.takePicture()
- return self.__optionFocus.findFocusByIcon(pic, self.__optionPaths['First']['option'])[0]
- '''
- 函数:是否在当前节点(移动后,判断是否移动到下一目标节点)上.
- 说明:
- 每次移动到下一目标节点(option)上时,self.__pos + 1,表示移动到下一层路径。
- 当self.__pos >= self.__optionPaths.__len__()时,表示到达value表格;
- 所以,该类的重点在self.__pos的移动;
- 参数:
- staticPic 静态图片路径。
- 主要用于将截图和聚焦、ocr识别分离,然后可以截图和聚焦中间添加其他处理。
- 返回:Boolean, 识别的文本/数字;
- 示例:
- '''
- def isOnTargetNode(self, staticPic=None):
- # 是否在value表中;
- isValueSheet = self.isOnValueSheet()
- self.info(u"当前层级在:%s" % ("value表" if isValueSheet else "路径表"))
- # 析出参数;
- if isValueSheet:
- curLevel = 'value'
- curParent = self.__optionPaths[g_level[self.__pos - 1]]['parent']
- curOption = self.__optionValueInfo['option']
- curOthers = self.__optionValueInfo['others']
- else:
- curLevel = g_level[self.__pos]
- curParent = self.__optionPaths[curLevel]['parent']
- curOption = self.__optionPaths[curLevel]['option']
- curOthers = self.__optionPaths[curLevel]['others']
- self.info("当前[%s]others=[%s]" % (curOption, curOthers))
- if curOthers.__len__() == 0:
- curOthers = {}
- else:
- curOthers = json.loads(curOthers)
- firstParent = self.__optionPaths['First']['parent']
- # 获取文本识别的参数;
- ocrConfigList = self.__optionConfig.getOptionOCRConfig(curOption)
- ocrThreshold = self.__optionConfig.getThresholdDict(firstParent) # 注意,此处使用firstParent;
- # 获取当前option的ocr值/value name下所有ocr值;
- if isValueSheet:
- if curOption.lower() == self.__optionName.lower():
- optionTextList = self.__optionExcel.getOptionValueText(curOption, self.__optionValue)
- else:
- optionTextList = self.__optionExcel.getOptionValueText(curOption)
- else:
- optionTextList = self.__optionExcel.getOptionText(curOption)
- # 获取option下所有兄弟项的ocr:option字典内容;
- siblingTextDict = self.__optionExcel.getOptionAllSiblingItemDict(curOption, not isValueSheet)
- # 获取所有option兄弟项(包括自己)的ocr值;
- siblingTextList = list(siblingTextDict.keys())
- # 是否获取数值文本;
- isNumberText = False
- # 如果是value表,且兄弟项文本为range
- # 注:value表中的option实际并没有兄弟项,取的是所有value项
- if isValueSheet and siblingTextList.__len__():
- if siblingTextList[0].startswith('range('):
- self.info(u"识别的内容是value表数字内容(range(x,y)类型)")
- isNumberText = True
- # 清除之前的value值;
- self.__optionValueText = ""
- # 是否为静态焦点识别(动态则为跑马灯);
- if curOthers.__len__() and 'marquee' in curOthers:
- return self.__getDynamicPicText(curOption, optionTextList, siblingTextList, ocrConfigList,
- ocrThreshold, curOthers['marquee'],
- isNumberText, isValueSheet)
- else:
- isFocus, isTarget, text = self.__getStaticPicText(self.takePicture() if staticPic is None else staticPic,
- curOption,
- optionTextList,
- siblingTextList,
- ocrConfigList,
- ocrThreshold,
- isNumberText, isValueSheet)
- return isTarget, text
- # endif
- '''
- 函数:是否移到目标节点上(在isOnOption后,判断__pos位置是否在__paths最后);
- 参数:无
- 返回:Boolean.
-
- '''
- def isOnTargetOption(self):
- return True if self.__pos == (self.__optionPaths.__len__() - 1) else False
- '''
- 函数:当前节点是否在value sheet层级中;
- 参数:无
- 返回:Boolean
- '''
- def isOnValueSheet(self):
- if self.__optionValue == "": # 该值空,表明不会移动到value sheet中;
- return False
- else:
- return True if self.__pos >= self.__optionPaths.__len__() else False
- '''
- 函数:移动到下一兄弟节点;
- 参数:无
- 返回:无
-
- 注意:
- sendKey的等待时间太短,会导致画面未响应,截图时还是截上一状态的,比如0.1秒就很经常出现这问题。
- 同时,按键等待时间,应该有所区分。
-
- 如果不截图,可以不考虑sendKey的等待时间.
- '''
- def move2NextSiblingNode(self):
- # 获取当前option信息;
- if self.getCurOptionInfo():
- # 析出move key;
- optionMoveKey = self.__curOptionInfo['option_move_key']
- if optionMoveKey.__len__() == 0:
- self.sendKey(self.__curOptionInfo['move_key'][1], 1, 1)
- else:
- self.sendKey(optionMoveKey[1], 1, 1)
- else:
- valueMoveKey = self.__optionValueInfo['move_key']
- self.sendKey(valueMoveKey[1], 1, 1)
- '''
- 函数:移动到上一兄弟节点
- 参数:无
- 返回:无
-
- 注意:
- sendKey的等待时间太短,会导致画面未响应,截图时还是截上一状态的,比如0.1秒就很经常出现这问题。
- 同时,按键等待时间,应该有所区分。
-
- 如果不截图,可以不考虑sendKey的等待时间.
- '''
- def move2PrevSiblingNode(self):
- # 获取当前option信息;
- if self.getCurOptionInfo():
- # 析出move key;
- optionMoveKey = self.__curOptionInfo['option_move_key']
- if optionMoveKey.__len__() == 0:
- self.sendKey(self.__curOptionInfo['move_key'][0], 1, 1)
- else:
- self.sendKey(optionMoveKey[0], 1, 1)
- else:
- valueMoveKey = self.__optionValueInfo['move_key']
- self.sendKey(valueMoveKey[0], 1, 1)
- '''
- 函数:返回到父节点
- 参数:
- 返回:无
- 注意:
- sendKey的等待时间太短,会导致画面未响应,截图时还是截上一状态的,比如0.1秒就很经常出现这问题。
- 同时,按键等待时间,应该有所区分。
- 如果不截图,可以不考虑sendKey的等待时间.
- '''
- def back2ParentNode(self):
- # 获取当前option信息;
- if self.getCurOptionInfo():
- tKeyDict = self.__curOptionInfo['toparent_key']
- peKeyDict = self.__prevOptionInfo['enter_key']
- else:
- tKeyDict = self.__optionValueInfo['toparent_key']
- peKeyDict = self.__prevOptionInfo['enter_key']
- '''
- 说明:
- toparent_key由于旧menutree无此参数,分以下数种情况:
- 当toparent_key没有配时:
- 是否为value层:
- 如果为value层,上一父层的enter_key是否为default:
- 如果是default,则toparent_key也按default处理;
- 如果不是default,则toparent_key按照return处理;
- 如果不为value层:
- toparent_key统一按return处理。
- 如果配了toparent_key:
- toparent_key是否为default:
- 如果是,按照default处理:
- 如果不是,按照配置的key来处理。
- tKeyDict中是否包含tolevel字段;
- 如果不包含,执行完toparent_key后self.__pos自减1
- 如果包含,执行完toparent_key后self.__pos等于tolevel指定的层级
- '''
- # 当toparent_key没有配时;
- if tKeyDict == "":
- # 是否为value层:
- if self.isOnValueSheet():
- # 如果为value层,上一父层的enter_key是否为default
- if strcmp(peKeyDict["key"][0], "default"):
- # 如果是default,则toparent_key也按default处理;
- self.info("当前层未配toparent_key,且父层enter_key为default,默认toparent_key为default")
- tKeyDict = {'key': ["default", ]}
- else:
- # 如果不是default,则toparent_key按照return处理;
- tKeyDict = {'key': ["return", ]}
- else:
- # 如果不为value层,toparent_key统一按return处理。
- tKeyDict = {'key': ["return", ]}
- # 如果配了toparent_key
- else:
- # toparent_key是否为default
- if strcmp(tKeyDict['key'][0], "default"):
- # 如果是default,则toparent_key也按default处理;
- self.info("当前层toparent_key为default")
- else:
- self.info("当前层toparent_key为%s" % str(tKeyDict['key']))
- self.__executeToParentKey(tKeyDict)
- '''
- 在忽略大小写的情况下,获取到该level对应的g_level的下标值
- '''
- def __getLevelIndex(self, level):
- s_level = []
- for lvStr in g_level:
- s_level.append(lvStr.lower())
- return s_level.index(level)
- '''
- 函数:进入当前节点,只对路径节点有效,value节点不处理;
- 参数:无
- 返回:无
- '''
- def enterNode(self):
- # 获取当前option信息;
- if self.getCurOptionInfo():
- # 析出enter key;
- # 优先使用option本身的enter_key
- optionEnterKey = self.__curOptionInfo['option_enter_key']
- # 如果option本身的enter_key为空,则使用该层parent的enter_key
- if optionEnterKey['key'] == "":
- optionEnterKey = self.__curOptionInfo['enter_key']
- self.__executeEnterKey(optionEnterKey)
- else:
- if not self.__isEnterKeyInValue:
- self.info(u"节点已在value上,且没有触发过enter key")
- enterKey = self.__optionValueInfo['enter_key']
- self.__executeEnterKey(enterKey)
- else:
- self.info(u"节点已在value上,已触发过enter key")
- '''
- 函数:设置当前节点位置;
- 参数:
- pos: 外部节点位置值。
- 返回:无
-
- 注意:此函数用于外部创建两个【路径具体子集关系】的COptionAction对象,才可以使用此函数。
- 假设, a对象路径{p,p1,p2,p3,p4, v1},b = {p, p1, p2, p3, p4, p5, p6, v2}, c = {p, p2, p5, p6, v3}
- 其中, v表示value值,不属于路径。那么,a和b具有子集关系, a与c或b与c都不具有子集关系。
-
- a移动到p4上,完成了v1设置后,a.back2ParentNode()后,此时如果要操作b并设置v2,就要b.SetCurPos(a.pos).
- '''
- def setCurPos(self, pos):
- if pos < 0 or pos > self.__optionPaths.__len__():
- self.error(u"pos值[%d]超出路径范围:[0-%d]" % (pos, self.__optionPaths.__len__()))
- return
- self.__pos = pos
- # 变更后要重新读一次当前option
- self.getCurOptionInfo()
- '''
- 函数:设置目标option的值, 只设置数值型value和输入型value(选择型value不需要此步骤).
- 参数:无
- 返回:无
-
- 注意:此函数必须是已聚焦在目标value节点上,否则无效。
- 重要:
- 在此函数enter后,UI是否返回到上一层父节点上,还是停留在本层节点不返回。
- 建议在excel中配置这个关键信息,以便此函数可以正确更改self.__pos的值。
-
- '''
- def setOptionValue(self):
- self.info(u"【在此函数enter后,UI是否返回到上一层父节点上,还是停留在本层节点不返回。\
- 建议在excel中配置这个关键信息,以便此函数可以正确更改self.__pos的值。】")
- if type(self.__optionValue) == str and self.__optionValue.__len__() == 0:
- self.error(u"[%s]的值为空,没有设置的值" % self.__optionName)
- return
- enterKey = self.__optionValueInfo['enter_key']
- moveKey = self.__optionValueInfo['move_key']
- valueText = self.__optionValueInfo['value_for_ocr']
- others = self.__optionValueInfo['others']
- if others.__len__():
- others = json.loads(others)
- else:
- others = {}
- # 是否有按键延时值;<注意:有些地方duration=0.1时,会变成长按的效果。>
- duration = float(others['duration']) if "duration" in others else 0.2
- # 值类型:
- # 0 表示默认选择类型.
- # 1 表示输入类型(有些输入完了,正确值会自动进入).
- # 2 表示进度条数据型.
- self.__valueType = VType.SelectType
- # 是否为数字文本(特指:range(0, 100));
- isNumberText = self.isNumberText(valueText)
- # 数值型value;
- if isNumberText:
- if moveKey[0] == 'input':
- # 标记值类型;
- self.__valueType = VType.InputType
- # 将数值转成字符;
- optionValue = self.__optionValue
- if type(optionValue) == int or type(optionValue) == float:
- optionValue = str(self.__optionValue)
- # 再将字符转成list;
- chList = list(optionValue)
- self.sendKey(chList, 1, duration)
- else:
- # 标记值类型;
- self.__valueType = VType.RangeType
- # 相差值;
- num = int(self.__optionValue) - int(self.__optionValueText)
- # 正->往右或下,负->往左或上;
- self.sendKey(moveKey[1] if num > 0 else moveKey[0], abs(num), duration)
- elif moveKey[0] == 'input':
- # 标记值类型;
- self.__valueType = VType.InputType
- # 将字符转成list;
- chList = list(self.__optionValue)
- self.sendKey(chList, 1, duration)
- # 最后,如果有进入键执行;
- if not strcmp(enterKey['key'][0], 'default') and enterKey['key'][0] != "":
- self.info(u"value节点具有enter key")
- self.__executeEnterKey(enterKey)
- '''
- 函数:获取当前层级的目标option详细信息.
- 参数:无
- 返回:Boolean, 获取成功返回True
- '''
- def getCurOptionInfo(self):
- if self.__optionPaths is None or self.__optionPaths.__len__() == 0:
- self.error(u"paths路径空")
- return False
- if self.__pos >= self.__optionPaths.__len__():
- # self.__curOptionInfo = None # 是否需要在到达value层级后置空,可能保留值会有比较多用处!
- self.warn(u"已到达value节点,无法获取路径信息")
- self.__prevOptionName = self.__curOptionName
- self.__prevOptionInfo = self.__curOptionInfo
- return False
- # 只有第一次或层级移动了才需要更新;
- if self.__curOptionInfo is None or self.__pos != self.__curOptionInfo['layers']:
- self.__curOptionName = self.__optionPaths[g_level[self.__pos]]['option']
- outResult, outData = self.__optionExcel.getOptionInfo(self.__curOptionName, self.__optionPaths)
- if outResult is False:
- return False
- self.__curOptionInfo = outData
- # 当处于第二层或以后的层级时,读取前一层的数据
- if self.__pos > 0:
- # 是否在value层
- if not self.isOnValueSheet():
- self.__prevOptionName = self.__optionPaths[g_level[self.__pos - 1]]['option']
- outResult, outData = self.__optionExcel.getOptionInfo(self.__prevOptionName, self.__optionPaths)
- if outResult is False:
- return False
- self.__prevOptionInfo = outData
- # 特殊情况:如果只一层且该层与value相同(信源source,要在value表中配置一份相同配置)。
- # if self.__curOptionInfo['first_parent'] == self.__curOptionName:
- # # 自动进入value表;
- # self.__pos += 1
- return True
- '''
- 函数:返回下一个option的详细信息;
- 参数:无
- 返回:Boolean、option info。
- '''
- def getNextOptionInfo(self):
- if self.__optionPaths is None or self.__optionPaths.__len__() == 0:
- self.error(u"paths路径空")
- return False, None
- if self.__pos + 1 >= self.__optionPaths.__len__():
- self.warn(u"已到达value节点,无法获取路径信息")
- return False, None
- nextOptionName = self.__optionPaths[g_level[self.__pos + 1]]['option']
- return self.__optionExcel.getOptionInfo(nextOptionName, self.__optionPaths)
- '''
- 函数:获取指定位置的option信息;
- 参数:
- pos 指定的层级位置;
- 返回:
- 成功返回option信息,失败返回None
- '''
- def getOptionInfo(self, pos):
- if pos > self.__optionPaths.__len__() and pos < 0:
- return None
- outResult, outData = self.__optionExcel.getOptionInfo(self.__optionPaths[g_level[pos]]['option'],
- self.__optionPaths)
- if outResult is False:
- return None
- return outData
- '''
- 函数:指定option是否存在该路径中;
- 参数:
- optionName 指定的option名称;
- 返回:
- 存在返回True+位置, 不存在返回False+位置;
- '''
- def isOptionInPaths(self, pos, optionNameOrValueName):
- bExist = False
- # 在路径表中;
- if 0 < pos < self.__optionPaths.__len__():
- posOptionName = self.__optionPaths[g_level[pos]]['option'].lower()
- siblingText = self.__optionExcel.getOptionAllSiblingItemName(posOptionName)
- for name in siblingText:
- if name.lower() == optionNameOrValueName.lower():
- bExist = True
- break
- # 在value表中;
- if pos == self.__optionPaths.__len__():
- valueOptionName = self.__optionPaths[g_level[self.__optionPaths.__len__() - 1]]['option']
- valueNames = self.__optionExcel.getOptionAllValueName(valueOptionName)
- for name in valueNames:
- if name.lower() == optionNameOrValueName.lower():
- bExist = True
- break
- return bExist
- '''
- 函数:检测路径是否有效;
- 参数:无
- 返回:Boolean, 检测正常返回True
- '''
- def checkRunOptionPath(self):
- outData = self.__optionExcel.checkOptionPaths(self.__optionPaths)
- if str(outData[1]) == 'NoExit':
- self.error(u"表格中不存在到达Option:[%s]的路径,在表格中排查到达该Option路径" % self.__optionName)
- return False
- if str(outData[1]) == 'Fail':
- self.error(u"表格中到达Option:[%s]的路径出现数据断层找不到First层级,在表格中排查到达该Option路径" % self.__optionName)
- return False
- return True
- '''
- 函数:指定的value_for_ocr或option_for_ocr数组是否为range(xx,xx)类型。
- 参数:
- textList value_for_ocr或option_for_ocr数组,一般只可能会是value_for_ocr
- 返回:Boolean, 如果是range(xx,xx)类型返回True.
- '''
- def isNumberText(self, textList):
- # 是否获取数值文本;
- isNumberText = False
- # 如果是value表,且兄弟项文本为range
- # 注:value表中的option实际并没有兄弟项,取的是所有value项
- if self.isOnValueSheet() and textList.__len__():
- if textList[0].startswith('range('):
- self.info(u"识别的内容是value表数字内容(range(0,xx)类型)")
- isNumberText = True
- return isNumberText
- '''
- 函数:获取静态图片文本内容
- 参数:(略,请看调用函数)
- 注意:
- 返回:Boolean、Boolean、文本识别内容。
- 是否成功聚焦、是否聚焦在目标节点上、聚焦框识别的文本内容。
- '''
- def __getStaticPicText(self, pic, optionName, optionTextList, siblingTextList, ocrConfigList, ocrThreshold,
- isNumberText, isValueSheet, aliveKey=None):
- # 获取图片焦点框;
- found, focusBox = self.__optionFocus.findFocusByIcon(pic, optionName, isValueSheet)
- if found is False:
- self.debug(u"未找到[%s]聚集框" % optionName)
- return False, False, None
- # 如果有鲜活键;
- self.sendAliveKey(aliveKey)
- # 获取文本区域框;
- textBox = self.__optionFocus.getFocusTextBox(optionName, focusBox, isValueSheet)
- # 配置文本图片路径,保存文本区域截图;
- text_pic = os.path.join(getSATTmpDIR(), "meuttree_area_text.png")
- self.__imgCMP.saveCropPic(pic, text_pic, textBox)
- if not os.path.exists(text_pic):
- self.error(u"%s截取文本图片失败:%s" % (optionName, text_pic))
- return False, False, None
- # 是否在某个兄弟项中;
- isOnSibling = False
- # 遍历所有ocr识别选项;
- for ocrConfig in ocrConfigList:
- # 如果有鲜活键;
- self.sendAliveKey(aliveKey)
- # 识别出当前聚焦文本;
- curFocusText = self.__optionOCR.getImageText(text_pic, ocrConfig, ocrThreshold)
- # 判断识别文本是来正常;
- if curFocusText == "ERR<Exp>" or curFocusText.__len__() == 0:
- continue
- # 转成小写;
- curFocusText = curFocusText.lower()
- self.info("[%s]当前识别出的文本=%s" % (optionName, curFocusText))
- # 是否取数字文本(肯定在value节点上);
- if isNumberText is True:
- # 特殊情况处理:某些情况下,会将包含数字以外的区域一起识别;
- curFocusText = curFocusText.strip('>')
- # 将数字分组
- numberTextList = strSplit(curFocusText)
- # 只判断最后一位是否为数字;
- if numberTextList.__len__() < 1:
- self.error(u"当前识别的文本不是数字文本:%s" % curFocusText)
- continue
- try:
- numberText = float(numberTextList[numberTextList.__len__() - 1])
- # 记录value值;
- self.__optionValueText = numberText
- return True, True, numberText
- except Exception:
- continue
- else:
- # 当前option识别的文本与所有兄弟项文本比较;
- for siblingText in siblingTextList:
- # 转为小写,保证所有比较都是小写;
- siblingText = siblingText.lower()
- # 兄弟项文本是否被包含在curFocusText中或相同;
- if siblingText.lower() in curFocusText.lower() or strcmp(siblingText, curFocusText):
- isOnSibling = True
- self.info(u"当前焦点在[%s], 目标焦点为[%s]" % (siblingText, optionName))
- self.info(u"optionTextList:%s" % optionTextList)
- # 再判断,该兄弟项是否为目标节点(curOption);
- for optionText in optionTextList:
- optionText = optionText.lower()
- # 若当前兄弟项为目标option返回True、文本;
- if strcmp(optionText, siblingText):
- return True, True, curFocusText
- # endif
- # endfor
- # 在兄弟项中,退出循环;
- break
- # endif
- # endfor
- if isOnSibling is False:
- self.error(u"未聚集到任何[%s]的兄弟项中" % optionName)
- else:
- self.info("未聚集到目标节点[%s],当前文本=%s" % (optionName, curFocusText))
- return found, False, curFocusText
- # endif
- # endfor
- # 默认返回;
- return found, False, 0 if isNumberText else ""
- '''
- 函数:获取动态图片文本内容
- 参数:(略,请看调用函数)
- 注意:
- 返回:Boolean、文本识别内容。成功识别出文本,返回True及文本内容。
- '''
- def __getDynamicPicText(self, optionName, optionTextList, siblingTextList, ocrConfigList, ocrThreshold, marqueeDict,
- isNumberText, isValueSheet):
- # 判断图片是否动态:截图两次,判断两次文本内容是否相同;
- firstFocus, firstTarget, firstText = self.__getStaticPicText(self.takePicture(), optionName, optionTextList,
- siblingTextList,
- [{"lan": "ChinesePRC+English", "type": 10001}],
- ocrThreshold, isNumberText, isValueSheet)
- if firstFocus is False:
- self.error(u"[%s]第一次截图未识别出聚焦框" % optionName)
- return False, None
- # 发送鲜活键, 保证界面鲜活;
- self.sendAliveKey(marqueeDict['alive_key'])
- # 第二次截图;
- secondFocus, secondTarget, secondText = self.__getStaticPicText(self.takePicture(), optionName, optionTextList,
- siblingTextList,
- [{"lan": "ChinesePRC+English", "type": 10001}],
- ocrThreshold, isNumberText, isValueSheet)
- if secondFocus is False:
- self.error(u"[%s]第二次截图未识别出聚焦框" % optionName)
- return False, None
- # 发送鲜活键, 保证界面鲜活;
- self.sendAliveKey(marqueeDict['alive_key'])
- # 比较两文本是否相同;
- if firstText.__len__() and firstText == secondText:
- self.info(u"截图两次,文本(%s)识别相同,聚焦的不是跑马灯Option" % firstText)
- return False, firstText
- # 文本不相同,为动态图片;
- menuList = marqueeDict['menu']
- # 如果只有一项跑马灯,且目标option亦在跑马灯列表中,则直接返回成功结果
- if menuList.__len__() == 1 and (optionName in menuList):
- self.info(u"该层菜单只有一项跑马灯, 找到即成功返回")
- return True, firstText
- picList = []
- # 如果有多项同级option都是跑马灯, 要成功识别文本需要间隔截图5次(大概会成功截图到最全的文本);
- for i in range(0, 5):
- picList.append(self.takePicture())
- # 间隔多久截图;
- time.sleep(marqueeDict['sleep_time'])
- # 发送鲜活键;
- self.sendAliveKey(marqueeDict['alive_key'])
- ocrTextList = []
- # 对截图进行文本识别分析;
- for pic in picList:
- isFocus, isTarget, text = self.__getStaticPicText(pic, optionName, optionTextList, siblingTextList,
- ocrConfigList,
- ocrThreshold, isNumberText, isValueSheet,
- marqueeDict['alive_key'])
- if isTarget is True:
- ocrTextList.append(text)
- # 发送鲜活键;
- self.sendAliveKey(marqueeDict['alive_key'])
- # 过滤重复的字符;
- ocrTextList = self.__removeDuplicateString(ocrTextList)
- self.info(u"识别到的跑马灯ocr文字列表:%s" % ocrTextList)
- # 获取动态文本的option字典;
- dynamicOptionOcrDict = self.__getOptionInfoDict(menuList)
- self.info(u"获取到的跑马灯Option对应的ocr字典:%s" % dynamicOptionOcrDict)
- # 遍历:识别结果与xls结果进行比较;
- for dynamicOption in dynamicOptionOcrDict:
- dynamicOptionOcrList = dynamicOptionOcrDict[dynamicOption]
- for dynamicOptionOcr in dynamicOptionOcrList:
- # 只要有3张满足,判断找到option;
- count = 0
- for ocrText in ocrTextList:
- if ocrText.lower() in dynamicOptionOcr:
- count += 1
- if count >= 3 and optionName == dynamicOption:
- return True, ocrText
- else:
- self.info(u"当前聚焦的跑马灯实际为:%s" % dynamicOption)
- return False, ocrText
- # endfor
- # endfor
- self.info("未能识别到当前聚焦的跑马灯Option")
- return False, 0 if isNumberText else None
- '''
- 函数:获取option名称数组内所有option的详细信息.
- 参数:
- optionNameList option名称数组。
- 返回:字典。
- {
- "option1": {self.__optionExcel.getOptionInfo(option1)[1]},
- "option2": {self.__optionExcel.getOptionInfo(option2)[1]},
- "option3": {self.__optionExcel.getOptionInfo(option3)[1]},
- }
- '''
- def __getOptionInfoDict(self, optionNameList):
- OptionInfoDict = {}
- for optionName in optionNameList:
- found, optionDict = self.__optionExcel.getOptionInfo(optionName)
- if found:
- OptionInfoDict[optionName] = optionDict
- # endif
- # endfor
- return OptionInfoDict
- '''
- 函数:找到两个字符串左边或者右边相同的部分
- 参数:
- str1:
- str2:
- direction 方向,默认为right
- 返回:
- '''
- def __findDuplicateString(self, str1, str2, direction="right"):
- index = 0
- if direction == "right":
- while True:
- index -= 1
- if abs(index) > str1.__len__() or abs(index) > str2.__len__():
- break
- if not str1[index] == str2[index]:
- break
- if index == -1:
- self.info(u"没有找到重复文本")
- return ""
- return str1[index + 1:]
- elif direction == "left":
- while True:
- if abs(index) >= str1.__len__() or abs(index) >= str2.__len__():
- break
- if not str1[index] == str2[index]:
- break
- index += 1
- return str1[:index]
- '''
- 函数:获取路径长度。
- 参数:
- 返回:
- '''
- def getPathLength(self):
- if self.__optionPaths is None:
- self.error(u"路径空,返回0")
- return 0
- self.info(u"路径长度:%d" % self.__optionPaths.__len__())
- return self.__optionPaths.__len__()
- '''
- 函数:去掉字符串数组中每个字符串 左边或右边相同的部分
- 参数:
- strList
- 返回:
- '''
- def __removeDuplicateString(self, strList):
- finishedList = strList
- directionList = ["left", "right"]
- for direction in directionList:
- same_str = self.__findDuplicateString(strList[0], strList[1], direction)
- if same_str == "":
- continue
- else:
- for i in range(2, strList.__len__()):
- same_str = self.__findDuplicateString(same_str, strList[i], direction)
- if same_str == "":
- break
- if same_str != "":
- finishedList = []
- for text in strList:
- if direction == "left":
- text = str[same_str.__len__():]
- else:
- text = str[:-same_str.__len__()]
- finishedList.append(text)
- # endfor
- # endif
- # endif
- # endfor
- return finishedList
- '''
- 函数:发送红老鼠按键;
- 参数:
- key 1、如果是字符串时,当作单个按键; 2、如果是list时,当作多个按键;
- count 执行多少次key;
- wait 1、执行单个key后,等待时长(因为电视机响应遥控需要时间);
- 2、执行list多个key后,每个key的等待时间;
- 返回:无
- '''
- def sendKey(self, key, count=1, duration=1.0):
- if key is not None and key.__len__() > 0:
- if type(key) == list:
- for k in key:
- # 清除前后空格;
- k = k.lstrip()
- k = k.rstrip()
- COptionAction.__redRat3.sendKey(k, 1, duration)
- else:
- key = str(key)
- # 清除前后空格;
- key = key.lstrip()
- key = key.rstrip()
- COptionAction.__redRat3.sendKey(key, count, duration)
- else:
- self.error(u"error:按键内容空")
- '''
- 函数:发送鲜活键;
- 参数:
- aliveKey 鲜活按键;
- 返回:无
- 注意:鲜活键等待时间是0.1秒,因为不考虑截图。
- '''
- def sendAliveKey(self, aliveKey):
- self.sendKey(aliveKey, 1, 0.1)
- '''
- 函数:按照新格式执行enter_key。
- 参数:
- eKeyDict 即enterKey的字典对象,包含的wait和duration等参数。
- wait,duration是为了兼容旧menutree按照others配置参数的处理方法。
- 如果others中有配waitTime和duration,优先采用others里的配置,否则则采用enterKey里的配置。
- '''
- def __executeEnterKey(self, eKeyDict):
- # 先从config读取
- wait = self.__optionConfig.getParentWaitTime(self.__curOptionInfo['parent'])
- # config中读取不到则尝试从others中读取
- if wait == 1.0:
- try:
- others = json.loads(self.__curOptionInfo['others'])
- if "waitTime" in others:
- if others['waitTime'] != "":
- wait = float(others['waitTime'])
- except Exception, e:
- pass
- # 读出来的enter_key,不为空或者default时,按格式执行enter_key
- if not strcmp(eKeyDict['key'][0], 'default') and eKeyDict['key'][0] != "":
- self.__executeKey(eKeyDict, wait=wait)
- if eKeyDict['tolevel'] != "":
- level = eKeyDict['tolevel']
- else:
- level = ""
- # value层的enter_key存在特殊字段,isback。用于判断执行enter_key以后会不会返回父层。
- # isback为0或为""时:
- # 则认为不返回父层;
- # 为1时:
- # 如果enter_key中没有配tolevel,则默认返回上一层父层;
- # 如果有,则默认返回到tolevel指定层级的父层。
- # 非value层,则self.__pos默认进入下一层。
- if self.isOnValueSheet():
- if eKeyDict['isback'] == "" or int(eKeyDict['isback']) == 0:
- pass
- else:
- if level == "":
- self.__pos -= 1
- else:
- self.__pos = self.__getLevelIndex(level) + 1
- else:
- self.__pos += 1
- # 重新获取次信息;
- self.getCurOptionInfo()
- def __executeToParentKey(self, tKeyDict):
- self.__executeKey(tKeyDict)
- # 层级变化
- if 'tolevel' in tKeyDict and tKeyDict['tolevel'] != "":
- level = tKeyDict['tolevel']
- self.__pos = self.__getLevelIndex(level) + 1
- else:
- self.__pos -= 1
- # 重新获取次信息;
- self.getCurOptionInfo()
- def __executeKey(self, keyDict, wait=1.0, duration=1.0):
- if wait != 1.0 and keyDict['wait'] != "":
- wait = float(keyDict['wait'])
- if duration != 1.0 and keyDict['duration'] != "":
- duration = float(keyDict['duration'])
- for key in keyDict['key']:
- self.sendKey(key, 1, duration)
- time.sleep(wait)
- def executeKey(self, keyDict, wait=1.0, duration=1.0):
- self.__executeKey(keyDict, wait, duration)
- def setValue(self, optionValue):
- self.__optionValue = optionValue
- if self.__optionValue != "" or self.__optionValue is not None:
- self.__optionValueInfo = self.__optionExcel.getOptionValueInfo(self.__optionName, self.__optionValue)
- return True if self.__optionValueInfo and self.__optionValueInfo.__len__() else False
- if __name__ == "__main__":
- exData = CExtraData()
- optionExcel = COptionExcel(exData)
- optionConfig = COptionConfig(exData, optionExcel)
- optionAction = COptionAction('source', None, optionConfig, optionExcel)
- # print optionAction.optionValueInfo
- print optionAction.curOptionInfo
- # ====================================== #
- # optionAction.callFirstOptionShortCutKey()
|