관리-도구
편집 파일: xargs.cpython-38.pyc
U ,�e� � @ sr d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl m Z d dl m Z d dl m Z d dl mZ d dl mZ d dl mZ d d l mZ d d lmZ d d lmZ d d lmZ ed �Zed�Zdd�dd�Zd3ddd�dd�Zdd�dd�Zddd�dd �Z G d d � d e �Z d4d d dd d!d"�d#d$�Z ej dd%d&�d'd(��Z!d)d*e� d+�d,d d-ddd.d/d0�d1d2�Z"dS )5� )� annotationsN)�Any)�Callable)� Generator)�Iterable)�MutableMapping)�Sequence)�TypeVar)� parse_shebang)� cmd_output_b)� cmd_output_p�TArg�TRet�int)�returnc C sL zt t�d��W S tk r$ Y nX z t�� W S tk rF Y dS X d S )Nr � )�len�os�sched_getaffinity�AttributeError�multiprocessing� cpu_count�NotImplementedError� r r �>/opt/hc_python/lib/python3.8/site-packages/pre_commit/xargs.pyr s r z MutableMapping[str, str] | None)�_envr c C sT | d k r | n t tdtj�}dt|� }|�� D ] \}}|t|�t|� d 7 }q.|S )N�environb� � )�getattrr �environr �items)r r �size�k�vr r r � _environ_size) s r% c C sH t jdkr2t �d�d t� } tt| d�d�} | S t jdkr@dS dS d S )N�posix� SC_ARG_MAXi i i �nti x )r �name�sysconfr% �max�min)�maximumr r r �_get_platform_max_length1 s r. �str)�cmdr c G s< d� | �}tjdkr&t|�d��d S t|�t�� ��S d S )N� �win32zutf-16ler )�join�sys�platformr �encode�getfilesystemencoding)r0 Zfull_cmdr r r �_command_length= s r8 c @ s e Zd ZdS )�ArgumentTooLongErrorN)�__name__� __module__� __qualname__r r r r r9 H s r9 z Sequence[str]z int | Noneztuple[tuple[str, ...], ...])r0 �varargs�target_concurrency� _max_lengthr c C s� |pt � }tdt�t|�| ��}t| �} g }g }tt|��}t| � d }|r�|� � }t|�d } || |kr�t|�|k r�|� |� || 7 }qJ|s�t |��qJ|� | t|� � g }t| � d }|� |� qJ|� | t|� � t|�S )N� r ) r. r+ �math�ceilr �tuple�list�reversedr8 �pop�appendr9 ) r0 r= r> r? Zmax_args�retZret_cmd� total_length�argZ arg_lengthr r r � partitionL s0 � � rK zYGenerator[Callable[[Callable[[TArg], TRet], Iterable[TArg]], Iterable[TRet]], None, None])�maxsizer c c s4 | dkrt V n tj�| ��}|j V W 5 Q R X d S )Nr )�map� concurrent�futures�ThreadPoolExecutor)rL �exr r r �_thread_mapperx s rR Fr )�colorr> r? �tuple[str, ...]�boolr ztuple[int, bytes])r0 r= rS r> r? �kwargsr c s |rt nt� d}d}zt�| �} W n8 tjk rZ } z|�� dd� W Y �S d}~X Y nX tjdkr�| d �� � d�r�t� d�} dt | � t d � d }t | |||�} d d d �� �fdd� } t t | �|�} t| ��>} | | | �}|D ](\}}}t|�t|�k�r|}||7 }q�W 5 Q R X ||fS )z�A simplified implementation of xargs. color: Make a pty if on a platform that supports it target_concurrency: Target number of partitions to run concurrently r � Nr r2 )z.batz.cmdzcmd.exei z /c i rT z tuple[int, bytes, bytes | None])�run_cmdr c s � | dt jd����S )NF)�check�stderr)� subprocess�STDOUT)rX �Zcmd_fnrV r r �run_cmd_partition� s ��z xargs.<locals>.run_cmd_partition)r r r Z normalize_cmdZExecutableNotFoundErrorZ to_outputr4 r5 �lower�endswith�find_executabler rK r, rR �abs)r0 r= rS r> r? rV �retcode�stdout�eZcmd_exeZ partitionsr^ �threadsZ thread_map�resultsZ proc_retcodeZproc_out�_r r] r �xargs� s. &�� ri )N)N)#� __future__r �concurrent.futuresrN � contextlibrA r r r[ r4 �typingr r r r r r r Z pre_commitr Zpre_commit.utilr r r r r r% r. r8 � RuntimeErrorr9 rK �contextmanagerrR ri r r r r �<module> s@ �,�