j8dZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z m Z m Z ddlmZddlmZddlmZmZmZddldd lmZd Zd Zdd ZGd deZGddeZy)z3Created on Fri May 06 2022 10:58:29 by codeskyblue ) annotationsN)IteratorListOptionalUnion) deprecated)adb_path)AdbConnectionErrorAdbError AdbTimeout)*) __version__sOKAYsFAILcBtj} |jd|j||f |jy#tjtj f$r}Yd}~|jyd}~wwxYw#|jwxYw)z Returns if server is running g?TNF)socket settimeoutconnectclosetimeouterror)hostportses N/var/www/api.ozpay.ru/ozpay/venv/lib/python3.12/site-packages/adbutils/_adb.py _check_serverrss A R 4,   NNFLL )    s#$A #B /B B  B BceZdZddZdZdZeddZdZdZ dZ eddZ dd Z dd Z dd Zdd Zdd ZddZddZddZddZdddZdZy) AdbConnectioncJ||_||_|j|_yN)_AdbConnection__host_AdbConnection__port _safe_connect_AdbConnection__conn)selfrrs r__init__zAdbConnection.__init__,s   ((* c|j}|j}tj}|jtjtj dt j}|dk(r|jtjtjd|jtjtjd|jtjtjd |jd|j||f|jd|S#tj$r}tdd}~wtj $r}t#d|zd}~wwxYw)NLinux connect to adb server timeoutz connect to adb server failed: %s)r r!r setsockopt SOL_SOCKET SO_KEEPALIVEplatformsystem IPPROTO_TCP TCP_KEEPIDLE TCP_KEEPCNT TCP_KEEPINTVLrrrr rr )r$adb_hostadb_portr sys_platformrs r_create_socketzAdbConnection._create_socket1s;;;; MMO V&&(;(;Q?( 7 " LL++V-@-@" E LL++V-?-? C LL++V-A-A2 F M LLO IIx* + LL H~~ ><= =|| M$%G!%KL L Ms$6D::F EF.E<<Fc |jS#t$rYnt$rYnwxYwtjdk(rt j nd}t jtdgd||jS)Nntrz start-serverg4@)r creationflags) r9r r osname subprocessCREATE_NO_WINDOWrunr )r$flagss rr"zAdbConnection._safe_connectGsq &&( (  !   /1ww$ ++A N3TQVW""$$s (((c|jduSr)r#r$s rclosedzAdbConnection.closedRs{{d""r&c|jy |jjtjt j d|jjd|_y#t $rY-wxYw)Ng{Gz?)r#shutdownrSHUT_WRtimesleepOSErrorrrDs rrzAdbConnection.close[sc ;;    KK  0 JJt      s>A// A;:A;c|SrrDs r __enter__zAdbConnection.__enter__hs r&c$|jyr)r)r$exc_typeexc tracebacks r__exit__zAdbConnection.__exit__ks  r&cH|j td|jS)NzConnection is closed)r#r rDs rconnzAdbConnection.connns" ;; 12 2{{r&c8|jj|S)z alias of conn.send(data) )rUsendr$datas rrWzAdbConnection.sendtsyy~~d##r&c~ |jj|S#tj$r t dwxYw)z alias of conn.recv(n) zadb recv timeout)rUrecvrrr r$ns rr[zAdbConnection.recvxs9 199>>!$ $~~ 1/0 0 1s<cj |j|S#tj$r tdwxYw)Nadb read timeout) _read_fullyrrr r\s rreadzAdbConnection.reads7 1##A& &~~ 1/0 0 1s2cP|jd}tj|dS)Nlittle)raint from_bytesrXs r read_uint32zAdbConnection.read_uint32s yy|~~dH--r&c|}d}|dkDr9|jj|}|s |S||z }|t|z }|dkDr9|S)Nr&r)rUr[len)r$r]tbufferchunks rr`zAdbConnection._read_fullysZ !eIINN1%E  eOFCKA !e  r&c |j|}t ||krt d|dt ||S#tj$r tdwxYw)Nr_z Expected z bytes, got )r`rrr riEOFErrorr$r]rYs r read_exactzAdbConnection.read_exactsf 1##A&D t9q=Yqcc$i[AB B ~~ 1/0 0 1s =Ac|jd}|jjdjt |jd|zy)Nutf-8z{:04x})encoderUrWformatri)r$cmd cmd_bytess r send_commandzAdbConnection.send_commands=JJw'  xs9~6==gFRSr&cJ|j|jdd}|S)Nrrreplaceerrorsradecoderos r read_stringzAdbConnection.read_strings$yy|""79"= r&cx|jd}|s tdt|d}|j|S)z. Raises: AdbError rczconnection closed)r~r re)r$lengthsizes rread_string_blockzAdbConnection.read_string_blocks? !!!$./ /62%%r&cjd} |jd}|sn||z }|r|j|dS|S)zp read until connection close :param encoding: default utf-8, if pass None, return bytes r&iryrzr|)r$encodingcontentrls rread_until_closezAdbConnection.read_until_closesK IIdOE u G  >Fw~~hy~9R7Rr&c|jd}|tk(rt|j|tk(rytd|z)NrczUnknown data: %r)ra_FAILr r_OKAYrXs r check_okayzAdbConnection.check_okaysEyy| 5=41134 4 U] )D011r&N)rstrrre)returnbool)rz socket.socket)rYbytesrre)r]rerrrre)rur)r]rerrrr)rr)rz str | NonerzUnion[str, bytes])__name__ __module__ __qualname__r%r9r"propertyrErrNrSrUrWr[rargr`rprwr~rrrrMr&rrr+s+ M, %##  $11 . T & S2r&rceZdZ d ddZeddZeddZdddZdZdZ dddZ ddd Z d d!d Z d"d Z d#d Zd$d Zdd%dZd dZeddded dZeddded&dZy)' BaseClientNc|s tjjdd}|s)ttjjdd}||_||_||_y)z Args: host (str): default value from env:ANDROID_ADB_SERVER_HOST port (int): default value from env:ANDROID_ADB_SERVER_PORT ANDROID_ADB_SERVER_HOSTz 127.0.0.1ANDROID_ADB_SERVER_PORTiN)r=environgetre_BaseClient__host_BaseClient__port_BaseClient__socket_timeout)r$rrsocket_timeouts rr%zBaseClient.__init__sO::>>";[IDrzz~~&?FGD  .r&c|jSr)rrDs rrzBaseClient.host {{r&c|jSr)rrDs rrzBaseClient.portrr&c|xs |j} t|j|j}|r|jj ||S#t $r tdwxYw)zO connect to adb server Raises: AdbTimeout r,)rrrrrUr TimeoutErrorr )r$r_conns rmake_connectionzBaseClient.make_connectionsb 2T22 >!$++t{{;E %%g.L ><= = >s >AA&c|j5}|jd|jt|j dcdddS#1swYyxYw)z? 40 will match 1.0.40 Returns: int z host:versionrN)rrwrrerr$cs rserver_versionzBaseClient.server_versionsO  ! ! # 2q NN> * LLNq**,b1 2 2 2s ;AAct|j|jr;|j5}|j d|j dddyy#1swYyxYw)zP adb kill-server Send host:kill if adb-server is alive z host:killN)rrrrrwrrs r server_killzBaseClient.server_killsX dkk 2%%' 1{+    3  s "AA&cX|j|5}g}|r|jd|gn|jd|jd|zdz|z|jdj ||j |j dddy#1swYyxYw)a~ Same as wait-for-TRANSPORT-STATE Args: serial (str): device serial [default None] transport (str): {any,usb,local} [default any] state (str): {device,recovery,rescue,sideload,bootloader,disconnect} [default device] timeout (float): max wait time [default 60] Raises: AdbError, AdbTimeout r host-serialrz wait-for--:N)rextendappendrwjoinr)r$serial transportstaterrcmdss rwait_forzBaseClient.wait_fors ! !' ! 2 aD ]F34 F# KK i/#5= > NN388D> * LLN LLN   s BB  B)c|j|5}|jd|z|j|jcdddS#1swYyxYw)a adb connect $addr Args: addr (str): adb remote address [eg: 191.168.0.1:5555] timeout (float): connect timeout Returns: content adb server returns Raises: AdbTimeout Example returns: - "already connected to 192.168.190.101:5555" - "unable to connect to 192.168.190.101:5551" - "failed to connect to '1.2.3.4:4567': Operation timed out" rz host:connect:N)rrwrr)r$addrrrs rrzBaseClient.connect%sR" ! !' ! 2 )a NN?T1 2 LLN&&( ) ) )s 4AAc |j5}|jd|z|j|jcdddS#1swYyxYw#t$r|rYywxYw)a adb disconnect $addr Returns: content adb server returns Raises: when raise_error set to True AdbError("error: no such device '1.2.3.4:5678') Example returns: - "disconnected 192.168.190.101:5555" zhost:disconnect:N)rrwrrr )r$r raise_errorrs r disconnectzBaseClient.disconnect;sl %%' -11D89 **, - - -  s.A4A AAAA A+*A+c# Kg}|j5}|jd|j |j}|j |}|j ||D]}||}?#1swYyxYww)aT Report device state when changes Args: limit_status: eg, ['device', 'offline'], empty means all status Returns: Iterator[DeviceEvent], DeviceEvent.status can be one of ['device', 'offline', 'unauthorized', 'absent'] Raises: AdbError when adb-server was killed zhost:track-devicesN)rrwrr_output2devices _diff_devices)r$ orig_devicesroutput curr_devicesevents r track_deviceszBaseClient.track_devicesPs  ! ! # ,q NN/ 0 LLN,,.#33F; !// lK EK +  , ,sBA"A77B<Bcg}|jD]S}|jjdd}t|dk7r3|\}}|j t d||U|S)N r()maxsplit) splitlinesstripsplitrir DeviceEvent)r$rdeviceslinefieldsrstatuss rrzBaseClient._output2devicesisn%%' >DZZ\''q'9F6{a#NFF NN;tVV< =  > r&c#Kt|j|D]}td|jdt|j|D]%}td|j|j'yw)NFabsentT)set differencerrr)r$origcurrds rrzBaseClient._diff_devicessslT%%d+ 9AeQXXx8 8 9T%%d+ 8AdAHHahh7 7 8sAc|j5}d}|rdj|}|j||j|j }g}|j D]C}|j }t|dk7r"|r |d|k7r-|jt|E|cdddS#1swYyxYw)Nzhost:list-forwardzhost-serial:{}:list-forwardr+r) rrtrwrrrrrir ForwardItem)r$rrlist_cmdritemsrpartss r forward_listzBaseClient.forward_listys  ! ! # q*H8??G NN8 $ LLN))+GE**, 2 u:?eAh&0 [%01  2   s B B;;Cc |j5}d|dg}|r|jd|j|dz|z|jdj||j dddy#1swYyxYw)z Args: serial (str): device serial local, remote (str): tcp: or localabstract: norebind (bool): fail if already forwarded when set to true Raises: AdbError rforwardnorebind;rN)rrrwrr)r$rlocalremoterrrs rrzBaseClient.forwardst ! ! # q!695D J' KK f, - NN388D> * LLN    s A A::Bz0.15.0z1.0.0zuse Device.reverse instead) deprecated_in removed_indetailscurrent_versionc |j5}|jd|z|jd|dz|zg}|jdj||jdddy#1swYyxYw)z Args: serial (str): device serial remote, local (str): tcp: or localabstract: norebind (bool): fail if already reversed when set to true Raises: AdbError host:transport:zreverse:forwardrrN)rrwrr)r$rrrrrrs rreversezBaseClient.reversesr ! ! # q NN,v5 6 LLN%v|e'; * LLN    s AA99Bzuse Device.reverse_list insteadc |j5}|jd|z|j|jd|j|j}g}|j D];}|j }t |dk7r"|jt|dd=|cdddS#1swYyxYw)Nrzreverse:list-forwardr+r() rrwrrrrrir ReverseItem)r$rrrrrrs r reverse_listzBaseClient.reverse_lists  ! ! # q NN,v5 6 LLN NN1 2 LLN))+GE**, 6 u:? [%)45  6    s B'CC )NNN)r Optional[str]rz Optional[int]rOptional[float]rrr)rrrr)Nanydevice<)rrrrrrrfloat)rrrrrr)F)rrrrrr)rzIterator[DeviceEvent])rr)rtyping.List[DeviceEvent]rr)rzUnion[None, str]rzList[ForwardItem])rrrzList[ReverseItem])rrrr%rrrrrrrrrrrrrrrrrrrMr&rrrs#"*. ///( /& >2 >),*,28 $$h"4 +- -"h"9 +-  - r&r)rrrrerr)__doc__ __future__rior=r0rr?rItypingrrrr deprecationradbutils._utilsr adbutils.errorsr r r adbutils._protoadbutils._versionrrrrobjectrrrMr&rrsh#   22"$DD) X2FX2v}}r&