
    i                     <   d Z ddlZddlZddlZddlmZ ddlmZ ddlm	Z	 ddl
mZ ddlmZ dd	lmZ  ee      Zed
ee	ddf   fd       Z	 	 ddeee   z  deez  dz  d
ej*                  fdZe	 ddeee   z  deez  dz  d
eeee   ee   f   ddf   fd       Zy)zFContains utilities to easily handle subprocesses in `huggingface_hub`.    N)	Generator)contextmanager)StringIO)Path)IO   )
get_loggerreturnc               #      K   t               } t        j                  }| t        _        	 |  |t        _        y# |t        _        w xY ww)a  Capture 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)r   sysstdout)outputprevious_outputs     r/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/huggingface_hub/utils/_subprocess.pycapture_outputr      s7      ZFjjOCJ%$
_
s   &A	9 A	AA	commandfolderc           	          t        | t              r| j                         } t        |t              rt        |      }t	        j
                  | fd|dd|xs t        j                         d|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.
    Tutf-8replace)r   checkencodingerrorscwd)
isinstancestrsplitr   
subprocessrunosgetcwd)r   r   r   kwargss       r   run_subprocessr#   5   si    4 '3--/&$V>>!biik      c           
   +     K   t        | t              r| j                         } t        j                  | ft        j
                  t        j
                  t        j                  dd|xs t        j                         d|5 }|j                  J d       |j                  J d       |j                  |j                  f ddd       y# 1 sw Y   yxY ww)a  Run 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")
    ```
    r   r   )stdinr   stderrr   r   r   Nz'subprocess is opened as subprocess.PIPE)r   r   r   r   PopenPIPESTDOUTr    r!   r&   r   )r   r   r"   processs       r   run_interactive_subprocessr,   `   s     @ '3--/				
oo  !biik	
 	
 , 
}}(S*SS(~~)T+TT)mmW^^++, , ,s   A>C AC
	C
CC)NT)N)__doc__r    r   r   collections.abcr   
contextlibr   ior   pathlibr   typingr   loggingr	   __name__loggerr   r   listCompletedProcessr#   tupler,    r$   r   <module>r:      s    M 	  
 % %     
H	 %	(D$"67 % %. !%
(49_($J(
   (V  !%.,49_.,$J., uRWbg%&d23	., .,r$   