U
e5d�- � @ s� d Z ddlmZmZ ddlZddlZddlZddlZddlZddl Z ddl
m Z dd� Z G dd � d e j
�Zd
d � Zdd d
�ZdS )z
Main program for 2to3.
� )�with_statement�print_functionN� )�refactorc C s( | � � } |� � }tj| |||dddd�S )z%Return a unified diff of two strings.z
(original)z (refactored)� )Zlineterm)�
splitlines�difflibZ unified_diff)�a�b�filename� r �$/usr/lib64/python3.8/lib2to3/main.py�
diff_texts s �r c s> e Zd ZdZd � fdd� Zdd� Z� fdd�Zd d
� Z� ZS ) �StdoutRefactoringToola2
A refactoring tool that can avoid overwriting its input files.
Prints output to stdout.
Output files can optionally be written to a different directory and or
have an extra file suffix appended to their name for use in situations
where you do not want to replace the input files.
r c sP || _ || _|r&|�tj�s&|tj7 }|| _|| _|| _tt | ��
|||� dS )aF
Args:
fixers: A list of fixers to import.
options: A dict with RefactoringTool configuration.
explicit: A list of fixers to run even if they are explicit.
nobackups: If true no backup '.bak' files will be created for those
files that are being refactored.
show_diffs: Should diffs of the refactoring be printed to stdout?
input_base_dir: The base directory for all input files. This class
will strip this path prefix off of filenames before substituting
it with output_dir. Only meaningful if output_dir is supplied.
All files processed by refactor() must start with this path.
output_dir: If supplied, all converted files will be written into
this directory tree instead of input_base_dir.
append_suffix: If supplied, all files output by this tool will have
this appended to their filename. Useful for changing .py to
.py3 for example by passing append_suffix='3'.
N) � nobackups�
show_diffs�endswith�os�sep�_input_base_dir� _output_dir�_append_suffix�superr �__init__) �selfZfixers�options�explicitr r �input_base_dir�
output_dir�
append_suffix�� __class__r r
r $ s
z StdoutRefactoringTool.__init__c O s* | j �|||f� | jj|f|�|� d S )N)�errors�append�logger�error)r �msg�args�kwargsr r r
� log_errorA s z StdoutRefactoringTool.log_errorc
sz |}| j rH|�| j�r6tj�| j |t| j�d � �}ntd|| jf ��| jrX|| j7 }||kr�tj� |�}tj�
|�s�|r�t� |� | � d||� | j
�s2|d }tj�|�r�zt�|� W n. tk
r� } z| � d|� W 5 d }~X Y nX zt�||� W n2 tk
�r0 } z| � d||� W 5 d }~X Y nX tt| �j} | ||||� | j
�s`t�||� ||k�rvt�||� d S )Nz5filename %s does not start with the input_base_dir %szWriting converted %s to %s.z.bakzCan't remove backup %szCan't rename %s to %s)r �
startswithr r �path�join�len�
ValueErrorr �dirname�isdir�makedirs� log_messager �lexists�remove�OSError�renamer r �
write_file�shutilZcopymode)
r Znew_textr Zold_text�encodingZ
orig_filenamer Zbackup�err�writer r r
r7 E sJ
� ��
�
z StdoutRefactoringTool.write_filec C s� |r| � d|� n�| � d|� | jr�t|||�}zP| jd k rl| j�" |D ] }t|� qHtj�� W 5 Q R X n|D ] }t|� qpW n$ tk
r� t d|f � Y d S X d S )NzNo changes to %sz
Refactored %sz+couldn't encode %s's diff for your terminal)
r2 r r Z output_lock�print�sys�stdout�flush�UnicodeEncodeError�warn)r �old�newr ZequalZ
diff_lines�liner r r
� print_outputl s$
�z"StdoutRefactoringTool.print_output)r r r ) �__name__�
__module__� __qualname__�__doc__r r) r7 rE �
__classcell__r r r r
r s
� 'r c C s t d| f tjd� d S )Nz WARNING: %s��file)r< r= �stderr)r& r r r
rA � s rA c
s t jdd�}|jddddd� |jdd d
g d d � |jd
dddddd� |jddd
g dd � |jddddd� |jdddd d� |jd d dd d� |jd dd!d� |jd"d#dd$d� |jd%d&dd'd(d � |jd)d*dd+d,d-d.� |jd/d0dd1d� |jd2dd+d,d3d.� d'}i }|�|�\}}|j�r@d4|d5< |j�s:td6� d4|_|j�rZ|j�sZ|� d7� |j
�rt|j�st|� d8� |j�s�|j �r�td9� |j�s�|j�r�|� d:� |j �r�t
d;� t�� �D ]}t
|� �q�|�s�d� t
d?tjd>� d@S dA|k�r(d4}|j�r(t
dBtjd>� d@S |j�r8d4|dC< |j�rFtjntj}tjdD|dE� t�dF�}tt�� ��} t� fdGdH�|jD ��}
t� } |j �r�d'} |j D ](}
|
dIk�r�d4} n| � � dJ |
� �q�| �r�| � | �n| }n
| � | �}|� |
�}t j!�"|�}|�r0|�#t j$��s0t j!�%|��s0t j!�&|�}|j�rT|�'t j$�}|�(dK|j|� t)t*|�|t*| �|j|j ||j|j
dL�}|j+�s�|�r�|�,� nTz|�||j|j-|j.� W n8 tj/k
�r� |j.dk�s�t0�t
dMtjd>� Y dS X |�1� t2t3|j+��S )Nz�Main program.
Args:
fixer_pkg: the name of a package where the fixers are located.
args: optional; a list of command line arguments. If omitted,
sys.argv[1:] is used.
Returns a suggested exit status (0, 1, 2).
z2to3 [options] file|dir ...)Zusagez-dz--doctests_only�
store_truezFix up doctests only)�action�helpz-fz--fixr# z1Each FIX specifies a transformation; default: all)rO �defaultrP z-jz --processesZstorer �intzRun 2to3 concurrently)rO rQ �typerP z-xz--nofixz'Prevent a transformation from being runz-lz --list-fixesz List available transformationsz-pz--print-functionz0Modify the grammar so that print() is a functionz-vz --verbosezMore verbose loggingz
--no-diffsz#Don't show diffs of the refactoringz-wz--writezWrite back modified filesz-nz --nobackupsFz&Don't write backups for modified filesz-oz --output-dir�strr zXPut output files in this directory instead of overwriting the input files. Requires -n.)rO rS rQ rP z-Wz--write-unchanged-fileszYAlso write files even if no changes were required (useful with --output-dir); implies -w.z --add-suffixzuAppend this string to all output filenames. Requires -n if non-empty. ex: --add-suffix='3' will generate .py3 files.T�write_unchanged_filesz&--write-unchanged-files/-W implies -w.z%Can't use --output-dir/-o without -n.z"Can't use --add-suffix without -n.z@not writing files and not printing diffs; that's not very usefulzCan't use -n without -wz2Available transformations for the -f/--fix option:r z1At least one file or directory argument required.rK zUse --help to show usage.� �-zCan't write to stdin.r z%(name)s: %(message)s)�format�levelz lib2to3.mainc 3 s | ]}� d | V qdS )�.fix_Nr )�.0�fix�� fixer_pkgr r
� � s zmain..�allrZ z7Output in %r will mirror the input directory %r layout.)r r r z+Sorry, -j isn't supported on this platform.)4�optparseZ OptionParserZ
add_option�
parse_argsrU r; rA r r r% Z
add_suffixZno_diffsZ
list_fixesr< r Zget_all_fix_namesr= rM r �verbose�logging�DEBUG�INFOZ basicConfigZ getLogger�setZget_fixers_from_packageZnofixr\ �add�union�
differencer r+ � commonprefixr r r0 r/ �rstrip�infor �sortedr" �refactor_stdinZ
doctests_onlyZ processesZMultiprocessingUnsupported�AssertionErrorZ summarizerR �bool)r^ r' �parserro �flagsr ZfixnamerY r$ Z avail_fixesZunwanted_fixesr Z all_presentr\ Z requestedZ fixer_namesr Zrtr r] r
�main� s�
� � � �
�
�
��
� � �
� �
� � �
��rt )N)rI Z
__future__r r r= r r rd r8 ra r r r ZMultiprocessRefactoringToolr rA rt r r r r
� s g