~@;jWBddlmZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZmZmZmZddlZdd lmZdd lmZdd lmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,Gdde-Z.dS)) annotationsN)byref)CDLL)c_long)HWND)Path)Popen)time)AnyDictTupleOptional) ImageGrab)AccessBridgeVersionInfo)ActorScheduler)By) JABException)Logger)Service) Win32Utils) JOBJECT64)TIMEOUT) JABElement) JABFixedFunccLeZdZdZddddddefdVfd ZdZdZdZe dWdZ e j dXdZ e dYdZ e j dZdZ e d[dZ e j d\dZ e d]dZej d^dZe d_d Zej d`d!Ze dad#Zej dbd%Ze dcd'Zej ddd)Zded*Zded+Zdfd-Zdgd/Zd0Zdhd2Zdid4Zefdjd5Zdkdld9Zdkdld:Zdkdld;Zdkdld<Z dkdmd=Z dkdmd>Z dkdmd?Z dkdld@Z!e"j#dd6fdndCZ$ dkdodEZ% dkdodFZ& dkdodGZ' dkdodHZ( dkdpdIZ) dkdpdJZ* dkdpdKZ+ dkdodLZ,e"j#dd6fdqdMZ-dNZ.dOZ/e"j#defdrdPZ0dQZ1dRZ2dSZ3dTZ4dUZ5xZ6S)s JABDriverzControls a Java application by Java Access Bridge. Args: Service ([type]): Host system to initialize the JAB and load JAB dll file. Ntitlestr file_pathr bridge_dllhwndrvmidraccessible_contextrtimeoutintreturnNonectt|t|_||_||_|jr|t|_ td|_ d|_ ||_ ||_||_||_d|_||_d|_d|_|t-|jdS)aCreate a new jab driver. Args: title (str, optional): Window title of Java application need to bind. Defaults to "". file_path (Path, optional): File path of the application to launch. Defaults to None. bridge_dll (str, optional): WindowsAccessBridge dll file path. Defaults to "". hwnd (HWND, optional): HWND of Java Window. Defaults to None. vmid (c_long, optional): vmid of Java Window. Defaults to None. accessible_context (JOBJECT64, optional): Any Accessible Context Component in Java Window. Defaults to None. timeout (int, optional): Default timeout set for JABDriver waiting. Defaults to TIMEOUT. pyjabN)superr__init__r win32utilsr _titleopen_applicationrservrlogger latest_log _bridge_dll_timeout_hwnd_vmid_pid_accessible_context_bridge _root_elementinit_jabrbridge_fix_bridge_functions) selfrr r!r"r#r$r% __class__s ]C:\Users\Administrator\AppData\Local\Temp\2\pip-target-b0z114gj\lib\python\pyjab/jabdriver.pyr,zJABDriver.__init__$s, i'')))$,," > $  ! ! # # #II Woo %    #5  ! T[!!7799999c|SNr>s r@ __enter__zJABDriver.__enter__Ns rAcNtj|jtjdSrC)oskillpidsignalSIGTERM)r>exc_typeexc_valexc_tbs r@__exit__zJABDriver.__exit__Qs &.)))))rAc|jjdkr)ddt|jgnt|j}t |d}||S)Njnlp javawsT)shell)r suffixjoinrr wait)r>cmdps r@r/zJABDriver.open_applicationTsh;?>;PTZ;Z;Zchh#dn"5"56777`cdhdr`s`s #T " " " rAc|jSrCr.rEs r@rzJABDriver.titleZs {rAc||_dSrCr\)r>rs r@rzJABDriver.title^s  rAc|jSrCr5rEs r@r"zJABDriver.hwndb zrAc||_dSrCr_r>r"s r@r"zJABDriver.hwndf  rAc|jSrCr7rEs r@rJz JABDriver.pidjs yrArJc||_dSrCre)r>rJs r@rJz JABDriver.pidns  rAc|jSrCr6rEs r@r#zJABDriver.vmidrr`rAc||_dSrCrh)r>r#s r@r#zJABDriver.vmidvrcrAc|jSrCr8rEs r@r$zJABDriver.accessible_contextzs ''rAc||_dSrCrk)r>r$s r@r$zJABDriver.accessible_context~s#5   rArc|jSrCr9rEs r@r<zJABDriver.bridges |rAr<c||_dSrCrn)r>r<s r@r<zJABDriver.bridges  rArc|jSrCr:rEs r@ root_elementzJABDriver.root_elements !!rArrc||_dSrCrq)r>rrs r@rrzJABDriver.root_elements)rAct}|d|j|dS)Nr*)r new_actorr-setup_msg_pumprun)r>scheds r@_run_actor_schedzJABDriver._run_actor_scheds@   !?!?!A!ABBB rAc|jd|j|j|_|j||js4|j r|j s&| |j |j |_|jr(||j\|_ |_ nh|j rR|j rK|j|j |j }|j|j ||_nt#d||jst#d|jd||_t+|j|j|j |j |_|jddS)Nzinit jab)rr%z8At least hwnd or vmid and accessible_context is requiredzHWND:z" is not Java Window, please check!)r<r"r#r$zinit jab success)r1infor0 load_libraryr3r< Windows_runryr"r#r$wait_java_window_by_titlerr4!_get_accessible_context_from_hwndgetTopLevelObjectgetHWNDFromAccessibleContext RuntimeError_is_java_windowget_pid_from_hwndrJrrr)r>top_level_objects r@r;zJABDriver.init_jabs $$$i,,T-=>>  !!!   di D,C 66j$-7DI 9 151W1W 22 .D #TYYY 42 #{<< 42    @@ +DIIJ ##DI.. VTtyTTTUU U))++&;#6     +,,,,,rAboolcPt|j|S)zReturn the specific window is or not a Java Window Args: hwnd (HWND): The hwnd of window. Returns: bool: True if is a Java Window. False is not a Java Window. )rr< isJavaWindowrbs r@rzJABDriver._is_java_windows"DK,,T22333rATuple[JOBJECT64, int]ct}t}|j|t |t |||jfS)zGets the AccessibleContext and vmID values for the given window. Args: hwnd (HWND): hwnd (HWND): The hwnd of window. Returns: Tuple: tuple of AccessibleContext and vmID )rrr<getAccessibleContextFromHWNDrvalue)r>r"r#r$s r@rz+JABDriver._get_accessible_context_from_hwndsVxx&[[ 00 %++u%788   "4:--rAc>tj|j\}}|SrC) win32processGetWindowThreadProcessIdr")r>_rJs r@rzJABDriver.get_pid_from_hwnds6tyAA3 rADict[str, str]ct}|j|jt ||j|j|j|jdS)aMGets the version information of the instance of Java Access Bridge instance your application is using. Returns: Dict[str]: Dict of AccessBridgeVersionInfo, contains: VMVersion bridgeJavaClassVersion bridgeJavaDLLVersion bridgeWinDLLVersion ) VMVersionbridgeJavaClassVersionbridgeJavaDLLVersionbridgeWinDLLVersion) rr<getVersionInfor#rrrrr)r>r{s r@get_version_infozJABDriver.get_version_infosU'(( ""49eDkk:::&*&A$($=#'#;    rAOptional[HWND]ct|j|D]}||r|cSdS)zGet Java Window hwnd by title. Args: title (str): Java window title Returns: Optional[HWND]: HWND if found Java Window, otherwise return None rN)r-get_hwnds_by_titler)r>rr"s r@get_java_window_hwndzJABDriver.get_java_window_hwndsTO66U6CC  D##D))     rAcZt} ||x}r|Sd|d}|j|kr!|j|||_t}t ||z }||krt d|d|d|)aWait until a Java Window exists in specific seconds. Args: title (str): The title of specific Java Window need to wait. timeout (int, optional): The timeout seconds. Defaults to TIMEOUT. Raises: TimeoutError: Timeout error occurs when wait time over the specific timeout Returns: HWND of Java window found in specific seconds. Trzno java window found by title ''z' in 'z'seconds)r rr2r1debuground TimeoutErrorry)r>rr%startr"log_outcurrentelapseds r@r~z#JABDriver.wait_java_window_by_titles $00u0===t  @@@@G')) !!'***")ffGGeO,,G'!!"TeTT7TTT  ! ! # # # $rAFrvisiblech||jjkr|jS|j||S)z: Find an JABElement given a name locator. rr)rrnamefind_element_by_namer>rrs r@rzJABDriver.find_element_by_name!: D%* * *$ $$99w9WW WrAch||jjkr|jS|j||S)zA Find an JABElement given a description locator. r)rr descriptionfind_element_by_descriptionrs r@rz%JABDriver.find_element_by_description*s; D%1 1 1$ $$@@uV]@^^ ^rAch||jjkr|jS|j||S)z: Find an JABElement given a role locator. r)rrrolefind_element_by_rolers r@rzJABDriver.find_element_by_role3rrAch||jjkr|jS|j||S)z; Find an JABElement given a state locator. r)rrstatesfind_element_by_statesrs r@rz JABDriver.find_element_by_states<sC D%, , ,$ $$;;W< rAch||jjkr|jS|j||S)zC Find an JABElement given an object depth locator. r)rr object_depthfind_element_by_object_depthrs r@rz&JABDriver.find_element_by_object_depthGsD D%2 2 2$ $$AAWB rAch||jjkr|jS|j||S)zD Find an JABElement given a children count locator. r)rrchildren_countfind_element_by_children_countrs r@rz(JABDriver.find_element_by_children_countTsD D%4 4 4$ $$CCWD rAch||jjkr|jS|j||SzF Find an JABElement given an index in parent locator. r)rrindex_in_parentfind_element_by_index_in_parentrs r@rz)JABDriver.find_element_by_index_in_parentasD D%5 5 5$ $$DDWE rAc:|j||Sr)rrfind_element_by_xpathrs r@rzJABDriver.find_element_by_xpathns  66UG6TTTrAbyr ctj|jtj|jtj|jtj|jtj |j tj |j tj |jtj|ji}||vrt%d|d||||S)zE Find an JABElement given a By strategy and locator. incorrect by strategy 'rr)rNAMEr DESCRIPTIONrROLErSTATESr OBJECT_DEPTHrCHILDREN_COUNTrINDEX_IN_PARENTrXPATHrkeysr)r>rrr dict_finds r@ find_elementzJABDriver.find_elementts GT. ND< GT. It2 OT>  tB   D Hd0   Y^^%% % %>>>>?? ?y}5'::::rAlist[JABElement]cg}||jjkr||j||j|||S)z? Find list of JABElement given a name locator. r)rrrappendextendfind_elements_by_namer>rr jabelementss r@rzJABDriver.find_elements_by_namej  D%* * *   t0 1 1 1   3 3% 3 Q Q   rAcg}||jjkr||j||j|||S)zF Find list of JABElement given a description locator. r)rrrrrfind_elements_by_descriptionrs r@rz&JABDriver.find_elements_by_descriptionsk  D%1 1 1   t0 1 1 1   : :PW : X X   rAcg}||jjkr||j||j|||S)z? Find list of JABElement given a role locator. r)rrrrrfind_elements_by_rolers r@rzJABDriver.find_elements_by_rolerrAcg}||jjkr||j||j|||S)z@ Find list of JABElement given a state locator. r)rrrrrfind_elements_by_statesrs r@rz!JABDriver.find_elements_by_statessj  D%, , ,   t0 1 1 1   5 5E7 5 S S   rAcg}||jjkr||j||j|||S)zH Find list of JABElement given an object depth locator. r)rrrrrfind_elements_by_object_depthrs r@rz'JABDriver.find_elements_by_object_depthsq  D%2 2 2   t0 1 1 1   ; ;W <     rAcg}||jjkr||j||j|||S)zI Find list of JABElement given a children count locator. r)rrrrrfind_elements_by_children_countrs r@rz)JABDriver.find_elements_by_children_countsq  D%4 4 4   t0 1 1 1   = =W >     rAcg}||jjkr||j||j|||SzK Find list of JABElement given an index in parent locator. r)rrrrr find_elements_by_index_in_parentrs r@rz*JABDriver.find_elements_by_index_in_parentsq  D%5 5 5   t0 1 1 1   > >W ?     rAc:|j||Sr)rrfind_elements_by_xpathrs r@rz JABDriver.find_elements_by_xpaths  77eW7UUUrActj|jtj|jtj|jtj|jtj |j tj |j tj |jtj|ji}||vrt%d|d||||S)zJ Find list of JABElement given a By strategy and locator. rrr)rrrrrrrrrrrrrrrrrrr)r>rrr dict_findss r@ find_elementszJABDriver.find_elementss GT/ ND= GT/ It3 OT?  tC   E Hd1   Z__&& & &>>>>?? ?z"~E7;;;;rAcP|j|jjdS)zK Maximizes the current java window that jabdriver is using r"N)r-_set_window_maximizerrr"rEs r@maximize_windowzJABDriver.maximize_window ) ,,$2C2H,IIIIIrAcP|j|jjdS)zJ Invokes the window manager-specific 'minimize' operation rN)r-_set_window_minimizerrr"rEs r@minimize_windowzJABDriver.minimize_windowrrAc t} t}t||z }t||z }|jd|d|||krt d|d|d|d |||S#t$r8d|d|d }|j|kr!|j|||_YnwxYw) NTz elapsed => z , remain => zJABElement with locator 'z' 'z' does not found in z seconds)rrz' does not found)r rr1rrrr2warning) r>rrr%rrrremainrs r@wait_until_element_existz"JABDriver.wait_until_element_exists# .ffGGeO,,G7W,--F K  IGIIII J J J'!!"cccuccRYccc .((Be(<<< . . .TbTTUTTT?g--K''000&-DO  . .sB?CCcX|}||dS)a Saves a screenshot of the current window to a PNG image file. Returns False if there is any IOError, else returns True. Use full paths in your filename. :Args: - filename: The full path you wish to save your screenshot to. This should end with a `.png` extension. :Usage: driver.get_screenshot_as_file('/Screenshots/foo.png') N)get_screenshotsave)r>filenameims r@get_screenshot_as_filez JABDriver.get_screenshot_as_file-s- " " rAcN|j|jj|jj}|d}|d}|d}|d}t j||||z||zfddS) z Gets the screenshot of the current window as a base64 encoded string which is useful in embedded images in HTML. :Usage: driver.get_screenshot_as_base64() rxywidthheightFT)bboxinclude_layered_windows all_screens)r-_set_window_foregroundrrr"boundsgetrgrab)r>rrrrrs r@rzJABDriver.get_screenshot=s ..D4E4J.KKK") JJsOO JJsOO 7##H%%~E F  %*    rAcT|j|jj||dS)a Sets the width and height of the current window. (window.resizeTo) :Args: - width: the width in pixels to set the window to - height: the height in pixels to set the window to :Usage: driver.set_window_size(800,600) )r"rrN)r-_set_window_sizerrr")r>rrs r@set_window_sizezJABDriver.set_window_sizeVs: (("'uV )     rAcT|j|jj||dS)a# Sets the x,y position of the current window. (window.moveTo) :Args: - x: the x-coordinate in pixels to set the window position - y: the y-coordinate in pixels to set the window position :Usage: driver.set_window_position(0,0) )r"lefttopN)r-_set_window_positionrrr")r>rrs r@set_window_positionzJABDriver.set_window_positiones: ,,"'aQ -     rAcL|j|jjS)zx Gets the x,y position of the current window. :Usage: driver.get_window_position() r)r-_get_window_positionrrr"rEs r@get_window_positionzJABDriver.get_window_positionts#339J9O3PPPrA)rrr rr!rr"rr#rr$rr%r&r'r()r'r)rrr'r()r'r)r"rr'r()r'r&)rJr&r'r()r'r)r#rr'r()r'r)r$rr'r()r'r)r<rr'r()r'r)rrrr'r()r'r()r"rr'r)r"rr'r)r'r)rrr'r)rrr%r&r'r)F)rrrrr'r)rr&rrr'r)rrrr rrr'r)rrrrr'r)rr&rrr'r)rrrrrrr'r)rrrr r%r&r'r)7__name__ __module__ __qualname____doc__rr,rFrPr/propertyrsetterr"rJr#r$r<rrryr;rrrrrr~rrrrrrrrrrrrrrrrrrrrrrrrrr rr __classcell__)r?s@r@rrs" ,0"(:(:(:(:(:(:(:T*** X \\X [[X ZZX [[(((X(6666X ]]"""X"**** )-)-)-)-X 4 4 4 4....     &    DK$$$$$<XXXXX_____XXXXX     /4     /4     /4     UUUUUG$;;;;;*/4     /4     /4     /4     /4"/4"/4"/4VVVVVG$<<<<<(JJJ JJJG$w.....*    2        QQQQQQQrAr)/ __future__rrHrKctypesrrrctypes.wintypesrpathlibr subprocessr r typingr r r rrPILrpyjab.accessibleinforpyjab.common.actorschedulerrpyjab.common.byrpyjab.common.exceptionsrpyjab.common.loggerrpyjab.common.servicerpyjab.common.win32utilsrpyjab.common.typesr pyjab.configrpyjab.jabelementrpyjab.jabfixedfuncrobjectrrDrAr@r.s""""""  ------------888888666666000000&&&&&&((((((......(((((( ''''''++++++^ Q^ Q^ Q^ Q^ Q^ Q^ Q^ Q^ Q^ QrA