CasperSecurity
o
��bcN. � @ 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)� )�lineterm)�
splitlines�difflib� unified_diff)�a�b�filename� r �#/usr/lib/python3.10/lib2to3/main.py�
diff_texts s �r c s@ e Zd ZdZ d � 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__) �self�fixers�options�explicitr r �input_base_dir�
output_dir�
append_suffix�� __class__r r r $ s
z StdoutRefactoringTool.__init__c O s0 | j �|||f� | jj|g|�R i |�� d S �N)�errors�append�logger�error)r �msg�args�kwargsr r r � log_errorA s z StdoutRefactoringTool.log_errorc sV |}| j r$|�| j�rtj�| j |t| j�d � �}n td|| jf ��| jr,|| j7 }||krJtj� |�}tj�
|�sC|rCt� |� | � d||� | j
s�|d }tj�|�rnzt�|� W n tym | � d|� Y nw zt�||� W n ty� | � d||� Y nw tt| �j}|||||� | j
s�t�||� ||kr�t�||� d S 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�shutil�copymode) r �new_textr
�old_text�encoding�
orig_filenamer! �backup�writer# r r r; E sN
���
� � � �z StdoutRefactoringTool.write_filec C s� |r
| � d|� d S | � d|� | jrct|||�}z8| jd urF| j� |D ]}t|� q%tj�� W d � n 1 s;w Y W d S W d S |D ]}t|� qHW d S tyb t d|f � Y d S w d S )NzNo changes to %sz
Refactored %sz+couldn't encode %s's diff for your terminal)
r6 r r � output_lock�print�sys�stdout�flush�UnicodeEncodeError�warn)r �old�newr
�equal�
diff_lines�liner r r � print_outputl s,
(�
� ���z"StdoutRefactoringTool.print_output)r r r ) �__name__�
__module__� __qualname__�__doc__r r- r; rP �
__classcell__r r r# r r s
� 'r c C s t d| f tjd� d S )Nz WARNING: %s��file)rE rF �stderr)r* r r r rJ � s rJ c
s t jdd�}|jddddd� |jdd d
g d d � |jd
dddddd� |jddd
g dd � |jddddd� |jdddd d� |jd d dd d� |jd d!dd"d� |jd#dd$d� |jd%d&dd'd� |jd(d)dd*d+d � |jd,d-dd.d/d0d1� |jd2d3dd4d� |jd5dd.d/d6d1� d*}i }|�|�\}}|jr�d7|d8< |js�td9� d7|_|jr�|js�|� d:� |j
r�|js�|� d;� |js�|j r�td<� |js�|jr�|� d=� |j r�t
d>� t�� �D ]}t
|� q�|s�d?S |s�t
d@tjdA� t
dBtjdA� dCS dD|v �rd7}|j�rt
dEtjdA� dCS |j�rd7|dF< |j�r d7|dG< |j�r&tjntj}tjdH|dI� t�dJ�}tt�� ��} t� fdKdL�|j D ��}
t� } |j �rud*} |j D ]}
|
dMk�r_d7} �qT| � � dN |
� �qT| �rr| � | �n| }n| � | �}|� |
�}t!j"�#|�}|�r�|�$t!j%��s�t!j"�&|��s�t!j"�'|�}|j�r�|�(t!j%�}|�)dO|j|� t*t+|�|t+| �|j|j ||j|j
dP�}|j,�s�|�r�|�-� n)z
|�||j|j.|j/� W n tj0�y� |j/dk�s�J �t
dQtjdA� Y dS w |�1� t2t3|j,��S )Rz�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 ...)�usagez-dz--doctests_only�
store_truezFix up doctests only)�action�helpz-fz--fixr' z1Each FIX specifies a transformation; default: all)r[ �defaultr\ z-jz --processes�storer �intzRun 2to3 concurrently)r[ r] �typer\ 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-ez--exec-functionz/Modify the grammar so that exec() 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.)r[ r` r] r\ 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.rV zUse --help to show usage.� �-zCan't write to stdin.r �
exec_functionz%(name)s: %(message)s)�format�levelz lib2to3.mainc 3 s � | ] }� d | V qdS )�.fix_Nr )�.0�fix�� fixer_pkgr r � <genexpr>� s � zmain.<locals>.<genexpr>�allrh z7Output in %r will mirror the input directory %r layout.)r r! r" z+Sorry, -j isn't supported on this platform.)4�optparse� OptionParser�
add_option�
parse_argsrb rC rJ r! r r) �
add_suffix�no_diffs�
list_fixesrE r �get_all_fix_namesrF rX r re �verbose�logging�DEBUG�INFO� basicConfig� getLogger�set�get_fixers_from_package�nofixrj �add�union�
differencer r/ � commonprefixr r r4 r3 �rstrip�infor �sortedr& �refactor_stdin�
doctests_only� processes�MultiprocessingUnsupported� summarizer_ �bool)rl r+ �parserr� �flagsr �fixnamerg r( � avail_fixes�unwanted_fixesr � all_presentrj � requested� fixer_namesr �rtr rk r �main� s�
� � � �
�
�
�
��
� � �
� �
� �
�
���r� r% )rT �
__future__r r rF r r rx r< ro r r r �MultiprocessRefactoringToolr rJ r� r r r r �<module> s g