-:j <dZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZeeZed ee ddffd Z dd eeezd ee zdzd ej*fdZe dd eeezd ee zdzd eee ee efddffdZy)zFContains utilities to easily handle subprocesses in `huggingface_hub`.N) Generator)contextmanager)StringIO)Path)IO) get_loggerreturnc#Kt}tj}|t_ ||t_y#|t_wxYww)aCapture output that is printed to terminal. Taken from https://stackoverflow.com/a/34738440 Example: ```py >>> with capture_output() as output: ... print("hello world") >>> assert output.getvalue() == "hello world " ``` N)rsysstdout)outputprevious_outputs c/var/www/html/tokenscope/api/venv/lib/python3.12/site-packages/huggingface_hub/utils/_subprocess.pycapture_outputrs7ZFjjOCJ% $ _ s&A 9 A  AA commandfolderc t|tr|j}t|tr t|}t j |fd|dd|xst jd|S)aX Method to run subprocesses. Calling this will capture the `stderr` and `stdout`, please call `subprocess.run` manually in case you would like for them not to be captured. Args: command (`str` or `list[str]`): The command to execute as a string or list of strings. folder (`str`, *optional*): The folder in which to run the command. Defaults to current working directory (from `os.getcwd()`). check (`bool`, *optional*, defaults to `True`): Setting `check` to `True` will raise a `subprocess.CalledProcessError` when the subprocess has a non-zero exit code. kwargs (`dict[str]`): Keyword arguments to be passed to the `subprocess.run` underlying command. Returns: `subprocess.CompletedProcess`: The completed process. Tutf-8replace)rcheckencodingerrorscwd) isinstancestrsplitr subprocessrunosgetcwd)rrrkwargss rrun_subprocessr#5si4'3--/&$V >>   !biik    c +Kt|tr|j}tj|ftj tj tj dd|xstjd|5}|jJd|jJd|j|jfdddy#1swYyxYww)aRun a subprocess in an interactive mode in a context manager. Args: command (`str` or `list[str]`): The command to execute as a string or list of strings. folder (`str`, *optional*): The folder in which to run the command. Defaults to current working directory (from `os.getcwd()`). kwargs (`dict[str]`): Keyword arguments to be passed to the `subprocess.run` underlying command. Returns: `tuple[IO[str], IO[str]]`: A tuple with `stdin` and `stdout` to interact with the process (input and output are utf-8 encoded). Example: ```python with _interactive_subprocess("git credential-store get") as (stdin, stdout): # Write to stdin stdin.write("url=hf.co username=obama ".encode("utf-8")) stdin.flush() # Read from stdout output = stdout.read().decode("utf-8") ``` rr)stdinr stderrrrrNz'subprocess is opened as subprocess.PIPE) rrrrPopenPIPESTDOUTr r!r&r )rrr"processs rrun_interactive_subprocessr,`s@'3--/    oo    !biik    , }}(S*SS(~~)T+TT)mmW^^++ , , ,sA>CAC  C CC)NT)N)__doc__r rr collections.abcr contextlibriorpathlibrtypingrloggingr __name__loggerrrlistCompletedProcessr#tupler,r$rr:sM  %% H % (D$"67%%.!% ( 49_( $J (   (V!%., 49_., $J .,uRWbg%&d23 .,.,r$