f _N.c@sdZddlmZmZddlZddlZddlZddlZddlZddl Z ddl m Z ddZ Gdd d e j Zd d Zdd d ZdS)z Main program for 2to3. )with_statementprint_functionN)refactorc Cs(|}|}tj||||ddddS)z%Return a unified diff of two strings.z (original)z (refactored))Zlineterm)Z splitlinesdifflibZ unified_diff)ZaZbfilenamer%/usr/lib64/python3.10/lib2to3/main.py diff_textss  r cs>eZdZdZd fdd ZddZfddZd d ZZS) 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. rc sP||_||_|r&|tjs&|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_diffsendswithossep_input_base_dir _output_dir_append_suffixsuperr __init__) selfZfixersoptionsexplicitr r input_base_dir output_dir append_suffixZ __class__rr r$s zStdoutRefactoringTool.__init__cOs0|j|||f|jj|g|Ri|dSN)errorsappendloggererror)rmsgargsZkwargsrrr log_errorAszStdoutRefactoringTool.log_errorc sZ|}|jrH||jr6tj|j|t|jd}ntd||jf|jrX||j7}||krtj |}tj |s|rt || d|||j s|d}tj|rzt|Wnty| d|Yn0zt||Wn"ty| d||Yn0tt|j}||||||j s@t||||krVt||dS)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)rZ startswithrrpathZjoinZlenZ ValueErrorrdirnameisdirZmakedirs log_messager ZlexistsZremoveZOSErrorZrenamerr write_fileshutilZcopymode) rZnew_textrZold_textZencodingZ orig_filenamerZbackupwriterrr r)EsH          z StdoutRefactoringTool.write_filecCs|r|d|n|d||jrt|||}zd|jdur|j,|D] }t|qHtjWdn$1st0Yn|D] }t|qWn"tyt d|fYdS0dS)NzNo changes to %sz Refactored %sz+couldn't encode %s's diff for your terminal) r(r r Z output_lockprintsysZstdoutZflushZUnicodeEncodeErrorwarn)rZoldZnewrZequalZ diff_linesZlinerrr print_outputls$    * z"StdoutRefactoringTool.print_output)rrr) Z__name__Z __module__Z __qualname____doc__rr$r)r/Z __classcell__rrrr r s   'r cCstd|ftjddS)Nz WARNING: %sZfile)r,r-stderr)r"rrr r.sr.c s"tjdd}|jddddd|jdd d gd d |jd dddddd|jddd gdd |jddddd|jddddd|jddddd|jd d!dd"d|jd#dd$d|jd%d&dd'd|jd(d)dd*d+d |jd,d-dd.d/d0d1|jd2d3dd4d|jd5dd.d/d6d1d*}i}||\}}|jrRd7|d8<|jsLtd9d7|_|jrl|jsl| d:|j r|js| d;|js|j rtd<|js|jr| d=|j rt d>tD]}t |q|sd?S|st d@tjdAt dBtjdAdCSdD|vr:d7}|jr:t dEtjdAdCS|jrJd7|dF<|jrZd7|dG<|jrhtjntj}tjdH|dItdJ}tt} tfdKdL|jD} t} |jrd*} |jD]*} | dMkrd7} q| dN| q| r| | n| }n | | }| | }t!j"#|}|rT|$t!j%sTt!j"&|sTt!j"'|}|jrx|(t!j%}|)dO|j|t*t+||t+| |j|j ||j|j dP}|j,s|r|-nRz|||j|j.|j/Wn6tj0y |j/dksJt dQtjdAYdS0|1t2t3|j,S)RzMain 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_onlyZ store_truezFix up doctests only)actionhelpz-fz--fixrz1Each FIX specifies a transformation; default: all)r3defaultr4z-jz --processesZstorerintzRun 2to3 concurrently)r3r5typer4z-xz--nofixz'Prevent a transformation from being runz-lz --list-fixeszList 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-dirZstrrzXPut output files in this directory instead of overwriting the input files. Requires -n.)r3r7r5r4z-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:rz1At least one file or directory argument required.r1zUse --help to show usage.iz-zCan't write to stdin.r exec_functionz%(name)s: %(message)s)Zformatlevelz lib2to3.mainc3s|]}d|VqdS).fix_Nr)Z.0fix fixer_pkgrr Z szmain..Zallr;z7Output in %r will mirror the input directory %r layout.)rrrz+Sorry, -j isn't supported on this platform.)4optparseZ OptionParserZ add_optionZ parse_argsr8r+r.rr r!Z add_suffixZno_diffsZ list_fixesr,rZget_all_fix_namesr-r2rr9ZverboseloggingZDEBUGZINFOZ basicConfigZ getLoggerZsetZget_fixers_from_packageZnofixr<ZaddZunionZ differencerr%Z commonprefixrrr'r&ZrstripZinfor Zsortedrrefactor_stdinZ doctests_onlyZ processesZMultiprocessingUnsupportedZ summarizer6Zbool)r>r#ZparserrAZflagsrZfixnamer:r Z avail_fixesZunwanted_fixesrZ all_presentr<Z requestedZ fixer_namesrZrtrr=r mains                              rBr)r0Z __future__rrr-rrr@r*r?rrr ZMultiprocessRefactoringToolr r.rBrrrr Zs  g