o JAfN.@s~ ddlmZmZddlZddlZddlZddlZddlZddlZddl m Z ddZ Gddde j Z d d Zd d d ZdS))with_statementprint_functionN)refactorc Cs* |}|}tj||||ddddS)Nz (original)z (refactored))Zlineterm) splitlinesdifflibZ unified_diff)abfilenamer #/usr/lib/python3.10/lib2to3/main.py diff_textss rcs>eZdZ d fdd ZddZfddZdd ZZS) StdoutRefactoringToolrc sR ||_||_|r|tjs|tj7}||_||_||_tt | |||dSN) nobackups show_diffsendswithossep_input_base_dir _output_dir_append_suffixsuperr__init__) selfZfixersoptionsexplicitrrinput_base_dir output_dir append_suffix __class__r r r$s zStdoutRefactoringTool.__init__cOs0|j|||f|jj|g|Ri|dSr)errorsappendloggererror)rmsgargskwargsr r r log_errorAszStdoutRefactoringTool.log_errorc sV|}|jr$||jrtj|j|t|jd}n td||jf|jr,||j7}||krJtj |}tj |sC|rCt || d|||j s|d}tj|rnzt|Wntym| d|Ynwzt||Wnty| d||Ynwtt|j}||||||j st||||krt||dSdS)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 startswithrrpathjoinlen ValueErrorrdirnameisdirmakedirs log_messagerlexistsremoveOSErrorrenamerr write_fileshutilcopymode) rZnew_textr Zold_textencodingZ orig_filenamerZbackupwriter!r r r8EsN          z StdoutRefactoringTool.write_filecCs|r |d|dS|d||jrct|||}z8|jdurF|j|D]}t|q%tjWdn 1s;wYWdSWdS|D]}t|qHWdStybt d|fYdSwdS)NzNo changes to %sz Refactored %sz+couldn't encode %s's diff for your terminal) r3rrZ output_lockprintsysstdoutflushUnicodeEncodeErrorwarn)roldnewr ZequalZ diff_linesliner r r print_outputls,     (  z"StdoutRefactoringTool.print_output)rrr)__name__ __module__ __qualname__rr*r8rF __classcell__r r r!r rs  'rcCstd|ftjddS)Nz WARNING: %sfile)r=r>stderr)r'r r r rBsrBc 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}||\}}|jrd7|d8<|jstd9d7|_|jr|js| d:|j r|js| d;|js|j rtd<|js|jr| d=|j rt d>tD]}t |q|sd?S|st d@tjdAt dBtjdAdCSdD|vrd7}|jrt dEtjdAdCS|jrd7|dF<|jr d7|dG<|jr'tjntj}tjdH|dItdJ}tt} tfdKdL|jD} t} |jrvd*} |jD]} | dMkr`d7} qU| dN| qU| rs| | n| }n| | }| | }t!j"#|}|r|$t!j%st!j"&|st!j"'|}|jr|(t!j%}|)dO|j|t*t+||t+| |j|j ||j|j dP}|j,s|r|-n!z |||j|j.|j/Wntj0yt dQtjdAYdSw|1t2t3|j,S)RNz2to3 [options] file|dir ...)usagez-dz--doctests_only store_truezFix up doctests only)actionhelpz-fz--fixr$z1Each FIX specifies a transformation; default: all)rPdefaultrQz-jz --processesstorerintzRun 2to3 concurrently)rPrRtyperQz-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-dirstrrzXPut output files in this directory instead of overwriting the input files. Requires -n.)rPrUrRrQz-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.rKzUse --help to show usage.-zCan't write to stdin.r exec_functionz%(name)s: %(message)s)formatlevelz lib2to3.mainc3s|] }d|VqdS).fix_Nr ).0fix fixer_pkgr r szmain..allr]z7Output in %r will mirror the input directory %r layout.)rrr z+Sorry, -j isn't supported on this platform.)4optparseZ OptionParserZ add_option parse_argsrWr<rBrrr&Z add_suffixZno_diffsZ list_fixesr=rZget_all_fix_namesr>rMrrZverboseloggingDEBUGINFO basicConfig getLoggersetZget_fixers_from_packageZnofixr_addunion differencerr, commonprefixrrr1r0rstripinforsortedr#refactor_stdinZ doctests_only processesZMultiprocessingUnsupportedZ summarizerTbool)rar(parserrtflagsrZfixnamer\r%Z avail_fixesZunwanted_fixesrZ all_presentr_Z requestedZ fixer_namesrZrtr r`r mains                                 ryr)Z __future__rrr>rrrgr9rdrrrZMultiprocessRefactoringToolrrBryr r r r s  g