jSddlZddlZddlZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddl mZmZddlmZmZGdd eZGd d eZy) N)DictListOptionalTuple)Image)retry) SCROLL_STEPS) HTTPErrorUiObjectNotFoundError)Exists intersectc eZdZdZidddddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,Zd-\ZZZfd.Zd/Z fd0Z fd1Z d2Z d3Z d4Zd5ZxZS)6SelectorzNThe class is to build parameters for UiSelector passed to Android device. text)N textContains)N textMatches)NtextStartsWith)N className)NclassNameMatches) N description)@NdescriptionContains)NdescriptionMatches)NdescriptionStartsWith)iN checkable)iFchecked)iF clickable)iF longClickable)i F scrollable)i@Fenabled)iF focusable)iF)iF)iF)iN)iN)i N)i@N)ir)ir)focusedselected packageNamepackageNameMatches resourceIdresourceIdMatchesindexinstance)maskchildOrSiblingchildOrSiblingSelectorc tt| |jdtt| |jgtt| |j g|D] }||||< yNr)superr __setitem___Selector__mask_Selector__childOrSibling!_Selector__childOrSiblingSelector)selfkwargsk __class__s W/var/www/api.ozpay.ru/ozpay/venv/lib/python3.12/site-packages/uiautomator2/_selector.py__init__zSelector.__init__.sb h)$++q9 h)$*?*?D h)$*G*GL AQiDG c:|j}|jddD]%}|j|r|j|'g}|jD]%\}}|j |dzt |z'ddj |zdzS)z) remove useless part for easily debugger r2)r3r4=z Selector [z, ])copypopgetitemsappendreprjoin)r<selectorkeyargsr>vs r@__str__zSelector.__str__5s99; V? "C<<$ S! "nn& +FQ KKC$q') * +diio-33rBc||jvrStt|||tt||j||j|j|dzyt d|z)Nrz%s is not allowed.)_Selector__fieldsr7rr8r9ReferenceError)r<r>rPr?s r@r8zSelector.__setitem__Asl   (D -a 3 ( $$(KK$($5 a8H8K$KM!!5!9: :rBc||jvrStt||tt||j ||j |j|dzyyr6)rSr7r __delitem__r8r9)r<r>r?s r@rVzSelector.__delitem__Js_   (D -a 0 ( $$(KK$($5q9I!9L8L$LN rBc*tfdD}tdi|}jD] }|jj|"jD].}|jj|j 0|S)Nc3xK|]1}|jjjfvs)||f3ywN)r9r:r;).0r>r<s r@ z!Selector.clone..Rs@ qA KK..0M0M> 5 q$q'l s*: :)dictrr:rJr;clone)r<r=rMrPss` r@r^zSelector.cloneQs D  %f%d++, 6A T** + 2 21 5 6d334 FA T22 3 : :1779 E FrBc ||jjd||jjtdi||S)Nchildr\r:rJr;rr<r=s r@razSelector.child\sA T " "#**73 T * *+2283Ef3EF rBc ||jjd||jjtdi||S)Nsiblingr\rbrcs r@rezSelector.siblingasA T " "#**95 T * *+2283Ef3EF rBcX||jr|||jdd<y||d<y)Nr1)r;)r<is r@update_instancezSelector.update_instancefs4 -- .BCD.. / 3J ? D rB)__name__ __module__ __qualname____doc__rSr9r:r;rArQr8rVr^rareri __classcell__)r?s@r@rrs1   | ,  \  L  | | n  _ ? _  o ?!" &#$%%'.&-#3H6:l6F 6  4;N   !rBrceZdZdefdZedZedZedZde e fdZ d0d e e dejfd Zd1d Zdee e e e ffd Zd2d Zd3dZd4defdZd5defdZdZd6dZd7dZd8dZd8dZd9dZd0dZd9dZ dZ!d0dZ"d0dZ#d0dZ$dZ%dZ&e%e&cZ'Z(d Z)d!Z*d"Z+d#Z,d$e fd%Z-ed&Z.d'Z/d(Z0d)Z1d*Z2d+Z3d,Z4d-Z5ed.Z6ed/Z7y):UiObjectrMcB||_||_|j|_yrY)sessionrMjsonrpc)r<rrrMs r@rAzUiObject.__init__os    rBc.|jjSrY)rr wait_timeoutr<s r@ruzUiObject.wait_timeoutts||(((rBct|S)z-check if the object exists in current window.)r rvs r@existszUiObject.existsxsd|rBcL|jj|jS)zui object info.)rsobjInforMrvs r@infoz UiObject.info}s||##DMM22rBreturncL|jj|jS)z!all matched ui objects info list.)rsobjInfoOfAllInstancesrMrvs r@ info_listzUiObject.info_lists||11$--@@rBN display_idcx|jj|}|j|jS)N)r)rr screenshotcropbounds)r<rims r@rzUiObject.screenshots. \\ $ $ $ ;wwt{{}%%rBc|j||j|\}}|jj||y)ag Click UI element. Args: timeout: seconds wait element show up offset: (xoff, yoff) default (0.5, 0.5) -> center The click method does the same logic as java uiautomator does. 1. waitForExists 2. get VisibleBounds center 3. send click event Raises: UiObjectNotFoundError timeout)offsetN) must_waitcenterrrclick)r<rrxys r@rzUiObject.clicks: w'{{&{)1 1a rBc|j}|jdxs|jd}|d|d|d|df\}}}}||||fS)z] Returns: left_top_x, left_top_y, right_bottom_x, right_bottom_y visibleBoundsrlefttoprightbottom)r{rH)r<r{rlxlyrxrys r@rzUiObject.boundss` yy/*@dhhx.@wPXIYYBBBBrBcz|j\}}}}|d}|\}}||z ||z } }|||zz} || |zz} | | fS)z Args: offset: optional, (x_off, y_off) (0, 0) means left-top, (0.5, 0.5) means middle(Default) Return: center point (x, y) ?r)r) r<rrrrrxoffyoffwidthheightrrs r@rzUiObject.centers`BB >F dRbv   $ 1v rBc|j|dkDr=tj||jsy|j|dz}|dkDr=y)z Click until element is gone Args: maxretry (int): max click times interval (float): sleep time between clicks Return: Bool if element is gone rTrF) click_existstimesleeprx)r<maxretryintervals r@ click_gonezUiObject.click_gonesP l JJx ;;     MH l rBcH |j|y#t$rYywxYw)NrTF)rr r<rs r@rzUiObject.click_existss*  JJwJ '$  s  !!durationc|j||j\}}|jj|||S)z Args: duration (float): seconds of pressed timeout (float): seconds wait element show up r)rrrr long_click)r<rrrrs r@rzUiObject.long_clicks: w'{{}1||&&q!X66rBc6|jdd}|jdd}j|t|dzt|dk\sd|vsd|vrfd }||i|Sjj j td i|S) Nrrrrrrrcjj||\}}jjj||SrY)rr pos_rel2absrsdragTorM)rrr<stepss r@drag2xyz!UiObject.drag_to..drag2xys?||//0131||**4==!QFFrBr\)rGrintlenrsrrMr)r<rOr=rrrrs` @r@drag_tozUiObject.drag_tos::j#.**Y- w'HsN# t9>SF]cVm G D+F+ +||""4==(2DV2DeLLrBc|dvsJ|j|j}|jdxs|jd}|d|d|d|df\}}}}||zdz||zdz} } |d k(r!|jj | | | || y |d k(r$|jj | | | |d z | y |dk(r!|jj | | || | y |dk(r$|jj | | |d z | | y y )aZ Performs the swipe action on the UiObject. Swipe from center Args: direction (str): one of ("left", "right", "up", "down") steps (int): move steps, one step is about 5ms percent: float between [0, 1] Note: percent require API >= 18 # assert 0 <= percent <= 1 )rrupdownrrrrrrrr)rrrN)rr{rHrrswipe) r< directionrr{rrrrrcxcys r@rzUiObject.swipes$;;;; yy/*@dhhx.@wPXIYYBBr'a"r'aB   LL  r2r2U  ; & LL  r2r26  ? & LL  r2r2U  ; ' ! LL  r2rAvr  ?"rBc |jj d fd fd}||||||||f\}}} } |jj|j||| | |S)z perform two point gesture. Usage: d().gesture(startPoint1, startPoint2, endPoint1, endPoint2, steps) c&||\}}||dS)N)rrr\)rrrel2abss r@pointzUiObject.gesture..points1a=DAq# #rBc6t|tk(r|S|SrY)typetuple)ptrs r@ctpzUiObject.gesture..ctps!%bU!25": : :rB)rr)rrrrsgesturerM) r<start1start2end1end2rrs1s2e1e2rrs @@r@rzUiObject.gesturesi ,,** $ ;Vc&k3t9c$iGBB||##DMM2r2r5IIrBcP|jj|j||SrY)rspinchInrMr<percentrs r@pinch_inzUiObject.pinch_in!s||##DMM7EBBrBcP|jj|j||SrY)rspinchOutrMrs r@ pinch_outzUiObject.pinch_out$s||$$T]]GUCCrBc| |j}|dz}|r5 |jj|jt |dz|S |jj|jt |dz|S#t $r7}t jd|zt|jcYd}~Sd}~wwxYw#t $r8}t jd|zt|j cYd}~Sd}~wwxYw)z Wait until UI Element exists or gone Args: timeout (float): wait element timeout Example: d(text="Clock").wait() d(text="Settings").wait(exists=False) # wait until it's gone N i) http_timeoutzwaitForExists readTimeout: %s) rurs waitForExistsrMrr warningswarnRuntimeWarningrx waitUntilGone)r<rxr http_waites r@waitz UiObject.wait's ?''GbL  %||11$--25gn2E?H2JJ )||11$--25gn2E?H2JJ  % =A,.{{}$ % ) =A,.;;=(( )s;3B 3C C ,B;5C;C D -C?9D?DcH|xs |j}|jd|S)z wait until ui gone Args: timeout (float): wait element gone timeout Returns: bool if element gone F)rxr)rurrs r@ wait_gonezUiObject.wait_goneHs'.T..yywy77rBcl|j||s"tdt|jddy)z3 wait and if not found raise UiObjectNotFoundError ir)codedatamethodN)rr strrM)r<rxrs r@rzUiObject.must_waitSs1yy)'T]]AS_e(fg g*rBc$|j|S)z alias of set_text )set_text)r<rs r@ send_keyszUiObject.send_keysXs}}T""rBc|j||s%|jj|jS|jj |j|SNr)rrsclearTextFieldrMsetText)r<rrs r@rzUiObject.set_text\sG w'<<..t}}= =<<'' t< .Itercd|_y)Nrg)r0rvs r@rAz(UiObject.__iter__..Iter.__init__s  rBc~|xjdz c_|jkr|jSt)Nr)r0 StopIteration)r<lengthobjs r@nextz$UiObject.__iter__..Iter.nexts2 a ::&tzz?*'/)rBN)rjrkrlrAr__next__)rrsr@Iterrs  *HrBr)r object)r<rrrs @@r@__iter__zUiObject.__iter__s'DJJ V 6 v rBc .d}|j|fi|S)NcHt||\}}}}||kr |d|dz SdSNrrrgr rect1rect2rrrrs r@ onrightofz!UiObject.right..onrightof5'0'> $D#uf586\5=5>1 Ir IrB_UiObject__view_beside)r<r=r&s r@rzUiObject.rights" J"t!!)6v66rBc .d}|j|fi|S)NcHt||\}}}}||kr |d|dz SdSr!r"r#s r@onleftofzUiObject.left..onleftofr'rBr()r<r=r,s r@rz UiObject.lefts" J"t!!(5f55rBc .d}|j|fi|S)NcHt||\}}}}||kr |d|dz SdSNrrrgr"r#s r@abovezUiObject.up..above5'0'> $D#uf59E\5<%/1 Ir IrBr()r<r=r0s r@rz UiObject.up" J"t!!%2622rBc .d}|j|fi|S)NcHt||\}}}}||kr |d|dz SdSr/r"r#s r@underzUiObject.down..underr1rBr()r<r=r5s r@rz UiObject.downr2rBc P|jd}d\}}t|jtdi|j }t |D]V\}}|||d} | dk\s|dks| |ks#tdi|} | j || t|j| }}X|S)Nr)rgNrr\)r{rprrrr enumerateri) r<onsideofr=rmin_distfoundrr0r{dist ui_selectors r@ __view_besidezUiObject.__view_besides8$"%T\\8+=f+=>HHJ $Y/ LKE4FDN3DqyhldXo&00 ++E2"&{(K%  L  rBch|j|jGfddt}|S) Args: dimention (str): one of "vert", "vertically", "vertical", "horiz", "horizental", "horizentally" action (str): one of "forward", "backward", "toBeginning", "toEnd", "to" c(eZdZdZdZdfd Zy)UiObject.fling.._Flingc d|_d|_yNTforward _verticalactionrvs r@rAz'UiObject.fling.._Fling.__init__!%' rBcl|dvr d|_|S|dvr d|_|S|dvr ||_|Std|zN)horiz horizental horizentallyF)vert verticallyverticalT)rDbackward toBeginningtoEndtozinvalid prop %srFrG ValueErrorr<rNs r@ __getattr__z*UiObject.fling.._Fling.__getattr__XAA%*DNK<<%)DNK#&DKK !2S!899rBcb|jdk(rj|jS|jdk(rj|jS|jdk(rj |j|S|jdk(rj |j|Sy)NrDrQrRrS)rG flingForwardrF flingBackwardflingToBeginning flingToEnd)r< max_swipesr=rsrMs r@__call__z'UiObject.fling.._Fling.__call__ s;;)+"//$..II[[J."004>>JJ[[M1"33Hdnn4>@@[[G+"--h.8::,rBN))rjrkrlrArXr`rsrMsr@_FlingrAs ( : :rBrc)rsrMr)r<rcrsrMs @@r@flingzUiObject.flings,,,== :V :>xrBch|j|jGfddt}|S)r?c,eZdZdZdZedffd Zy) UiObject.scroll.._Scrollc d|_d|_yrCrErvs r@rAz)UiObject.scroll.._Scroll.__init__#rHrBcl|dvr d|_|S|dvr d|_|S|dvr ||_|Std|zrJrUrWs r@rXz,UiObject.scroll.._Scroll.__getattr__'rYrBrac|jdvr;|jdk(r jn j}||j|S|jdk(rj |j||S|jdk(rj |j||S|jdk(r&j tdi||jSy)N)rDrQrDrRrSrTr\)rG scrollForwardscrollBackwardrFscrollToBeginning scrollToEndscrollTor)r<rr_r=rrsrMs r@r`z)UiObject.scroll.._Scroll.__call__5s;;"996:kkY6NW22T[TjTjF!(DNNEBB[[M1"44Xt~~5?HH[[G+"..x/95BB[[D("++Hh6H6H,0NN<<)rBN)rjrkrlrArXr r`rbsr@_Scrollrg"s ( :&2c J$CD)B 8h #=3 #MO#(NK, ,0$0C0(11$7633 ((T++rBrp)loggingrrtypingrrrrPILrruiautomator2._protor uiautomator2.exceptionsr r uiautomator2.utilsr r r]rrrpr\rBr@rs@ ..,D0]!t]!@VvVrB