jW^fddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z m Z m Z  ddl mZddlmZmZddlmZddlmZddlmZmZdd ldd lmZdd lmZmZmZm Z dd l!m"Z"dd l#m$Z$dZ%ejLe'Z(GddZ)GddZ*y#e$r ddlmZYuwxYw)) annotationsN)ListOptionalUnionoverload)Literal)ImageUnidentifiedImageError) deprecated)ShellReturnRaw) AdbConnection BaseClient)*) StrOrPathLike) StopEventadb_path get_free_port list2cmdline) __version__)AdbErroriXceZdZdZ d5 d6dZdZed7dZdef d8dZ d9dZ d:dZ d:d Z d:d Z d:d Zed;d Zd ZeddZededd d?dZedefdd d@dZededf dAdZdeddf dBdZeefddd dCdZeefddd dDdZeeddf dEdZedddf dFdZefdGd ZefdGd!ZdHdId"ZdJd#ZdKd$ZdLdMd%ZdNd&ZdHdOd'ZdNd(ZdPd)Z dQd*Z!dRd+Z"e#d,d-e$d./dSd0Z% dTd1Z&d2Z'dUd3Z( dV dWd4Z)y)X BaseDevicezBasic operation for a deviceNc||_||_||_i|_i|_|s |s t d|j y)z Args: client (BaseClient): AdbClient instance serial (str): device serial transport_id (int): transport_id z)serial, transport_id must set atleast oneN)_client_serial _transport_id _properties _featuresr_prepare)selfclientserial transport_ids V/var/www/api.ozpay.ru/ozpay/venv/lib/python3.12/site-packages/adbutils/_device_base.py__init__zBaseDevice.__init__)sC  )lFG G cy)zrewrite in sub classNr s r$rzBaseDevice._prepare=sr&c|jSN)rr)s r$r"zBaseDevice.serial@s ||r&c|jj|}|r|jr3|jd|jd||j |S|j r3|jd|j d||j |St d|jr0|jd|j|j |S|j r|jjdk\rA|jd|j |j |jd |S|jd |j z|j |St d) Ntimeoutzhost-transport-id::z host-serial:zshould not reach herezhost:transport-id:)zhost:tport:serial:zhost:transport:) rmake_connectionr send_command check_okayr RuntimeErrorserver_versionread)r commandr.cs r$open_transportzBaseDevice.open_transportDsY LL ( ( ( 9 !!!3D4F4F3Gq RS 0/dll^1WIFG *'##:;;!!!3D4F4F3GHI <<..0B6NN%7 ~#FGLLNFF1I NN#4t||#CDLLN##:;;r&cp|j|5}|jcdddS#1swYyxYwr+)r:read_string_block)r cmdr9s r$_get_with_commandzBaseDevice._get_with_commandgs3   % )&&( ) ) )s,5c$|jdS)z/return device state {offline,bootloader,device}z get-stater>r)s r$ get_statezBaseDevice.get_stateks%%k22r&c$|jdS)z1return the real device id, not the connect serialz get-serialnor@r)s r$ get_serialnozBaseDevice.get_serialnoos%%n55r&c$|jdS)zexample return: usb:12345678Yz get-devpathr@r)s r$ get_devpathzBaseDevice.get_devpathss%%m44r&cf|jd}t|jd|_|S)z Return example: 'abb_exec,fixed_push_symlink_timestamp,abb,stat_v2,apex,shell_v2,fixed_push_mkdir,cmd' features,)r>setsplitr)r rGs r$ get_featureszBaseDevice.get_featuresws. ))*5X^^C01r&cb|j|j|jdS)N)serialnodevpathstate)rCrErAr)s r$infozBaseDevice.infos0))+'')^^%  r&c8dj|jS)NzAdbDevice(serial={}))formatr"r)s r$__repr__zBaseDevice.__repr__s%,,T[[99r&ct|Sr+)Propertyr)s r$propzBaseDevice.props ~r&c |jrtd|jgn tg}|j| tj|tj tj jdS#tj$rD}|jddr(td||jjddYd }~y d }~wwxYw) zRun adb command use subprocess and get its content Returns: string of output Raises: EnvironmentError z-s)stdinstderrutf-8 raise_errorT subprocessignoreerrorsN) rrextendr\ check_outputDEVNULLSTDOUTdecodeCalledProcessErrorgetEnvironmentErroroutput)r argskwargscmdses r$ adb_outputzBaseDevice.adb_outputs48<< D$,,/hj\ D **J..z7H7HfWo ,, zz-.& $(Q/ sABC:CCct|ttfr t|}|j }|j d|z|j |S)zOpen a shell connection to the device Args: cmdargs (str | list | tuple): command args Returns: AdbConnection: shell connection object zshell:) isinstancelisttuplerr:r3r4)r cmdargsr9s r$ open_shellzBaseDevice.open_shellsI ge} -"7+G    ! x')* r&rZTcyr+r(r rrstreamr.encodingrstrips r$shellzBaseDevice.shellsr&F)rvr.rxcyr+r(rus r$ryzBaseDevice.shellsr&)rxcyr+r(rus r$ryzBaseDevice.shellsr&cyr+r()r rrrvr.rxs r$ryzBaseDevice.shellsr&cHt|ttfr t|}|j |}|r|S|r|j j | |j|}|j|r|r|jS|S|S#|jwxYw)aRun shell inside device and get it's content Args: rstrip (bool): strip the last empty line (Default: True) stream (bool): return stream instead of string output (Default: False) timeout (float): set shell timeout encoding (str): set output encoding (Default: utf-8), set None to make return bytes rstrip (bool): strip the last empty line, only work when encoding is set Returns: string of output when stream is False AdbConnection when stream is True Raises: AdbTimeout Examples: shell("ls -l") shell(["ls", "-l"]) shell("ls | grep data") rw) rorprqrrsconn settimeoutread_until_closecloserx)r rrrvr.rwrxr9rhs r$ryzBaseDevice.shells: ge} -"7+G OOG $ H  FF  g & ''':F GGI &,6==? 8& 8  GGIs BB!)rxv2cyr+r(r rrr.rwrxrs r$shell2zBaseDevice.shell2 sr&cyr+r(rs r$rzBaseDevice.shell2sr&cyr+r()r rrr.rxrs r$rzBaseDevice.shell2!sr&c t|ttfr t|}|r^|js-t |j jd|_d|jvrd}tjd|r|j||}n|j||}|rt|j|j|jj!|d|j"j!|d|j$j!|d}|r]|jj'|_|j"j'|_|j$j'|_|S)a" Run shell command with detail output Args: cmdargs (str | list | tuple): command args timeout (float): set shell timeout, seconds encoding (str): set output encoding (Default: utf-8), set None to make return bytes rstrip (bool): strip the last empty line, only work when encoding is set shell_v2 (bool): attempt to use the shell_v2 protocol (and fail if not supported by device) Returns: ShellOutput Raises: AdbTimeout rHshell_v2Fz.shell_v2 specified but not supported by devicereplacer^r8 returncoderhrYstdout)rorprqrrrIrKrJloggerwarning _shell_v2 _shell_v1 ShellReturnr8rrhrdrYrrx)r rrr.rwrxrresults r$rzBaseDevice.shell2*s/. ge} -"7+G >>!$T%6%6%8%>%>s%C!D/OP ^^GW5F^^GW5F  !,,}}++HY+G}}++HY+G}}++HY+G F & 4 4 6 & 4 4 6 & 4 4 6  r&c"t|tsJd}|d|dz}|j||dd}|j|j }|dk(r t d||t ||t|zd}|d|}t|||S) NzX4EXIT:z; echo z$?F)r.rwrxzshell output invalid)r8rrh) rostrryrfindencoderintlenr )r rrr.MAGICnewcmdrhrindexrs r$rzBaseDevice._shell_v1^s'3'''WUG2.. 67TRW Xelln- R<166B BU 3 567 g*VTTr&c|j|} |jd||jtj}tj}tj}d} |j d}|d} t j|ddd} | dk(r7|j | } | dk(r#|j| |j| n3| d k(r#|j| |j| n | d k(r| d}nt|||j|j|j |jS#|jwxYw) Nr-z shell,v2:rlittle) byteorderr) r:r3r4ioBytesIO read_exactr from_byteswriter getvaluer) r rrr.r9 stdout_buffer stderr_buffer output_buffer exit_codeheadermsg_idlengthdatas r$rzBaseDevice._shell_v2jsL     0!  NNYwi0 1 LLNJJLMJJLMJJLMIaq xHQ;||F+Q;!''-!''-q[!''-!''-q[ $QI!""$$--/$--/$--/  GGIAGGIs D?E$$E6cnd}|r|dz }|d|d|z }|j|}|jy)Nforwardz :norebindr/;r:r)r localremotenorebindr=r9s r$rzBaseDevice.forwardsD  ; C 5'6(##    $  r&ct|trdt|z}|jD]j}|j|j k(s|j |k(s-|jjdsIt|jtddcSt}|jdt|z||S)z(forward remote port to local random porttcp:N) rorr forward_listr"rrr startswithrrr)r rf local_ports r$ forward_portzBaseDevice.forward_ports fc "c&k)F""$ 3ADLL(HH&GG&&v.1773v;=122  3#_  Vc*o-v6r&cd}|j|5}|j}g}|jD]8}|j}t |dk7r"|j t |:|cdddS#1swYnxYw|jj}|Dcgc]}|j|jk(s| c}Scc}w)Nz list-forwardr) r:r< splitlinesrJrappend ForwardItemrrr"r)r r=r9contentitemslinepartsitems r$rzBaseDevice.forward_lists   % ))+GE**, 2 u:? [%01  2     ))+!&F$++*EFFFsAA==B'C C cd} |jd|}|r|jyy#t$r|rY"wxYw#|r|jwwxYw)a1 Remove existing forward local connection. Args: local (str): Local address of the forward to close, for example tcp: raise_non_found (bool): Whether to raise an `AdbError` if the local forward address doesn't exist, default is True. Nz killforward:)r:rr)r rraise_non_foundr9s r$forward_removezBaseDevice.forward_removesh#' ##l5'$:;A      s, ;>;>AcB|jdjy)z)Remove all forwarded network connections.zkillforward-allNrr)s r$forward_remove_allzBaseDevice.forward_remove_alls -.446r&c(|j5}dg}|r|jd|j|dz|z|jdj||j |j dddy#1swYyxYw)a Args: serial (str): device serial remote, local (str): - tcp: - localabstract: - localreserved: - localfilesystem: norebind (bool): fail if already reversed when set to true Raises: AdbError zreverse:forwardrrr/N)r:rr3joinr4)r rrrr9ris r$reversezBaseDevice.reversesy " a%&D J' KK u, - NN388D> * LLN LLN   s A.BBc|j5}|jd|j|jdddy#1swYyxYw)z'Remove all reverse network connections.zreverse:killforward-allNr:r3r4r r9s r$reverse_remove_allzBaseDevice.reverse_remove_allsD  " a NN4 5 LLN LLN   s 2A  Ac|j5}|jd||j|jdddy#1swYyxYw)a Remove existing reverse remote connection. Args: serial (str): device serial remote (str): - tcp: - localabstract: - localreserved: - localfilesystem: Raises: AdbError zreverse:killforward:Nr)r rr9s r$reverse_removezBaseDevice.reverse_removesM " a NN1&: ; LLN LLN   s 5AAc V|j5}|jd|j|j}g}|j D];}|j }t |dk7r"|jt|dd= ddd|S#1swYSxYw)Nzreverse:list-forwardrr) r:r3r4r<rrJrr ReverseItem)r r9rrrrs r$ reverse_listzBaseDevice.reverse_lists  " 6a NN1 2 LLN))+GE**, 6 u:? [%)45  6 6  6 s BBB(cn|j}|jd|j|j}|dk(r t d|j}|dk7r|dk7r t d||j}|dk(r|j}|j}|j}|j}|j}|j} |j} |j} |j} |j} |j}d}| d k(rd }|dk(s|r|d z }|d k7r|dk7r t d |j |}t ||k7rtd|t |tj|||f|}|S)z}Capture device screen and return PIL.Image object (Not very stable) Raises: NotImplementedError z framebuffer:zUnsupported version 16 zUnsupported bpp(bits per pixel)rRGBrBGRARGBAzUnsupported color formatzframebuffer size not match) r:r3r4 read_uint32NotImplementedErrorr7rr r frombytes)r r9versionbppsizewidthheight red_offset red_length blue_offset blue_length green_offset green_length alpha_offset alpha_length color_formatbufferimages r$ framebufferzBaseDevice.framebuffer s    ! ~& --/ b=%&>? ?mmo "9%&GM M}} 19==?D ]]_ ]]_ mmo mmo }} }} }} }}  !  L "9 C L 6 !le&;%&@A A v;$ ()EtSQW[Y Y ufovF r&z2.6.0z3.0.0zuse sync.push instead) deprecated_in removed_incurrent_versiondetailsc:|jj||S)z alias for sync.push )syncpush)r rrs r$rzBaseDevice.push:syy~~eV,,r&crdfd }|j} ||S#|jxYw)z Used to connect a socket (unix of tcp) on the device Returns: socket object Raises: AdbError, ValueError ctjk(rKttsJ|j dt z|j |jStjtjfvrBtt sJ|j dz|j |jStjtjtjtjfvr<|j dzt z|j |jStd)Nrzlocalabstract:r/zUnsupported network type)NetworkTCProrr3rr4UNIXLOCAL_ABSTRACTLOCAL_FILESYSTEMLOCALDEVLOCAL_RESERVED ValueErrorr)r9addressnetworks r$_connectz.BaseDevice.create_connection.._connectKs'++%!'3///vG 45 66MW\\7+A+ABB!'3////'9: 66M((  &&  w}s7|;< 66M!!;WEEr&)r9r return socket.socketr)r r r r r9s `` r$create_connectionzBaseDevice.create_connection?s: *    ! A;   GGI s#6c|j5}|jd|j|jcdddS#1swYyxYw)zjrestart adbd as root Return example: cannot run as root in production builds zroot:N)r:r3r4rrs r$rootzBaseDevice.rootgsH " (a NN7 # LLN%%' ( ( (s 1A  Ac|j5}|jdt|z|j|j cdddS#1swYyxYw)zurestart adbd listening on TCP on PORT Return example: restarting in TCP mode port: 5555 ztcpip:N)r:r3rr4r)r portr9s r$tcpipzBaseDevice.tcpipssQ  " (a NN8c$i/ 0 LLN%%' ( ( (s =AA!crAttrtjttjsJ|r|j dd fd } d d}t }|j |d}tj|jdd} tjd ||| ||fd j|S) a Args: file (str): file path to save logcat clear (bool): clear logcat before start re_filter (str | re.Pattern): regex pattern to filter logcat command (str): logcat command, default is "logcat -v time" Example usage: >>> evt = device.logcat("logcat.txt", clear=True, re_filter=".*python.*") >>> time.sleep(10) >>> evt.stop() zlogcat --clearc0yj|duS)NT)search)r re_filters r$ _filter_funcz'BaseDevice.logcat.._filter_funcs" ##D)5 5r&c |jjddd}|jsM|j}|sn:||r!|j ||j |jsM|j |j |jy#j |j |jwxYw)NrUTF-8r)rwr_)rmakefile is_stoppedreadlinerflushrdone)rvfdstevent filter_funcfsrcrs r$ _copy2filez%BaseDevice.logcat.._copy2files {{++C')+T**,==?D"4( 4(  **,      s A:B..2C T)rvwrr~logcat)nametargetridaemon)rrrbool)rvr r#z typing.TextIOr$rr%ztyping.Callable[[str], bool]) rorrecompilePatternryrpathlibPathopen threadingThreadstart) r fileclearrr8rr'r$rvr#s ` r$r)zBaseDevice.logcat~s& )S)JJy1 i4 44  JJ' ( 6  !   6  ( GD1||D!&&sW&=$|4  %' r&)NN)r!rr" Optional[str]r#z Optional[int])rr9)r8r9r.Optional[float]rr )r=rrr)rr)rdict)rz 'Property')rrUnion[str, list, tuple]rr ) rrr<rvz Literal[True]r.r:rw str | Nonerr ) rrr<rvLiteral[False]r.r:rw Literal[None]rbytes) rrr<rvr>r.r:rwrrr)rrr<rvr>r.r:rr) rrr<rvr-r.r:rwr=rz'typing.Union[AdbConnection, str, bytes])rrr<r.r:rwr?rr )rrr<r.r:rwrrr)rrr<r.r:rr)rrr<r.r:rwr=rz"Union[ShellReturn, ShellReturnRaw])rrrr.r:rr )F)rrrrrr-)rUnion[int, str]rr)rzList[ForwardItem]T)rrrr-rNone)rrC)rrrrrr-)rr)rzList[ReverseItem])rz Image.Image)rrrr)r rr rArr)rr)NFNzlogcat -v time) r7zOptional[StrOrPathLike]r8r-rz Optional[Union[str, re.Pattern]]r8rrr)*__name__ __module__ __qualname____doc__r%rpropertyr"_DEFAULT_SOCKET_TIMEOUTr:r>rArCrErKrPrSrVrmrsrryrrrrrrrrrrrrrr rrrrrr)r(r&r$rr&s&_c *7N[(#(,H_!$!6E! !F)365  :*  $;& (!     "'#:(  !    "'#: (!    "'#: (!  #:& ,(,,! ,  , 1,\$;  (!   $;  (!   $;  (!  $;&  2(2!2 2 ,2hBY UBY#J  G&7.( -^g';`wx-y-&&)8& &P ( ()-6:' =%==4 =  =  =r&rcTeZdZddZdZd d dZedZedZedZ y) rUc||_yr+)_d)r ds r$r%zProperty.__init__s r&cTd|jd|jd|jS)Nzproduct:z model:z device:)r*modeldevicer)s r$__str__zProperty.__str__s'$))GDJJrjs#   22*."*3)LL)$   8 $U U p99Y*)*sB"" B0/B0