File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/util/__pycache__/tool_support.cpython-310.pyc
o
���g� � @ 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
< d1d
d�Zd2dd�Zd2dd�Zd3dd�Zd4d d �Zd5d d!�Z d6d"d#�Z
d$d$d%�d7d)d*�Z e j
d8d,d-��Ze j
d9d/d0��Zd$S ):�code_writer_cmdr �parserr �args�bool�suppress_output�diffs_detectedr � source_root�pyproject_toml_path� tool_script�strc C s^ t |�jj| _| jt d� | _| j�� sJ �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 �existsr r � add_argument)�selfr � r �G/usr/local/lib/python3.10/dist-packages/sqlalchemy/util/tool_support.py�__init__+ s �
�zcode_writer_cmd.__init__�tempfile�return�Nonec C �" | � t|�dd| j� �d�� d S )N�zimportsz--toml-config ��
entrypoint�options��_run_console_scriptr r �r r" r r r � run_zimports? �
��z code_writer_cmd.run_zimportsc C r% )N�blackz --config r' r* r, r r r � run_blackH r. zcode_writer_cmd.run_black�pathr) �Dict[str, Any]c
C s� t jdk}|d }t�d�D ] }|j|kr|} nqtd|� ���|�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) � )r3 �stdout�stderrr zimport %s; %s.%s()�cwdN)�os�namer �importlib_metadata_get� Exception�get�shlex�splitr �
subprocess�DEVNULL�run�sys�
executable�module�attrr r )
r r1 r) �is_posix�entrypoint_name�entry�impl�cmdline_options_str�cmdline_options_list�kwr r r r+ Q s>
�� ������
�z#code_writer_cmd._run_console_script�textc G s | j stj�d�|�� d S d S )N� )r rC r7 �write�join)r rN r r r � write_statusx s �z code_writer_cmd.write_status�destination_path�Union[str, Path]c C sb | j jr
| j||d� d S | j jrt|� d S | �d|� d�� t|�j|ddd� | �d� d S )N)�source�Writing �...�utf-8�
)�encoding�newline�done
)r �check� _run_diffr6 �printrR r �
write_text)r rN rS r r r �write_output_file_from_text| s �z+code_writer_cmd.write_output_file_from_textc C s� | j jr| j||d� t�|� d S | j jr7t|��}t|�� � W d � n1 s+w Y t�|� d S | � d|� d�� t
� ||� | � d� d S )N)� source_filerV rW r\ ) r r] r^ r9 �unlinkr6 �openr_ �readrR �shutil�move)r r" rS �tfr r r � write_output_file_from_tempfile� s
� z/code_writer_cmd.write_output_file_from_tempfileN)rU rb rU �
Optional[str]rb c C s� |r t |dd�� }t|�}W d � n1 sw Y n|d ur(|jdd�}nJ d��t |dd��2}tjt|�|t|��� ddd d
�}t|�}|r\d| _td � |�� W d � d S W d � d S 1 sgw Y d S ) NrX )rZ T)�keependsFz!source or source_file is requiredz<proposed changes>� rY )�fromfile�tofile�n�linetermr5 )
rd �list�
splitlines�difflib� unified_diffr �as_posixr r_ rQ ) r rS rU rb rh � source_lines�dp�d� d_as_listr r r r^ � s0
��
��
"�zcode_writer_cmd._run_diff�Iterator[ArgumentParser]c c s � | j V d S )N)r �r r r r �
add_arguments� s � z code_writer_cmd.add_arguments�Iterator[None]c c sj � | j �� | _| jjrd| _d| _n | jjrd| _nd| _d V | jjr.| jr.t�d� d S t�d� d S )NFTr r ) r �
parse_argsr r] r r r6 rC �exitr{ r r r � run_program� s � zcode_writer_cmd.run_program)r r )r" r r# r$ )r1 r r) r2 r# r$ )rN r r# r$ )rN r rS rT r# r$ )r" r rS r r# r$ )rS rT rU rj rb rj r# r$ )r# rz )r# r} )�__name__�
__module__� __qualname__�__annotations__r! r- r0 r+ rR ra ri r^ �
contextlib�contextmanagerr| r� r r r r r
# s*
'
� r
)�__doc__�
__future__r �argparser r r� rs r9 �pathlibr r> rf r@ rC �typingr r r r r
r5 r r
r r r r �<module> s&