3
\- @ 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_statementprint_functionN )refactorc C s( | j } |j }tj| |||ddddS )z%Return a unified diff of two strings.z
(original)z (refactored) )Zlineterm)
splitlinesdifflibZ unified_diff)abfilename r $/usr/lib64/python3.6/lib2to3/main.py
diff_texts s
r c s> e Zd ZdZd fdd Zdd Z fddZd 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 sR || _ || _|r(|jtj r(|tj7 }|| _|| _|| _tt | j
||| 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_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 j|||f | jj|f|| d S )N)errorsappendloggererror)r msgargskwargsr r r
log_errorA s z StdoutRefactoringTool.log_errorc
! s| |}| j rH|j| jr6tjj| j |t| jd }ntd|| jf | jrX|| j7 }||krtjj |}tjj
| r|rtj | | j d|| | j
s4|d }tjj|rytj| W n. tk
r } z| j d| W Y d d }~X nX ytj|| W n2 tk
r2 } z| j d|| W Y d d }~X nX tt| j} | |||| | j
sbtj|| ||krxtj|| 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 Zbackuperrwrite)r r r
r6 E s@
z StdoutRefactoringTool.write_filec C s |r| j d| n| j d| | jrt|||}yX| jd k rp| j& x|D ] }t| qJW tjj W d Q R X nx|D ] }t| qvW W n" tk
r t d|f d S X d S )NzNo changes to %sz
Refactored %sz+couldn't encode %s's diff for your terminal)
r1 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____doc__r r( r6 rD
__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
r@ s r@ 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|\}}|jr@d4|d5< |js:td6 d4|_|jr\|j r\|j d7 |j
rx|j rx|j d8 |j r|j rtd9 |j r|jr|j d: |j rt
d; x tj D ]}t
| qW |sd t
d?tjd> d@S dA|kr4d4}|jr4t
dBtjd> d@S |jrDd4|dC< |jrRtjntj}tjdD|dE tjdF}ttj } t fdGdH|jD }
t } |j rd'} x2|j D ](}
|
dIkrd4} n| j dJ |
qW | r| j | n| }n
| j | }|j |
}t j!j"|}|rD|j#t j$ rDt j!j%| rDt j!j&|}|jrh|j't j$}|j(dK|j| t)t*||t*| |j|j ||j|j
dL}|j+s|r|j, nBy|j||j|j-|j. W n& tj/k
r t
dMtjd> dS X |j0 t1t2|j+S )NzMain 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)actionhelpz-fz--fixr" z1Each FIX specifies a transformation; default: all)rM defaultrN z-jz --processesZstorer intzRun 2to3 concurrently)rM rO typerN 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.)rM rQ rO rN 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.)rJ zUse --help to show usage. -zCan't write to stdin.r z%(name)s: %(message)s)formatlevelz lib2to3.mainc 3 s | ]} d | V qdS )z.fix_Nr ).0fix) fixer_pkgr r
s zmain..allz.fix_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_argsrS r: r@ r r r$ Z
add_suffixZno_diffsZ
list_fixesr; r Zget_all_fix_namesr< rK r verboseloggingDEBUGINFOZ basicConfigZ getLoggersetZget_fixers_from_packageZnofixrY addunion
differencer r* commonprefixr r r/ r. rstripinfor sortedr! refactor_stdinZ
doctests_onlyZ processesZMultiprocessingUnsupportedZ summarizerP bool)rZ r& parserrk flagsr ZfixnamerW r# Z avail_fixesZunwanted_fixesr Z all_presentrY Z requestedZ fixer_namesr Zrtr )rZ r
main s
ro )N)rH Z
__future__r r r< r r r` r7 r] r r r ZMultiprocessRefactoringToolr r@ ro r r r r
s g