o 3g>@s`ddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddlm Z ddlm Z m Z ddl mZddlmZmZddlmZmZddlmZdd lmZmZmZdd lmZmZdd lmZdd l m!Z!dd l"m#Z#m$Z$ddl%m&Z&GdddeZ'erddl(m)Z)m*Z*e+dZ,GdddeZ-GdddeZ.GdddeZ/e/Z0dS)) annotationsN)contextmanager)PIPEPopen)mkdtemp)CommandNotFoundConcreteCommand)posix_daemonizewin32_daemonize iter_lines)IS_WIN32ProcInfoStaticProperty) BaseMachine PopenAddons)BaseEnv) ShellSession) LocalPath LocalWorkdir) RemotePathc@s8eZdZeZddZddZddZddZd d Zd S) PlumbumLocalPopencOst|i||_dSN)r_procselfargskwargsrN/home/garg/my-data/venv/lib/python3.10/site-packages/plumbum/machines/local.py__init__szPlumbumLocalPopen.__init__cC|Srr rrrr__iter__ szPlumbumLocalPopen.__iter__cCs |jSr)r __enter__r"rrrr$# zPlumbumLocalPopen.__enter__cOs|jj|i|Sr)r__exit__rrrrr&&szPlumbumLocalPopen.__exit__cCs t|j|Sr)getattrr)rnamerrr __getattr__)s zPlumbumLocalPopen.__getattr__N) __name__ __module__ __qualname__r r r#r$r&r)rrrrrs r)IMAGE_SUBSYSTEM_WINDOWS_CUIget_pe_subsystemz plumbum.localcs:eZdZdZdZe ZfddZddZddZ Z S) LocalEnvz>The local machine's environment; exposes a dict-like interfacercsLtjttjjtjdtr d|vr"|j dur$|j |d<dSdSdSdS)N)_currHOME) superr rospathpathsepenvironcopyr homer" __class__rrr <s zLocalEnv.__init__cCs:tj}|t_ztjtj|}W|t_|S|t_w)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)r3r6getdictr4 expanduser expandvarsrexprprevoutputrrrexpandBs zLocalEnv.expandcCs2tj}|t_z tj|}W|t_|S|t_w)zExpand home shortcuts (e.g., ``~/foo/bar`` or ``~john/foo/bar``) :param expr: An expression containing home shortcuts :returns: The expanded string)r3r6r;r4r<r>rrrr<Rs zLocalEnv.expanduser) r*r+r,__doc__ __slots__r CASE_SENSITIVEr rBr< __classcell__rrr9rr/6s r/c@s4eZdZdZdZd ddZeddZd d d ZdS) LocalCommandrautocCs&t|||dkrtjdS|dS)NrI)rr localcustom_encoding)r executableencodingrrrr hs zLocalCommand.__init__cCstSr)rJr"rrrmachinemszLocalCommand.machineNcKsRt|tr|f}|jj|j|d|f|dur|jn||dur"|jn|d|S)Nr)cwdenv) isinstancestrrN_popenrL formulaterOrP)rrrOrPrrrrpopenqs  zLocalCommand.popen)rI)rNN) r*r+r,rD QUOTE_LEVELr propertyrNrUrrrrrGds  rGc@s eZdZUdZeeZeZe Z e dZ iZded<ddZer;dgedd ejjZed d Zned d Zed dZddZddZddZeeedddfddZd*ddZ erjddZ!nddZ!d d!Z"d+d"d#Z#e$d$d%Z%e$d,d&d'Z&d(d)Z'e(e j)e Z*dS)- LocalMachineaThe *local machine* (a singleton object). It serves as an entry point to everything related to the local machine, such as working directory and environment manipulation, command creation, etc. Attributes: * ``cwd`` - the local working directory * ``env`` - the local environment * ``custom_encoding`` - the local machine's default encoding (``sys.getfilesystemencoding()``) rz dict[tuple[str, str], LocalPath]_program_cachecCs g|_dSr)_as_user_stackr"rrrr r%zLocalMachine.__init__PATHEXTz :.exe:.batcCsP|}|jjD]}|jD]}|||}|dr$|s$|Sq qdSNx)lowerrPr4 _EXTENSIONSaccessis_dir)clsprognamepextfnrrr_whichs    zLocalMachine._whichcCs4|jjD]}||}|dr|s|SqdSr])rPr4rarb)rcrdrergrrrrhs cCs||jddf}tt |j|WdS1swY|g}d|vr8||dd|ddg7}|D]}||}|rL||j|<|Sq:t|t |jj )a4Looks 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:`LocalPath ` PATHr[N_-.) rPget contextlibsuppressKeyErrorrYreplacerhrlistr4)rcrdkey alternativespnr4rrrwhichs    zLocalMachine.whichcGsTt|jg}|D]}t|trtd|||jt|qtt j j |S)zA factory for :class:`LocalPaths `. Usage: ``p = local.path("/usr", "lib", "python2.7")`` z Cannot construct LocalPath from ) rRrOrQr TypeErrorappendrPr<rr3r4join)rpartsparts2rerrrr4s  zLocalMachine.pathcCs$z||WdStyYdSw)NFT)rrcmdrrr __contains__s   zLocalMachine.__contains__cCs`t|tr t|St|ts)t|}d|vsd|vr"tt|St||St d|)aoReturns a `Command` object representing the given program. ``cmd`` can be a string or a :class:`LocalPath `; 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:: ls = local["ls"] /\zcmd must not be a RemotePath: ) rQrrGrr3fspathrJr4rvrwr|rrr __getitem__s   zLocalMachine.__getitem__NFc  Ks8|rd| d<trEd| vrE|tjdfvrEtt|} | tkrEt} | | d<ttdr9| j tj j O_ tj j | _ n | j tj O_ tj | _ |durL|j}|j|g} i}| D]} | sZqUt| trc| } || qU|jru|jd|\}}td|t|ft||||t||d| }t|_|j|_||_|S)NTstart_new_session startupinfo _subprocessz Running %r)rLstdinstdoutstderrrOrP)r sysrr.rRr- subprocess STARTUPINFOhasattrdwFlagsrSTARTF_USESHOWWINDOWSW_HIDE wShowWindowrOrPrQrr;updaterZloggerdebugrtime _start_timerKargv)rrLrrrrrOrP new_sessionr subsystemsuienvs_envprocrrrrSsX        zLocalMachine._popenrTcCs$tr t|||||St|||||S)a On POSIX systems: Run ``command`` as a UNIX daemon: fork a child process to setpid, redirect std handles to /dev/null, umask, close all fds, chdir to ``cwd``, then fork and exec ``command``. Returns a ``Popen`` process that can be used to poll/wait for the executed command (but keep in mind that you cannot access std handles) On Windows: Run ``command`` as a "Windows daemon": detach from controlling console and create a new process group. This means that the command will not receive console events and would survive its parent's termination. Returns a ``Popen`` object. .. note:: this does not run ``command`` as a system service, only detaches it from its parent. .. versionadded:: 1.3 )r r r )rcommandrOrrrxrrrdaemonic_popen9szLocalMachine.daemonic_popenc csddl}td}|ddd}|}||}zt|}Wn ty*tddw|d}|d }|d } |d } |D]} tt | || | | | | |VqAdS) z Returns information about all running processes (on Windows: using ``tasklist``) .. versionadded:: 1.3 rNtasklistz/Vz/FOCSVz"tasklist must at least have headerz Image NamePIDStatusz User Name) csvrJ splitlinesreadernext StopIteration RuntimeErrorindexrint) rrrrAlinesrowsheaderimgidxpididxstatidxuseridxrowrrrlist_processesRs(         zLocalMachine.list_processesc 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,argsrrH N)rpopstripsplitrrry)rrrlinerzrrrrms  *ccs0t|}|D] }||jr|Vq dS)z| Process grep: return information about all processes whose command-line args match the given regex pattern N)recompilersearchr)rpatternpatprocinforrrpgrep|s   zLocalMachine.pgrepcCst|dj|dS)zCreates a new :class:`ShellSession ` object; this invokes ``/bin/sh`` and executes commands on it over stdin/stdout/stderrsh)r)rrU)rrrrrsessionszLocalMachine.sessionccs.|t}z |VW|dS|w)zeA context manager that creates a temporary directory, which is removed when the context exitsN)r4rdelete)rnew_dirrrrtempdirs  zLocalMachine.tempdirc #strdur djfddndur$jfddn jfddz dVWjddSjdw)aNRun nested commands as the given user. For example:: head = local["head"] head("-n1", "/dev/sda1") # this will fail... with local.as_user(): head("-n1", "/dev/sda1") :param username: The user to run commands as. If not given, root (or Administrator) is assumed N Administratorcs4ddddddd|DdgdfS)Nrunasz /savecredz/user:"rcss|]}t|VqdSr)rR).0arrr sz9LocalMachine.as_user....)ryrvrrusernamerrsz&LocalMachine.as_user..csdgt|dfS)Nsudorrrvrr"rrrscsddgt|dfS)Nrz-urrrrrrsr)r rZrxrrrrras_users"     zLocalMachine.as_usercCr!)zUA shorthand for :func:`as_user("root") `)rr"rrras_rootszLocalMachine.as_root)rNNT)Fr)+r*r+r,rCrrrOr/rPrgetfilesystemencodingrKplatformunamerY__annotations__r r rmr_rr3r4r5r` classmethodrhrvr~rrrSrrrrrrrrrGrLpythonrrrrrXsT         B    ) rX)1 __future__rrnloggingr3rrrrrrrrtempfilerplumbum.commandsrrplumbum.commands.daemonsr r plumbum.commands.processesr plumbum.libr rrplumbum.machines.baserrplumbum.machines.envrplumbum.machines.sessionrplumbum.path.localrrplumbum.path.remoterrplumbum.machines._windowsr-r. getLoggerrr/rGrXrJrrrrs@        .F