o i @sddlZddlZddlZddlZddlmZmZmZmZddl m Z ddl m Z de dee fdd Zdee de fd d Zdd d dee ee fdeededejfddZdS)N)AnyListOptionalUnion) is_windows)ErrorscommandreturncCstj|t dS)zSplit a string command using shlex. Handles platform compatibility. command (str) : The command to split RETURNS (List[str]): The split command. )posix)shlexsplitrr rH/home/ubuntu/.local/lib/python3.10/site-packages/weasel/util/commands.py split_command srcCsddd|DS)zJoin a command using shlex. shlex.join is only available for Python 3.8+, so we're using a workaround here. command (List[str]): The command to join. RETURNS (str): The joined command  css|]}t|VqdS)N)r quote).0cmdrrr szjoin_command..)joinrrrr join_commandsrF)stdincapturerrc Cst|tr t|}|}n|}d|}ztj|tj|dd|r#tj nd|r)tj ndd}Wnt yBt t j j||dddw|jdkrs|rsd|d }|d |j7}|jdurf|d 7}||j7}t|}||_||_||jdkr~t|j|S) aRun a command on the command line as a subprocess. If the subprocess returns a non-zero exit code, a system exit is performed. command (str / List[str]): The command. If provided as a string, the string will be split using shlex.split. stdin (Optional[Any]): stdin to read from or None. capture (bool): Whether to capture the output and errors. If False, the stdout and stderr will not be redirected, and if there's an error, sys.exit will be called with the return code. You should use capture=False when you want to turn over execution to the command, and capture=True when you want to run the command more like a function. RETURNS (Optional[CompletedProcess]): The process object. rutf8FN)envinputencodingcheckstdoutstderrr) str_commandtoolzError running command: z zSubprocess exited with status z$ Process log (stdout and stderr): ) isinstancestrrr subprocessrunosenvironcopyPIPESTDOUTFileNotFoundErrorrE501format returncoder SubprocessErrorretr sysexit)r rrcmd_listcmd_strr2messageerrorrrr run_commandsF           r9)r(r r&r3typingrrrrcompatrerrorsrr%rrboolCompletedProcessr9rrrrs(