U
e5d- @ s| d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlZddl m
Z
dd Z G dd de
j Z
d d
Zd
d d ZdS ) )with_statementprint_functionN )refactorc C s( | } | }tj| |||ddddS )Nz
(original)z (refactored) )Zlineterm)
splitlinesdifflibZ unified_diff)abfilename r $/usr/lib64/python3.8/lib2to3/main.py
diff_texts s r c s: e Zd Zd
fdd Zdd Z fddZdd Z ZS ) StdoutRefactoringToolr c sP || _ || _|r&|tjs&|tj7 }|| _|| _|| _tt |
||| d S N) nobackups
show_diffsendswithossep_input_base_dir _output_dir_append_suffixsuperr __init__) selfZfixersoptionsexplicitr 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 r )errorsappendloggererror)r msgargskwargsr r r
log_errorA s z StdoutRefactoringTool.log_errorc
sz |}| j rH|| jr6tj| j |t| jd }ntd|| jf | jrX|| j7 }||krtj |}tj
|s|rt | | d|| | j
s2|d }tj|rzt| 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|| ||krvt|| 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 pathjoinlen
ValueErrorr dirnameisdirmakedirs log_messager lexistsremoveOSErrorrenamer r
write_fileshutilZcopymode)
r Znew_textr Zold_textencodingZ
orig_filenamer Zbackuperrwriter! r r
r8 E sJ
z StdoutRefactoringTool.write_filec C s |r| d| n| d| | jrt|||}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)
r3 r r Z output_lockprintsysstdoutflushUnicodeEncodeErrorwarn)r oldnewr ZequalZ
diff_linesliner r r
print_outputl s$
z"StdoutRefactoringTool.print_output)r r r )__name__
__module__ __qualname__r r* r8 rF
__classcell__r r r! r
r s 'r c C s t d| f tjd d S )Nz WARNING: %sfile)r= r> stderr)r' r r r
rB s rB 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 }||\}}|jr@d4|d5< |js:td6 d4|_|jrZ|jsZ| d7 |j
rt|jst| d8 |js|j rtd9 |js|jr| d: |j rt
d; t D ]}t
| q|sd t
d?tjd> d@S dA|kr(d4}|jr(t
dBtjd> d@S |jr8d4|dC< |jrFtjntj}tjdD|dE tdF}tt } t fdGdH|jD }
t } |j rd'} |j D ](}
|
dIkrd4} n| dJ |
q| r| | n| }n
| | }| |
}t j!"|}|r0|#t j$s0t j!%|s0t j!&|}|jrT|'t j$}|(dK|j| t)t*||t*| |j|j ||j|j
dL}|j+s|r|, nDz|||j|j-|j. W n( tj/k
r t
dMtjd> Y dS X |0 t1t2|j+S )NNz2to3 [options] file|dir ...)Zusagez-dz--doctests_only
store_truezFix up doctests only)actionhelpz-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-dirstrr 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.Twrite_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)formatlevelz lib2to3.mainc 3 s | ]} d | V qdS ).fix_Nr ).0fix 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.)3optparseZ OptionParserZ
add_option
parse_argsrU r< rB r r r& Z
add_suffixZno_diffsZ
list_fixesr= r Zget_all_fix_namesr> rM r verboseloggingDEBUGINFOZ basicConfigZ getLoggersetZget_fixers_from_packageZnofixr\ addunion
differencer r, commonprefixr r r1 r0 rstripinfor sortedr# refactor_stdinZ
doctests_onlyZ processesZMultiprocessingUnsupportedZ summarizerR bool)r^ r( parserro flagsr ZfixnamerY r% Z avail_fixesZunwanted_fixesr Z all_presentr\ Z requestedZ fixer_namesr Zrtr r] r
main s
rs )N)Z
__future__r r r> r r rd r9 ra r r r ZMultiprocessRefactoringToolr rB rs r r r r
s g