관리-도구
편집 파일: tool_support.cpython-38.pyc
U -?�f� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlZddlZddlZddl m Z ddl Z ddl Z ddl Z ddlZddlmZ ddlmZ dd lmZ dd lmZ dd lmZ d d lmZ G dd� d�ZdS )z�support routines for the helpers in tools/. These aren't imported by the enclosing util package as the are not needed for normal library use. � )� annotations)�ArgumentParser)� NamespaceN)�Path)�Any)�Dict)�Iterator)�Optional)�Union� )�compatc @ s� e Zd ZU ded< ded< ded< ded< ded < ded < d d �d d�Zd dd�dd�Zd dd�dd�Zd ddd�dd�Zd dd�dd�Zd d dd �d d �Z d d dd �d!d"�Z d#d#d$�d d%d%dd&�d'd(�Z e j d)d*�d+d,��Ze j d-d*�d.d/��Zd#S )0�code_writer_cmdr �parserr �args�bool�suppress_output�diffs_detectedr � source_root�pyproject_toml_path�str)� tool_scriptc C s^ t |�jj| _| jt d� | _| j�� s,t�t� | _| jjdddd� | jjddddd � d S ) Nzpyproject.tomlz--stdout� store_truez)Write to stdout instead of saving to file)�action�help�-cz--checkz�Don't write the files back, just return the status. Return code 0 means nothing would change. Return code 1 means some files would be reformatted)r r ) r �parentr r �exists�AssertionErrorr r � add_argument)�selfr � r �L/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/util/tool_support.py�__init__+ s ��zcode_writer_cmd.__init__�None)�tempfile�returnc C s" | � t|�dd| j� �d�� d S )NZzimportsz--toml-config �� entrypoint�options��_run_console_scriptr r �r r$ r r r! � run_zimports? s ��z code_writer_cmd.run_zimportsc C s" | � t|�dd| j� �d�� d S )N�blackz --config r& r) r+ r r r! � run_blackH s ��zcode_writer_cmd.run_blackzDict[str, Any])�pathr( r% c C s� t jdk}|d }t�d�D ]}|j|kr |} qBq td|� ���|�dd�}tj||d�|g }i } | jr~t j | d< | d < t j t j d d |j|j|jf g| fd t| j�i| �� d S )zeRun a Python console application from within the process. Used for black, zimports �posixr' �console_scriptsz*Could not find entrypoint console_scripts.r( � )r0 �stdout�stderrr zimport %s; %s.%s()�cwdN)�os�namer Zimportlib_metadata_get� Exception�get�shlex�splitr � subprocess�DEVNULL�run�sys� executable�module�attrr r ) r r/ r( Zis_posixZentrypoint_name�entry�implZcmdline_options_strZcmdline_options_list�kwr r r! r* Q s> � �������z#code_writer_cmd._run_console_script)�textr% c G s | j stj�d�|�� d S )N� )r r? r4 �write�join)r rF r r r! � write_statusx s z code_writer_cmd.write_statuszUnion[str, Path])rF �destination_pathr% c C s^ | j jr| j||d� nB| j jr*t|� n0| �d|� d�� t|�j|ddd� | �d� d S )N)�source�Writing �...�utf-8� )�encoding�newline�done )r �check� _run_diffr3 �printrJ r � write_text)r rF rK r r r! �write_output_file_from_text| s �z+code_writer_cmd.write_output_file_from_text)r$ rK r% c C s� | j jr"| j||d� t�|� n\| j jrVt|��}t|�� � W 5 Q R X t�|� n(| � d|� d�� t � ||� | � d� d S )N)� source_filerM rN rS ) r rT rU r6 �unlinkr3 �openrV �readrJ �shutil�move)r r$ rK �tfr r r! � write_output_file_from_tempfile� s z/code_writer_cmd.write_output_file_from_tempfileN)rL rY z Optional[str])rK rL rY r% c C s� |r&t |dd��}t|�}W 5 Q R X n"|d k r<|jdd�}n dsHtd��t |dd��H}tjt|�|t|��� ddd d �}t|�}|r�d| _t d � |�� W 5 Q R X d S ) NrO )rQ T)�keependsFz!source or source_file is requiredz<proposed changes>� rP )�fromfile�tofile�nZlinetermr2 ) r[ �list� splitlinesr �difflibZ unified_diffr �as_posixr rV rI ) r rK rL rY r_ Z source_linesZdp�dZ d_as_listr r r! rU � s& �zcode_writer_cmd._run_diffzIterator[ArgumentParser])r% c c s | j V d S )N)r �r r r r! � add_arguments� s z code_writer_cmd.add_argumentszIterator[None]c c sf | j �� | _| jjr"d| _d| _n| jjr2d| _nd| _d V | jjrX| jrXt�d� n t�d� d S )NFTr r ) r � parse_argsr rT r r r3 r? �exitrk r r r! � run_program� s zcode_writer_cmd.run_program)�__name__� __module__� __qualname__�__annotations__r" r, r. r* rJ rX r` rU � contextlib�contextmanagerrl ro r r r r! r # s( '� r )�__doc__� __future__r �argparser r rt rh r6 �pathlibr r: r] r< r? �typingr r r r r r2 r r r r r r! �<module> s$