관리-도구
편집 파일: rust.cpython-38.pyc
U ,�e^ � @ st d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl m Z d dl m Z d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d d lmZ d d lmZ d d lmZ d d lm Z d d lm Z dZ ej Z ej!Z"ej#dd�dd�dd��Z$ddd�dd�Z%dddd�dd�Z&ej'd dd d �d d ��Z(d d!d"d#�d$d%�Z)ddd"d&�d'd(�Z*d dd)d"d*�d+d,�Z+dS )-� )� annotationsN)� Generator)�Sequence)� lang_base)� parse_shebang)� envcontext)�PatchesT)�Var)�Prefix)� cmd_output_b)�make_executable)�win_exeZrustenv� )�maxsize�str)�returnc C s$ t dddd�d dkrdS tjS d S )N�cargo� --versionF)�checkr �system)r �C�DEFAULT� r r �G/opt/hc_python/lib/python3.8/site-packages/pre_commit/languages/rust.py�get_default_version s r )�language_versionr c C s | t jkrdS | S dS )z=Transform the language version into a rust toolchain version.ZstableN)r r )r r r r �_rust_toolchain+ s r r )� target_dir�versionr c C s: dt j�| d�t jtd�fff|dkr4dt|�ffnd�S )N�PATH�binr ZRUSTUP_TOOLCHAINr )�os�path�join�pathsepr r )r r r r r � get_env_patch3 s ���r% r z Generator[(None, None, None)])�prefixr r c c s2 t �| t|�}tt||��� d V W 5 Q R X d S )N)r �environment_dir�ENVIRONMENT_DIRr r% )r& r �envdirr r r �in_env? s r* zset[str]�None)r& �additional_dependenciesr c C sN g }|D ]0}|� d�\}}}|� d|p(d� �}|�|� qt�| d|�� d S )N�:�@�*r �add)r r0 )� partition�appendr Z setup_cmd)r& r, Zcrates�dep�name�_�specZcrater r r �_add_dependenciesF s r7 )� toolchainr) r c C s� t �� ��}td|fd|ff��� t�d�d kr�tjdkr>d}nd}tj� |�}t j � |t d��}t|d��}t�||� W 5 Q R X t|� t|d d d d d � tdddd| � W 5 Q R X W 5 Q R X d S )NZ CARGO_HOME� RUSTUP_HOMEZrustup�win32z https://win.rustup.rs/x86_64zhttps://sh.rustup.rsz rustup-init�wbz-yz--quietz--no-modify-pathz--default-toolchain�noner8 �installz--no-self-update)�tempfile�TemporaryDirectoryr r �find_executable�sys�platform�urllib�request�urlopenr! r" r# r �open�shutil� copyfileobjr r )r8 r) Z rustup_dir�url�respZ rustup_init�fr r r �install_rust_with_toolchainS s4 � �rL z Sequence[str])r& r r, r c C s t �| t|�}dd� |D �}t|�| }dh}|D ]J}|td�d � }|�d�\}} } | dkrp|�|d| f� q2|�|f� q2t�� ��} | � t | |�� |dkr�t t |�|� | � t �� �} | � td | ff�� t|�d kr�t| |� |D ]"} td d d d|f| �d| ji� q�W 5 Q R X d S )Nc S s h | ]}|� d �r|�qS )�cli:)� startswith)�.0r3 r r r � <setcomp>� s z&install_environment.<locals>.<setcomp>)z--path�.rM r- � r r r9 r r r= z--binsz--root�cwd)r r' r( �set�lenr1 r0 � contextlib� ExitStack� enter_contextr* rL r r> r? r r7 r Z prefix_dir)r& r r, r) Zcli_depsZlib_depsZpackages_to_installZcli_dep�packager5 Z crate_version�ctxZtmpdir�argsr r r �install_environmentq s@ � ���r\ ),� __future__r rV � functools�os.pathr! rG rA r> �urllib.requestrC �typingr r Zpre_commit.constants� constantsr Z pre_commitr r Zpre_commit.envcontextr r r Zpre_commit.prefixr Zpre_commit.utilr r r r( Zbasic_health_checkZ health_checkZbasic_run_hookZrun_hook� lru_cacher r r% �contextmanagerr* r7 rL r\ r r r r �<module> s>