ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZmZddlmZmZmZm Z mZddlmZej.dZe j2rdd lmZndd lmZd ZGd d ej<ZGddZ ejBjEe dZ#dZ$GddejJZ&ejNjEe&dZ(ejReZ)nejRjTZ)ejVeZ+ne(ejVjTZ+ejXeZ,ne(ejXjTZ,e j2r$ejZe+Z+ejZe,Z,ee+re#e+Z+ej\Z.ej^Z0ejbZ1ejdZ2ejfZ3ejhZ4ejjZ5ejlZ6ejnZ7ejpZ8ejrZ9ejtZ:ejvZ; ejxZ< ej|Z>e j~Z@dZAGddZBd7dZCdZDdZEd8dZFd8dZGd9dZHdZIdZJeJeId ZKd!ZLdaMd"ZNd#ZOd$ZPd%ZQd&ZRd'ZSd(ZTd8d)ZUe j2rd*ZVejZXnd+ZVe jZXd,ZZd:d-Z[da\d.Z]d/Z^d0Z_d1e`fd2Zaejd3Zce j2rd4ZdedejzZf d;d5Zgy d;d6Zgy#e=$rYwxYw#e=$rYwxYw)<N)BinaryIO_)getattropen)encodingerrorpolicypycompattypelib) resourceutilosutil)windows)posixcB |jS#t$rYywxYw)NF)isattyAttributeError)fps ???{{ Mr$c$eZdZdefdZdZdZy)rCr1c||_yr/r0r2s rr3zWriteAllWrapper.__init__r4r$c.t|j|Sr/r6r7s rr9zWriteAllWrapper.__getattr__r:r$c|jj}t|}t|}d}||kr|||d}|r||z }||kr|S)Nr)r1r& memoryviewlen)r r?write1mtotal_to_write total_writtencs rr&zWriteAllWrapper.writes^ qMQ n,q()A" n,r$N)r'r(r)rr3r9r&r+r$rrCrC~sX( r$rCctt|tr|St|tjr|St|Sr/)rBrCrGrIrOs r_make_write_allr]s5&/* &"++,  6 ""r$cF|dk\rtd|zStd| zS)zpreturn a message describing a subprocess status (codes from kill are negative - not os.system/wait encoding)rsexited with status %dskilled by signal %dr)codes r explainexitr`s/ qy)*T11 # $u ,,r$c4eZdZdZdZdZdZdZdZdZ y) _pfilez;File-like wrapper for a stream opened by subprocess.Popen()c ||_||_yr/)_proc_fp)r procrs rr3z_pfile.__init__s r$cj|jj|jjSr/)recloserdwaitr s rrhz _pfile.closes! zz  r$c,t|jSr/)iterrerjs r__iter__z_pfile.__iter__sDHH~r$c.t|j|Sr/)rrer7s rr9z_pfile.__getattr__stxx&&r$c|Sr/r+rjs r __enter__z_pfile.__enter__s r$c$|jyr/)rh)r exc_type exc_valueexc_tbs r__exit__z_pfile.__exit__s  r$N) r'r(r)r*r3rhrmr9rprur+r$rrbrbs#E! 'r$rbcv|dk(r t||S|dk(r t||Stjd|z)Nrbswbsunsupported mode: %r) _popenreader _popenwriterr ProgrammingError)cmdmodebufsizes rpopenr~sA u}C)) C))  !84!? @@r$ctjt|d|ttj}t ||j S)NT)shellr} close_fdsstdout) subprocessPopen tonativestrclosefdsPIPErbrr{r}ps rrxrxs=C  A !QXX r$ctjt|d|ttj}t ||j S)NT)rr}rstdin)rrrrrrbrrs rryry s=Coo  A !QWW r$c tjt|ddttjtjt |}|j |jfS)NT)rr}rrrenv)rrrrr tonativeenvrr)r{rrs rpopen2rsP Coo   A 77AHH r$c.t||\}}}}|||fSr/)popen4)r{rrrstderrrs rpopen3r&s$%c3/E661 &&  r$c tjt|d|ttjtjtjt |}|j |j|j|fS)NT)rr}rrrrr) rrrrrrrrr)r{rr}rs rrr+s]Coo   A 77AHHahh ))r$ctjt|dttjtj}|j |\}}|S)z9filter string S through command CMD, returning its outputT)rrrr)rrrrr communicate)r?r{rpoutperrs r pipefilterr9sGCoo  Aq!JD$ Kr$cHd\}} tjd\}}tj|d}|j ||j tjd\}}tj ||j d|}|j d|}t|}tjdk(r|d zrd }|r,tjtd |t|fzt|d 5}|jcd d d  |rtj| |rtj|SS#t $rY%wxYw#t $rYSwxYw#1swYnxYw |rtj|n#t $rYnwxYw |rtj|y y #t $rYy wxYw# |rtj|n#t $rYnwxYw |rtj|ww#t $rYwwxYwxYw)zfilter string S through a pair of temporary files with CMD. CMD is used as a template to create the real command to be run, with the strings INFILE and OUTFILE replaced by the real names of the temporary files generated.)NNs hg-filter-in-)prefixwbshg-filter-out-sINFILEsOUTFILEOpenVMSrscommand '%s' failed: %srwN)r mkstemposfdopenr&rhreplacesystem sysplatformr Abortrr`rreadunlinkOSError)r?r{innameoutnameinfdroutfdr_s r tempfilterrFs !OFG''/?@ f YYtT "    !))1BCw kk)V,kk*g.c{   : -$(D ++,-k$6G0HH '5 !R779" !  &!  '"       " ! !  &!     '"     &!     '"   sDGE1 G"E:E" EE" E.-E.1E:6G?F F#"F#'G G  G H!G)(H!) G52H!4G55H!9HH! HH!HH!)s tempfile:spipe:ctjD];\}}|j|s|||t|dj cSt ||S)zMfilter a string through a command that transforms its input to its outputN) _filtertableitems startswithrVlstripr)r?r{namefns rfilterrqsW!&&(b >>$ aSY[)0023 3) a r$ct_tjjd}tj d}|rt |tStjrUttdddk(r"t tjdtSt tjtStjs[tjjt|ddd k(r.t tj |j"tSt t%d xs0tjjtj&d tS) z]return location of the 'hg' executable. Defaults to $HG or 'hg' in the search path. NHG__main__frozen macosx_appEXECUTABLEPATH__file__hgshgr) _hgexecutabler environgetsysmodules_sethgexecutabler mainfrozenrr sysexecutablerDrpathbasenamefsencoderfindexesysargv)rmainmods r hgexecutabler}s     ! !% (++j) R  $ $ &sHd+|; !1!12C!DE !!7!78 ""  *b!ABdJ X..w/?/?@ A  G"''"2"283C3CA3F"G  r$c|ay)z#set location of the 'hg' executableN)r)rs rrrs Mr$ct|dd} |xr||jk(S#tj$rYywxYw)NfilenoF)rrrGUnsupportedOperation)fstdfrs r _testfilenorsE Q$ 'F3&(dkkm33 " "s+AAc6t|tjSr/)rr __stdin__rs risstdinrs q#-- ((r$c6t|tjSr/)rr __stdout__rs risstdoutrs q#.. ))r$c|j||}}t|trtj|j }tj tjtj}tj||j tj|tj|d}t|trotj|j }tjtj |j tj|d}||fS)aDuplicate streams and redirect original if (uin, uout) are stdio If uin is stdin, it's redirected to /dev/null. If uout is stdout, it's redirected to stderr so the output is still readable. Returns (fin, fout) which point to the original (uin, uout) fds, but may be copy of (uin, uout). The returned streams can be considered "owned" in that print(), exec(), etc. never reach to them. rbr)r>rrrduprrdevnullO_RDONLYdup2rhrrr)uinuoutfinfoutnewfdnullfds r protectstdiors JJLTC3szz|$R[[1  % iit$4 t{{}%  /yy% 9r$c|j||f||ffD]L\}}||us tj|j|j|j Ny)z@Restore (uin, uout) streams from possibly duplicated (fin, fout)N)r>rrrrh)rrrrruifs r restorestdiorsPJJL:d|,3 C< GGAHHJ - GGI-r$cdttj}|r(|jfd|j Dt |d<|S)z>return environ with optional override, useful for shelling outcD||dury|durytj|S)z9convert python object into string that is useful to shellF0T1)r bytestr)vals rpy2shellzshellenviron..py2shells+ ;#, $;$$r$c38K|]\}}||fywr/r+).0kvrs r zshellenviron..s@1Ax{#sr)dictr rupdaterr)rrrs @r shellenvironrsE% x C @ @@CJ Jr$c@tj|t|Sr/)rE shelltocmdexerr{rs r shelltonativers%% c"  r$c|Sr/r+rs rrrs r$c6tjt|S)zHconvert the environment from bytes to strings suitable for Popen(), etc.)r rapplyr)rs rrrs ??; ,,r$c  tjt|}| t |rIt j t|dtt|tjt|}nt jt|dtt|tjt|t jt j}t|jj dD]}|j#||j%|j&}tj(dk(r|dzrd}|S#t$rYAwxYw)zenhanced shell command execution. run with environment maybe modified, maybe in different dir. if out is specified, it is assumed to be a file-like object that has a write() method. stdout and stderr will be redirected to out.T)rrrcwd)rrrrrrr$rrr)rr> Exceptionrrrcallrrrr rrrSTDOUTrlreadliner&ri returncoder)r{rroutrrcrflines rrrs     w C {hsm __  C  S1    C  S1??$$ --s3D IIdO4 __z)b1f  I7    sE EEc8tjr9dtjvryt t ddrt j Sytjxs@tjjdxstjjdS)zAre we running in a GUI?sSSH_CONNECTIONFisguiNTsDISPLAYsWAYLAND_DISPLAY) r isdarwinr rrrrrDrr+r$r_guir!s|  0 0 0 VWd +<<> !    8##J/ 8##$67 r$c.t tatSr/)_is_guirr+r$rguir5s& Nr$ctjr9ttdddk(rtj dgSt jgStS)zReturn the command used to execute current hg This is different from hgexecutable() because on Windows we want to avoid things opening new shell windows like batch files, so we get either the python call or current executable. rNrr) rrrrr rr r _gethgcmdr+r$rhgcmdr<sM 3$ '< 7$$%678 8**+ + ;r$returnctfd}d}ttdd}|tj||} t|}|s[|vs t |s/|s( |$tjtj |yt jd|s[||$tjtj |S#|$tjtj |wxYw)a_Execute the argument list in a detached process. condfn is a callable which is called repeatedly and should return True once the child process is known to have started successfully. At this point, the child process PID is returned. If the child process fails to start or finishes before condfn() evaluates to True, return -1. cLjtjyr/)addrri)signumframe terminateds rhandlerzrundetached..handler^srwwy!r$NSIGCHLDrg?)setrsignal spawndetachedtestpidrtimesleep)argscondfnr prevhandlerrpidrs @r rundetachedr*Ls J"Kfi.GmmGW5  D!(z!vx  " MM&..+ 6  JJsO(  " MM&..+ 6   " MM&..+ 6 s(C C*C(C:c#Ktjtjggfd} tjtj|dr'tjtjdrty#t$rd=YDwxYw#r'tjtjdrtwxYww)a Inhibit SIGINT handling on a region of code. Note that if this is called in a non-main thread, it turns into a no-op. Args: warn: A callable which takes no arguments, and returns True if the previous signal handling should be restored. cr*tjtjdd=jdy)NrT)r!SIGINTappend)r&oldsiginthandler shouldbailwarns rdisabledsiginthandlerz.uninterruptible..disabledsiginthandlers6 6 MM&--)9!)< = #$r$rN)r! getsignalr- ValueErrorKeyboardInterrupt)r1r2r/r0s` @@runinterruptibler6ws((78J $ $ MM&--)> ?   MM&--)9!)< = # #  $ # $  MM&--)9!)< = # # s:.C$BB!2C BB!BB!!3CCc  d}|Ftj}|j||j|j dt j tjt||t|dt|||} ||| j||jyy#|jwwxYw)z1Spawn a command without waiting for it to finish.NrT)rrr creationflagsrrr) r unnamedtempfiler&r>seekrrrrr_creationflagsrirh) scriptrrrr ensurestart record_wait stdin_bytesrrs r runbgcommandrAs E& 002 K(  1    V4$, A&AFF#  !u  !s B B66C c ^ tjr t|r|}n+t|tr|g}dj d|D}|d|z}d} nd} d}d} |t j} nFtj} | j|| j| jd|t j}|t j}t j|d|d| |||  |%| #t| trJ| j |s1t!j" fd  } d| _| j'y j(} | || yy#t$r ||d wxYw#|&| #t| trJ| jwwwxYw) aSpawn a command without waiting for it to finish. When `record_wait` is not None, the spawned process will not be fully detached and the `record_wait` argument will be called with a the `Subprocess.wait` function for the spawned process. This is mostly useful for developers that need to make sure the spawned process finished before a certain point. (eg: writing test) c32K|]}t|ywr/) shellquote)rxs rrzrunbgcommand..s:cz!}csNs( %s ) &TFr)rrrrrrstart_new_sessioncjSr/)ri)rsrzrunbgcommand..s r$)target)r rrrBbytesjoinrDEVNULLr:r&r>r;rrintrh threadingThreaddaemonstartri)r{rrrrr>r?r@r=rGrtr rs @rrArAs$    E F#u%eYY:c::F   6)F $  % K ""** 002 K(  1 ~#++~#++  "3 A&5+<%eS111    7AAH GGIJ&J''' &C   &5+<%eS111 ,=&sBE**E??F*F,)rwrr/)Nr)NNN)FNNTNN)h contextlibrrGrr!rrrPr$typingri18nrr rrrr r r r utilsr importmodrrDrrErrrHrr-rIregisterrMrPBinaryIO_ProxyrCIOBaser]rbufferrrrFrgethgcmdrgetusergetpid hidewindowreadpipe setbinarysetsignalhandlerrE shellsplitr"sshargsr# setprocnamer unblocksignalisposixrr`rbr~rxryrrrrrrrrrrrrrrrrr strfromlocalridentityrrrrrrrOr*contextmanagerr6DETACHED_PROCESSCREATE_NEW_PROCESS_GROUPr<rAr+r$rros    !   ( # &$:bll:   ./'@g,,( ?# #$99 IE II  E:: YF SZZ.. /F:: YF SZZ.. /F  X   'F X   'F &>  'F                  ,,   &&       $$K ((M   -2A  ! * "L  8 )*2$  ''K##K- #L  (  ( ( V $$B "   - - .$XR(E    s$ M" MMMM! M!