j\dddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z mZmZmZddlmZddlmZddlmZddlmZddlmZdd lmZmZmZej>e Z!Gd d e"Z#Gd d e"Z$de%de%fdZ&de%de%fdZ'dee%e(fde(fdZ)de%de*fdZ+de%de%fdZ,de%de%fdZ-Gdde%Z.GddZ/Gdde0Z1Gdd e%ejdZ3Gd!d"ejhZ5Gd#d$e5Z6Gd%d&e6Z7Gd'd(e0Z8Gd)d*e8Z9y)+)absolute_importN)AnyCallableDictListOptionalTupleUnion)etree)Image) Direction)AbstractXPathBasedDevice)XPathElementNotFoundError) deprecated inject_callswipe_in_boundsc eZdZy)TimeoutExceptionN)__name__ __module__ __qualname__S/var/www/api.ozpay.ru/ozpay/venv/lib/python3.12/site-packages/uiautomator2/xpath.pyrrsrrceZdZdZy) XPathErrorzbasic error for xpath pluginN)rrr__doc__rrrrrs&rrsreturnc|j}tjdd|}tjdd|}tjdd|}|S)Nz[$@#&].z\.+z^\.|\.$)stripresubrs r safe_xmlstrr'#sH  A xa A vsAA {B"A Hrc$dj|S)zquick way to quote stringz{!r})formatr&s r string_quoter*,s == rvct|tr|St|tr|jdSt dt ||)Nzutf-8 Invalid type) isinstancebytesstrencode ValueErrortype)r+s r str2bytesr41s=!U!Sxx  ^T!Wa 00rxpath_expressioncb tj|y#tj$rYywxYw)NTF)r XPathXPathSyntaxError)r5s ris_xpath_syntax_okr99s/ $%  ! !s ..cj|jd}|ddjd|ddDz}|S)zl Convert a string from kebab-case to camelCase. Example: "hello-world" -> "helloWorld" -rr"c3<K|]}|jywN) capitalize).0parts r z(convert_to_camel_case..Js'Pd(9'PsN)splitjoin)rpartscamel_case_strs rconvert_to_camel_caserGAs: GGCLE1X'PeABi'P PPN rxpathc|}|jdjdrn|jdrdj|dd}n|jdrdjt|}n|jd r/|j d rd jt|dd }n|jd r*|dd}d jt|t |}nK|j d r |dd }d jt|}ndjt|}|j d}t|s td|tjd|||S)z*make xpath to be computer recognized xpath(/@z//*[@resource-id={!r}]rBN^zX//*[re:match(@text, {0}) or re:match(@content-desc, {0}) or re:match(@resource-id, {0})]%z9//*[contains(@text, {0}) or contains(@content-desc, {0})]z}//*[{0} = substring(@text, string-length(@text) - {1} + 1) or {0} = substring(@content-desc, string-length(@text) - {1} + 1)]z?//*[starts-with(@text, {0}) or starts-with(@content-desc, {0})]z7//*[@text={0} or @content-desc={0} or @resource-id={0}]z Invalid xpathzxpath %s -> %s) lstrip startswithr)r*endswithlenrstripr9rloggerdebug)rH orig_xpathtexts r strict_xpathrYNspJ ||C##C(   # (//ab :   # jqq      # 5>>##6KRR q %    # QRyPWW  D    Sbz M T TT"   JPP    LL E e $*55 LL!:u5 LrcHeZdZfdZfdZddZdeddfdZd dZxZ S) r7ct|tr|St|}|r!tjd|t|St |||S)Nc$|j|Sr=)joinpathabs rzXPath.__new__..|sAr)r.r7rY functoolsreducesuper__new__)clsvalueargsrH __class__s rrez XPath.__new__wsJ eU #LU# ##$>eElS S7?3. .rc(dt|dS)NzXPath())rd__repr__)selfris rrlzXPath.__repr__s(*+1--rrctr=)NotImplementedErrorrmrgs r__and__z XPath.__and__s!!rsubpathcJ|jdsd|z}t||zS)NrK)rQr7)rmrrs rr]zXPath.joinpaths(!!#&GmGTG^$$rc$|j|Sr=) find_elementsrmsources rallz XPath.alls##D))r)rgr7r XPathSelector)rw PageSource) rrrrerlrqr0r]rx __classcell__ris@rr7r7vs*/."%%% *rr7ceZdZdefdZedeedfddfdZejde jfdZ dede dfd Zy ) rz xml_contentc@tjdd|}||_y)Nz[\u200B-\u200F\uFEFF]r")r$r% _xml_content)rmr~clean_xml_contents r__init__zPageSource.__init__sFF#;RM-rdatarc<t|tr t|S|Sr=)r.r0rz)rs rparsezPageSource.parses dC d# # rctjt|j}|j dD]0}t |j jddxsd|_2|S)Nz//nodeclassr"node) r fromstringr4rrHr'attribpoptag)rm_rootrs rrootzPageSource.roots\  4+<+KK) KD"4;;??7B#?@JFDH K rrH XMLElementc||jj|ddi}|Dcgc] }t|c}Scc}w)Nr$$http://exslt.org/regular-expressions namespaces)rrHr)rmrHmatchesrs rruzPageSource.find_elementss7))//%T;a4b/c-45T 4 555s9N)rrrr0r staticmethodr rrbcached_propertyr _Elementrrrurrrrzrzsw.C. E#|+, enn 6364 +=6rrzceZdZdefdZdZdZedZedZ de fdZ d$d Z e d d efdZe d d$dZe d d%defdZe d dZe d dZe d dZd$dedeefdZej2dfdedeeefdedfd Zd$ded!eeee fdd"fd#Zy )& XPathEntrydc.||_t|dsJy)z= Args: d (uiautomator2 instance) wait_timeoutN)_dhasattr)rmrs rrzXPathEntry.__init__s q.)))rcxdh}||vr td||dk(r|j|yt|d|z|y)Ntimeoutz invalid key_)r2implicitly_waitsetattr)rmkeyrg valid_keyss r global_setzXPathEntry.global_setsI   j ]C0 0 )    ' D#)U +rc&||j_y)zset default timeout when clickNrrrmrs rrzXPathEntry.implicitly_waits&rc.|jjSr=rrms rrzXPathEntry.wait_timeoutsww###rc.|jjSr=)rwatcherrs r_watcherzXPathEntry._watcherswwrrc\tj|jjSr=)rzrrdump_hierarchyrs rget_page_sourcezXPathEntry.get_page_sources  6 6 899rNcHt|||jdkDSNrrSrxrmrHrws rmatchzXPathEntry.matchs"4v&**,-11rzuse d.watcher.when(..) insteadreasonxqueryc8|jj|Sr=)rwhen)rmrs rrzXPathEntry.whens}}!!&))rzuse d.watcher.run() insteadc8|jjyr=)rrunrvs r run_watcherszXPathEntry.run_watcherss rzuse d.watcher.start(..) insteadintervalc8|jj|Sr=)rstart)rmrs rwatch_backgroundzXPathEntry.watch_backgrounds}}""8,,rzuse d.watcher.stop() insteadc8|jjy)zstop watch backgroundNrstoprs r watch_stopzXPathEntry.watch_stops rzuse d.watcher.remove() insteadc8|jjyr=rrs r watch_clearzXPathEntry.watch_clears rremovedc2tj|z}tj|kri|jtd|tjz }tjt d|tj|krhyy)zrun watchers when sleepr?N)timermaxsleepmin)rmsecondsdeadline left_times r sleep_watchzXPathEntry.sleep_watchsh99;(iikH$    Ax$))+56I JJs3 * +iikH$rrHrc@t||}|j|y)z Find element and perform click Args: xpath (str): xpath string timeout (float): pass pre_delay (float): pre delay wait time before click Raises: TimeoutException rN)DeviceXPathSelectorclick)rmrHrselectors rrzXPathEntry.clicks'ud3w'r directionrNc&|tjk(rtj}nx|tjk(rtj}nT|tj k(rtj }n0|tjk(rtj}n td||dkDsJ||}t|D]X}|jr.|jj|d|jcS|jj|dZy)z Need more tests scroll up the whole screen until target element founded Returns: bool (found or not) zInvalid directionrg?rN)r FORWARDUPBACKWARDDOWN HORIZ_FORWARDLEFTHORIZ_BACKWARDRIGHTr2rangeexistsr swipe_extget_last_matchrmrHr max_swipestargetis r scroll_tozXPathEntry.scroll_tos  )) )! I ),, ,!I )11 1!I )22 2!I0)< <A~~ez" .A}}!!)S1,,.. GG  i -  . rrwrcRt|||rtj|SdSr=)rrzrrs r__call__zXPathEntry.__call__s'"5$F 0@0@0H]]X\]]rr=)g@)rrrrrrrpropertyrrrzrrrr0rrfloatrrrrrrr rr rrrrrrrs~*2* ,'$$::278*3*9*45689--:-567789y!,", (3 (% ($,5+<+<  C(  ! " D^c^8E#z/4J+K^Wl^rrceZdZdZdZy)OperatorANDORN)rrrrrrrrrr!s C Brrc@eZdZejdededfdZy)AbstractSelectorrwrrcyr=rrvs rrxzAbstractSelector.all's rN)rrrabcabstractmethodrzrrxrrrrr&s- * l);  rrceZdZdeeeeffdZdZe deedfddfdZ dZ ddZ ddZ ed d eeeefddfd Zd eddfd ZdededfdZy)ryrgct|trt||_n4t|ttfr||_nt dt ||d|_d|_y)Nr-) r.r0r7 _base_xpathrr2r3 _operator _next_xpathrps rrzXPathSelector.__init__-sR eS !$U|D  '78 9$D ^T%[%@ @-17;rc,tj|S)z copy self)copyrs rrzXPathSelector.copy7syyrrct|tr|jSt|tr|t |St d|)Nz Invalid value)r.ryrr0r7r2)rfrgs rcreatezXPathSelector.create;s> e] +::<  s #uU|$ $_e4 4rc|jrFdt|jd|jjdt|jdSdt|jdS)Nz#( rk)rreprrrgrrs rrlzXPathSelector.__repr__Dsc >>T--./q1E1E0FaTM]M]H^G__`a aT--./q1 1rczt|}tj||_tj|_|Sr=)ryrrrrrrmrgrs rrqzXPathSelector.__and__Js. $ %,,U3 ll rczt|}tj||_tj|_|Sr=)ryrrrrrrs r__or__zXPathSelector.__or__Ps. $ %,,U3 kk rzuse and_ or & insteadr_xpathcht|ttfrtjd||S||zS)z] add xpath to condition list the element should match all conditions c ||zSr=rr^s rraz%XPathSelector.xpath..]s Qr)r.listtuplerbrc)rmr s rrHzXPathSelector.xpathVs2 ftUm ,##$6E E&= rc|jst|jts t d|j }|jj ||_|S)z! add child xpath z9can't use child when base is not XPath or operator is set)rr.rr7rrr])rmr news rchildzXPathSelector.childasN >>D,<,  8;;.H H =>!!3T^^DDrN)rry)rrrr r0r7rrr classmethodrrlrqr rrrrHrrzrrxrrrryry,s|j|jdSr)rxrrs rrz"DeviceXPathSelector.get_last_matchsxx))*1--rc6|jjS)z get element text Returns: string of node text Raises: XPathElementNotFoundError )r7rXrs rget_textzDeviceXPathSelector.get_textsxxzrrXc|j}|j|jjj |jjj |yr=)r7rrr clear_text send_keys)rmrXr3s rset_textzDeviceXPathSelector.set_texts@ XXZ   ""$ !!$'rctj|xs |jz} |jrytj|kDrytjd;)z wait until element found TF皙?rr.rrrmrrs rr6zDeviceXPathSelector.waitsK99;'"AT-A-AB{{yy{X% JJsO rc<|jr|jSy)zG Returns: None or matched DeviceXMLElement N)rrrs rrzDeviceXPathSelector.matchs ;;&&( ( rctj|xs |jz}tj|kr:|jsytjdtj|kr:y)zr Args: timeout (float): seconds Returns: True if gone else False Tr@FrArBs r wait_gonezDeviceXPathSelector.wait_gonesW99;'"AT-A-ABiikH$;; JJsOiikH$rc|jdj\}}tjd|||jj j ||y)Nrz click %d, %d)rxcenterrUinforrrrmxys r click_nowaitz DeviceXPathSelector.click_nowaitsFxxz!}##%1 NAq) a#rc |j|}|jy#t$rF|jstj dt |j|jcYSwxYw)zfind element and perform clickrzelement not found, run fallback)rN)r7rrrrUrHrrrmrr3s rrzDeviceXPathSelector.clicks[ :'*B HHJ( :>> KK9 :t~~9 9  :s"%A A43A4ch |j|}|jy#t$rYywxYw)zreturn if clickedrTF)r7rrrNs r click_existsz DeviceXPathSelector.click_exists s5 '*B HHJ(  s "% 11c@|jjy)z#find element and perform long clickN)r7 long_clickrs rrRzDeviceXPathSelector.long_clicks  rcB|j}|jS)ztake element screenshot)r7 screenshot)rmr3s rrTzDeviceXPathSelector.screenshots XXZ}}rrc|jdr td|tt|s td||j }t ||S)z In IPython console, attr:_ipython_canary_method_should_not_exist_ will be called So here ignore all attr startswith _ rz Invalid attr)rQAttributeErrorrr1r7getattr)rmrr3s r __getattr__zDeviceXPathSelector.__getattr__sM >>#  5 5'- 5 5 XXZr3rr=)rr)rr1)'rrrr r0rrrrzrryr#rqr rboolr+rrr.r0rrxrr7rr:r>r6rrErLrrPrRr rTrXr{r|s@rrrzsI2eC)9$9:2J2X`akXl2] ## XhsDy&9:  .*.G(:.G$?Q:RG###%". (3- (S( D)x 23)  $ :D EKK  s  rrc.eZdZdejfdZdZdZdZdde fdZ dZ dd e d e fd Z dd eededfdZej&deeeeeffdZedeeeeeffdZedZedeeeffdZedeeeffdZy )relemc||_y)zX Args: elem: lxml node d: uiautomator2 instance N)r[)rmr[s rrzXMLElement.__init__-s  rc,t|jSr=)hashr[rs r__hash__zXMLElement.__hash__5sDIIrc:|jt|k(Sr=)r_r^rps r__eq__zXMLElement.__eq__8s}}$u+--rcx|j\}}dj|jj||S)Nz()rrJrK)rGr)r[rrIs rrlzXMLElement.__repr__;s9{{}19@@ aA  r strip_indexc|jj}|j|j}|rtjdd|}|S)zget element full xpathz\[\d+\]r")r[ getroottreegetpathr$r%)rmrcrpaths r get_xpathzXMLElement.get_xpathAs@yy$$&||DII& 66*b$/D rc&|jddS)z- Returns: (x, y) r)offsetrs rrGzXMLElement.centerIs {{3$$rpxpych|j\}}}}|t||zz|t||zzfS)z Offset from left_top Args: px (float): percent of width py (float): percent of height Example: offset(0.5, 0.5) means center )rectr-)rmrkrlrJrKwidthheights rrjzXMLElement.offsetPs<#ii1eV3urz?"AFRK(8$888rNrHrrc|#t|jjS|jj}|j}g}|.||k7r)|j ||j}|||k7r)t |}|j |ddi}|Dcgc]}|j|}}t|D]"}|j||vst|cSycc}w)z( Returns parent element Nr$rr) rr[ getparentreappendrYrHrfreversed)rmrHreelsrm all_pathss rrzXMLElement.parent^s =dii1134 4yy$$& IImT JJqM AmT U#** t%KL /66T\\!_6 6# %A||A)+!!}$ %7s"C-c|jjjd}|syttt j d|\}}}}||||fS)zJ Returns: tuple of (left, top, right, bottom) bounds)rrrrz\d+)r[rr7mapr-r$findall)rmrzlxlyrxrys rrzzXMLElement.boundsusR !!%%h/S"**VV"<=BBBBrc<|j\}}}}||||z ||z fS)zN Returns: (left_top_x, left_top_y, width, height) )rz)rmr}r~rrs rrnzXMLElement.rects, BB2rBwR''rcL|jjjdS)NrX)r[rr7rs rrXzXMLElement.textsyy##F++rc@t|jjSr=)dictr[rrs rrzXMLElement.attribsDII$$%%rcBi}t|jjD]B\}}|dvr |dvr|dk(|t|<!|dk(rt |||<5||t|<Dt |j j|d<|j j|d<|j\}}}}||||d|d<|jjd |d <|jjd |d <|jjd |d<|S)N)rzrpackage content-desc) checkablechecked clickableenabled focusablefocused scrollablezlong-clickablepasswordselectedzvisible-to-usertrueindex childCount className)lefttoprightbottomrzr packageNamercontentDescriptionz resource-id resourceName) rritemsrGr-rSr[ getchildrenrrzr7)rmretkr+r}r~rrs rrHzXMLElement.infos%++- 2DAqBBRR01V )!,-gQA01)!,- 2  5 5 78L99==KBB!#BrJH "[[__Y7M$(KKOON$C !"kkoom<N r)F)rr=)rrrr rrr_rarlrYrhrGrrjrr0r rrbrr r-rzrrnrXrrrrHrrrrr,sU^^. T% 9 9% 9%HSM%U;M5N%.  c3S01    (eCc3./((,,&S#X&&d38nrrc eZdZdedeffd ZdZdZdZdde e e fde fd Z e jfde e e fd efd Ze jd fd e de ded e dfdZddeefdZdZxZS)r1r3rcFt||j||_yr=)rdrr[r)rmr3rris rrzDeviceXMLElement.__init__s ! rcv|j\}}|jjj||y)z: click element, 100ms between down and up N)rGrrrrIs rrzDeviceXMLElement.clicks+{{}1 a#rcv|j\}}|jjj||y)zJ Sometime long click is needed, 400ms between down and up N)rGrrrRrIs rrRzDeviceXMLElement.long_clicks+{{}1 ""1a(rc|jjj}|j|jS)z, Take screenshot of element )rrrTcroprz)rmims rrTzDeviceXMLElement.screenshots-\\__ ' ' )wwt{{##rrscalecZt|jj|j||S)z Args: direction: one of ["left", "right", "up", "down"] scale: percent of swipe, range (0, 1.0) Raises: AssertionError, ValueError )rrrrz)rmrrs rswipezDeviceXMLElement.swipes!t|| YNNrrc|dk(rtj}n|dk(rtj}t|j dj }|j |dt|j dj |z }|jdz}|Dcgc]$}|jj|s#|&}}t|dkDScc}w)z Args: direction: Direction eg: Direction.FORWARD Returns: bool: if can be scroll again forwardbackwardz//*333333?)rrKr) r rrrrrxrrhrQrS)rmrrv new_elementsppathr3s rscrollzDeviceXMLElement.scrolls  !!))I * $!**I$,,u%))+, 9C (4<<.2245;  3&(MbBLLN,E,Ee,LrMM3x!|Ns )$C"C"rrHrrc|dkDsJ|j|}t|D]3}|jr|jcS|j |r3yyr)rrrrrrs rrzDeviceXMLElement.scroll_tos_A~~e$z" A}},,..;;y)   rwsizec|j\}}}}|xs$|jjj\}}||z ||z ||z ||z fS)z Args: wsize (tuple(int, int)): window size Returns: list of 4 float, eg: 0.1, 0.2, 0.5, 0.8 )rzrr window_size)rmrr}r~rrwwwhs rpercent_boundszDeviceXMLElement.percent_boundssSBB7$,,//557BRb"r'2733rc|jjj\}}|j\}}}}||z ||z fS)zGReturns: (float, float): eg, (0.5, 0.5) means 50%, 50% )rrrrn)rmrrrwhs r percent_sizezDeviceXMLElement.percent_sizesB,,.BYY 1aBBr)rr=)rrrrrrrrRrTr r r0rrrrYrr-rrrrrr{r|s@rr1r1s:z$)$ OuY^4 OU O9B8I8Iin 5d.2;1B1BVX  %. PS  ! "  4HUO 4 rr1): __future__rrrenumrbloggingr$rtypingrrrrrr r lxmlr PILr uiautomator2._protor uiautomator2.abstractruiautomator2.exceptionsruiautomator2.utilsrrr getLoggerrrU Exceptionrrr0r'r*r/r4rYr9rGrYr7rzobjectrEnumrABCrryrrr1rrrrsh'  DDD):=GG   8 $ y ''  3 3 CC 1sEz"1u1 S S %%%P*C*2660w^w^tsDII  sww L$L\o -o d}}@Y zY r