o 3gi>@sddlmZddlZddlZddlmZddlmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZmZmZGd d d eZGd d d eZGdddeZGdddZGddde ZdS)) annotationsN)NamedTemporaryFile)CommandNotFoundConcreteCommandshquote)ProcInfo) BaseMachine)BaseEnv) LocalPath) RemotePath RemoteWorkdirStatRescs`eZdZdZddgZfddZddZdd Zd d Zd d Z ddZ ddZ ddZ Z S) RemoteEnvz?The remote machine's environment; exposes a dict-like interface_origremotecs|jd}|ddkr#|ds#tdd|ddD}n'dd}d d|D}d d|D}fd d|D}d d |D}tj|jd|d||_ |j |_ dS)Nz env -0; echorcss$|] }d|vr|ddVqdS=Nsplit.0linerO/home/garg/my-data/venv/lib/python3.10/site-packages/plumbum/machines/remote.py sz%RemoteEnv.__init__..rz env; echocss|] }|ddVqdSrrrrrrr scss$|] }t|dkr|dVqdS)rrN)lenrrrrr!s"c3s&|]}|d|dfVqdS)z printenv "z"; echoN)run)rkeysessionrrr"$cSsBi|]\}}|ddkr|ddr|ds||ddqS)rr r)rstrip)rr rrrr #s "z&RemoteEnv.__init__..:)_curr) _sessionrr%dictr splitlinessuper__init__pathrr(copyr)selfrenv0r(linesrkeysruns __class__r!rr-s    zRemoteEnv.__init__cCs$t|||jjd|dSNzunset )r __delitem__rr)rr0namerrrr8-s zRemoteEnv.__delitem__cCs0t||||jjd|dt|dS)Nexport r)r __setitem__rr)rr)r0r:valuerrrr<1s"zRemoteEnv.__setitem__cGs,tj||g|R|jjd|dSr7)r poprr)r)r0r:defaultrrrr>5sz RemoteEnv.popcOsFtj|g|Ri||jjdddd|DdS)Nr; css&|]\}}|dt|VqdS)rN)r)rkvrrrr<r#z#RemoteEnv.update..)r updaterr)rjoingetdictitemsr0argskwargsrrrrC9szRemoteEnv.updatecC |j|S)aEExpands any environment variables and home shortcuts found in ``expr`` (like ``os.path.expanduser`` combined with ``os.path.expandvars``) :param expr: An expression containing environment variables (as ``$FOO``) or home shortcuts (as ``~/.bashrc``) :returns: The expanded string)rexpandr0exprrrrrK?s zRemoteEnv.expandcCrJ)zExpand home shortcuts (e.g., ``~/foo/bar`` or ``~john/foo/bar``) :param expr: An expression containing home shortcuts :returns: The expanded string)r expanduserrLrrrrNIs zRemoteEnv.expandusercCs|j|jd<i}|jD]\}}||jvrt|||<q|jD]\}}||jvr2d||<q$||j|kr@|j|||<q$|S)znReturns the difference between the this environment and the original environment of the remote machinePATH)r.rDr(rFrstr)r0deltarArBrrrgetdeltaUs    zRemoteEnv.getdelta)__name__ __module__ __qualname____doc__ __slots__r-r8r<r>rCrKrNrS __classcell__rrr5rrs   rc@sFeZdZdZdZdddZeddZdd Zdd d Z dddZ dS) RemoteCommandrrautocCs,||_t|||dkr|jdS|dS)Nr\)rrr-custom_encoding)r0r executableencodingrrrr-ls zRemoteCommand.__init__cCs|jSNr[r0rrrmachinerszRemoteCommand.machinecCsd|jd|jdS)NzRemoteCommand(z, ))rr^rarrr__repr__vzRemoteCommand.__repr__rcKs|jj||fi|Sr`)rpopenrGrrrrfyszRemoteCommand.popen. nohup.outNTcCs|j|||||S)zRuns a command detached.)rbdaemonic_popen)r0cwdstdoutstderrappendrrrnohup|szRemoteCommand.nohup)r\)r)rgrhNT) rTrUrVrX QUOTE_LEVELr-propertyrbrdrfrnrrrrrZhs   rZc@s eZdZdS)ClosedRemoteMachineN)rTrUrVrrrrrqsrqc@s,eZdZddgZddZddZddZd S) ClosedRemote_obj __weakref__cCs ||_dSr`)rs)r0objrrrr-s zClosedRemote.__init__cCsdSr`rrarrrclosezClosedRemote.closecCst|jd)Nz has been closed)rqrsr9rrr __getattr__szClosedRemote.__getattr__N)rTrUrVrXr-rvrxrrrrrrs  rrc@sJeZdZdZeZeddZdNddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZeddZdOddddZddZd d!Zd"d#Zd$d%Zd&d'Zejd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Z d8d9Z! : ;dPdd?Z#d@dAZ$dBdCZ%dDdEZ&dFdGZ'dHdIZ(dJdKZ)dLdMZ*d:S)QBaseRemoteMachineaRepresents a *remote machine*; serves as an entry point to everything related to that remote machine, such as working directory and environment manipulation, command creation, etc. Attributes: * ``cwd`` - the remote working directory * ``env`` - the remote environment * ``custom_encoding`` - the remote machine's default encoding (assumed to be UTF8) * ``connect_timeout`` - the connection timeout There also is a _cwd attribute that exists if the cwd is not current (del if cwd is changed). cCst|ds t||_|jS)N_cwd)hasattrr rzrarrrrjs  zBaseRemoteMachine.cwdutf8 FcCs>||_||_|j|d|_||_t||_d|_i|_ dS)N new_session) r]connect_timeoutr"r) _get_unameunamerenv_python_program_cache)r0r_rrrrrr-s   zBaseRemoteMachine.__init__cCsP|jjddd\}}}|dkr|S|jjddd\}}}|dkr&|SdS)Nrretcoderz7python3 -c 'import platform;print(platform.uname()[0])'Windowsr)rstrip)r0rcout_rrrrs zBaseRemoteMachine._get_unamecCsd|jjd|dS)N)r6rTrarrrrdrezBaseRemoteMachine.__repr__cCs|Sr`rrarrr __enter__rwzBaseRemoteMachine.__enter__cCs |dSr`)rv)r0trBtbrrr__exit__s zBaseRemoteMachine.__exit__cCs|jt||_dS)z_closes the connection to the remote machine; all paths and programs will become defunctN)r)rvrrrarrrrvs zBaseRemoteMachine.closecGsRt|jg}|D]}t|trtd|||t|qt|g|RS)zA factory for :class:`RemotePaths `. Usage: ``p = rem.path("/usr", "lib", "python2.7")`` z!Cannot construct RemotePath from )rQrj isinstancer TypeErrorrmrNr )r0partsparts2prrrr.s  zBaseRemoteMachine.pathcCs||jddf}tt |j|WdS1swY|g}d|vr8||dd|ddg7}|D]!}|jjD]}||}|drZ| sZ||j|<|Sq@q:t ||jj)a2Looks up a program in the ``PATH``. If the program is not found, raises :class:`CommandNotFound ` :param progname: The program's name. Note that if underscores (``_``) are present in the name, and the exact name is not found, they will be replaced in turn by hyphens (``-``) then periods (``.``), and the name will be looked up again for each alternative :returns: A :class:`RemotePath ` rOrPNr-rgx) rget contextlibsuppressKeyErrorrreplacer.accessis_dirr)r0prognamer alternativesr:rfnrrrwhichs      zBaseRemoteMachine.whichcCstt|tr|j|ur|||Std|t|ts3||d|vs'd|vr-||S||Std|)arReturns a `Command` object representing the given program. ``cmd`` can be a string or a :class:`RemotePath `; if it is a path, a command representing this path will be returned; otherwise, the program name will be looked up in the system's ``PATH`` (using ``which``). Usage:: r_ls = rem["ls"] z3Given path does not belong to this remote machine: /\zcmd must not be a LocalPath: )rr rrZrr r.r)r0cmdrrr __getitem__s    zBaseRemoteMachine.__getitem__cCs|js|d|_|jS)z?A command that represents the default remote python interpreterpython3)rrarrrpythons zBaseRemoteMachine.pythonr~cCt)zCreates a new :class:`ShellSession ` object; this invokes the user's shell on the remote machine and executes commands on it over stdin/stdout/stderr NotImplementedError)r0isattyrrrrr"zBaseRemoteMachine.sessioncCr)a1Downloads a remote file/directory (``src``) to a local destination (``dst``). ``src`` must be a string or a :class:`RemotePath ` pointing to this remote machine, and ``dst`` must be a string or a :class:`LocalPath `rr0srcdstrrrdownloadzBaseRemoteMachine.downloadcCr)a/Uploads a local file/directory (``src``) to a remote destination (``dst``). ``src`` must be a string or a :class:`LocalPath `, and ``dst`` must be a string or a :class:`RemotePath ` pointing to this remote machinerrrrrupload#rzBaseRemoteMachine.uploadcKr)zSpawns the given command on the remote machine, returning a ``Popen``-like object; do not use this method directly, unless you need "low-level" control on the remote processrrGrrrrf*rzBaseRemoteMachine.popenc csp|d}|ddd}|d|D]!}|}tt|dt|d|dd|d d Vqd S) z Returns information about all running processes (on POSIX systems: using ``ps``) .. versionadded:: 1.3 psz-ez-ozpid,uid,stat,argsrrrr@N)r+r>rrrintrD)r0rr2rrrrrlist_processes0s  4z BaseRemoteMachine.list_processesccs0t|}|D] }||jr|Vq dS)z| Process grep: return information about all processes whose command-line args match the given regex pattern N)recompilersearchrH)r0patternpatprocinforrrpgrep=s   zBaseRemoteMachine.pgrepccsB|jd\}}}||}z |VW|dS|w)zlA context manager that creates a remote temporary directory, which is removed when the context exitsz1mktemp -d 2>/dev/null || mktemp -d tmp.XXXXXXXXXXN)r)rr.rdelete)r0rr local_dirrrrtempdirFs zBaseRemoteMachine.tempdircCs6|jdt|d}|d|d|S)Nzls -a rrgz..)r)rrr+remove)r0rfilesrrr _path_listdirVs  zBaseRemoteMachine._path_listdircCs\|dd}|dd}|jd|d|dd}t|dkr,||ds,gS|S)Nr@z\ z for fn in rz; do echo $fn; donerr)rr)rr+r _path_stat)r0rrmatchesrrr _path_glob\s  zBaseRemoteMachine._path_globcC4|jdvrdnd}|j|t|ddS)NDarwinFreeBSDzstat -c '%u,%U' zstat -f '%u,%Su' r,rr)rrrrr0rstat_cmdrrr _path_getuidg "zBaseRemoteMachine._path_getuidcCr)Nrzstat -c '%g,%G' zstat -f '%g,%Sg' rrrrrrr _path_getgidorzBaseRemoteMachine._path_getgidc Cs|jdvrd}nd}|jj|t|dd\}}}|dkrdS|d}|d}tt |ddgt dd |d dDR}||_ |S) Nrz+stat -c '%F,%f,%i,%d,%h,%u,%g,%s,%X,%Y,%Z' z-stat -f '%HT,%Xp,%i,%d,%l,%u,%g,%z,%a,%m,%c' rrrcss|]}t|VqdSr`)r)rsrrrrrsz/BaseRemoteMachine._path_stat..r) rr)rrrrr>lowerr rtuple text_mode) r0rrrrrstatresrresrrrrws 0zBaseRemoteMachine._path_statcCs|jdt|dS)Nzrm -rf r)rr)r0rrrr _path_deletezBaseRemoteMachine._path_deletecC$|jdt|dt|dS)Nzmv r@rrrrr _path_move$zBaseRemoteMachine._path_movecCr)Nzcp -r r@rrrrr _path_copyrzBaseRemoteMachine._path_copyNTcCs.|rdnd}d|t|}|j|dS)Nz-p rPzmkdir )rr)r)r0rmodeminus_pp_strrrrr _path_mkdirs zBaseRemoteMachine._path_mkdircCs"|jd|ddt|dS)Nzchmod or@r)r0rrrrr _path_chmods"zBaseRemoteMachine._path_chmodcCs|jd|dS)Nztouch )r)r)r0r.rrr _path_touchrezBaseRemoteMachine._path_touchcCsdg}|r |d|dur|dur||d|n|dur)|t|n |dur5|d||t||jd|dS)Nchownz-Rr'r@)rmrQrr)rrD)r0rownergroup recursiverHrrr _path_chowns zBaseRemoteMachine._path_chowncCs,|d|}|jrt|tr||jS|S)Ncat)r]rrQencode)r0rdatarrr _path_reads  zBaseRemoteMachine._path_readcCsr|jrt|tr||j}t}||||d||j |WddS1s2wYdS)Nr) r]rrQrrwriteflushseekrr:)r0rrfrrr _path_writes   "zBaseRemoteMachine._path_writecCs4|rdnd}|jd|t|dt|dS)Nz-s rPzln r@r)r0rrsymlink symlink_strrrr _path_links (zBaseRemoteMachine._path_linkcCs|jd|dS)Necho rrrLrrrrKrzBaseRemoteMachine.expandcCsBtdd|dDs|S|dd}|jd|dS)Ncss|]}|dVqdS)~N) startswith)rpartrrrrsz/BaseRemoteMachine.expanduser..r$z\$rr)anyrrr)rr)r0rM expr_replrrrrNs zBaseRemoteMachine.expanduser)r|r}F)F)NT)+rTrUrVrWrZrprjr-rrdrrrvr.rrrr"rrrfrrrcontextmanagerrrrrrrrrrrrrrrrrrKrNrrrrrysV             ry) __future__rrrtempfilerplumbum.commandsrrr plumbum.librplumbum.machines.baserplumbum.machines.envr plumbum.path.localr plumbum.path.remoter r r rrZ Exceptionrqrrryrrrrs      Y