U
|
¸ý°d ã @ sº d Z ddlmZ ddlmZ ddlmZ ddlZddlZddlZddl m
|
Z
|
ddl Z ddl Z ddl Z ddlZddlmZ ddlmZ dd lmZ dd
|
lmZ d d lmZ G d d dZdS )zsupport 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é )Úcompatc @ sö e Zd ZU ded< ded< ded< ded< ded < ded
|
< d d d dZd ddddZd ddddZd dddddZd ddddZd 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 )/Ú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,tt | _| jjdddd | jjddddd d S )
|
Nzpyproject.tomlz--stdoutÚ
|
store_truez)Write to stdout instead of saving to file)ÚactionÚhelpú-cz--checkzDon'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 úSd:\z\workplace\vscode\pyvenv\venv\Lib\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 )NZblackz --config r% r( r* r r r Ú run_blackG 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| ji| 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' Ú )r. Ú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) P s>
|
|
ÿ ÿþýûÿùøz#code_writer_cmd._run_console_script)Útextr$ c G s | j stj d |¡¡ d S )Nú )r r= r2 ÚwriteÚjoin)r rD r r r Ú write_statusw s z code_writer_cmd.write_status)rD Údestination_pathr$ c C sX | j jr| j||d n<| j jr*t| n*| d| d¡ t| |¡ | d¡ d S )N)ÚsourceúWriting ú...údone
|
)r ÚcheckÚ _run_diffr1 ÚprintrH r Ú
|
write_text)r rD rI r r r Úwrite_output_file_from_text{ s
|
z+code_writer_cmd.write_output_file_from_text)r# rI 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_filerK rL rM ) r rN rO r4 Úunlinkr1 ÚopenrP ÚreadrH ÚshutilÚmove)r r# rI Útfr r r Ú write_output_file_from_tempfile s
|
z/code_writer_cmd.write_output_file_from_tempfileN)rJ rS z Optional[str])rI rJ rS r$ c C s |r"t |}t|}W 5 Q R X n"|d k r8|jdd}n dsDtdt |@}tjt|||dddd}t|}|rd| _td |¡ W 5 Q R X d S )
|
NT)ÚkeependsFz!source or source_file is requiredz<proposed changes>é Ú
|
)ZfromfileZtofileÚnZlinetermr0 ) rU ÚlistÚ
|
splitlinesr ÚdifflibZ unified_diffr rP rG ) r rI rJ rS rY Z source_linesZdpÚdZ d_as_listr r r rO 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 rN r r r1 r= Úexitrc r r r Ú run_program· s zcode_writer_cmd.run_program)Ú__name__Ú
|
__module__Ú __qualname__Ú__annotations__r! r+ r, r) rH rR rZ rO Ú
|
contextlibÚcontextmanagerrd rg r r r r r " s(
|
' û r )Ú__doc__Ú
|
__future__r Úargparser r rl ra r4 Úpathlibr r8 rW r: r= Útypingr r r r r0 r r r r r r Ú<module> s"
|
