a `N`0@sdZddlmZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl m Z ddl m Z dd l mZd Zd d Zd dZdddZdddZddZddZddZedkredS)aYAPF. YAPF uses the algorithm in clang-format to figure out the "best" formatting for Python code. It looks at the program as a series of "unwrappable lines" --- i.e., lines which, if there were no column limit, we would place all tokens on that line. It then uses a priority queue to figure out what the best formatting is --- i.e., the formatting with the least penalty. It differs from tools like autopep8 and pep8ify in that it doesn't just look for violations of the style guide, but looks at the module as a whole, making formatting decisions based on what's the best format for each line. If no filenames are specified, YAPF reads the code from stdin. )print_functionN)tokenize)errors)file_resources) py3compat)style)yapf_apiz0.31.0c Cs8t}||dd}|jr0tdtdS|j}|jrHt|dS|j rft |j dkrf| d|j durzt |j nd}|j s|js|jr| dg}tjjrqz|tWqtyYqYqtyYdS0q|dur|jstt}dd|D}t|d|d<z.tjt d !|d d |||j"d \}}WnJt#j$y} z.t%&d | j'dd| j'dfWYd} ~ n d} ~ 00t(d |dSt)t} t*|j |j+|j,pg| } | st%&dt-| ||j|j|j|j|j"|j.|j/|j0d } | r4|js0|j/r4dSdS)a[Main program. Arguments: argv: command-line arguments, such as sys.argv (including the program name in argv[0]). Returns: Zero on successful program termination, non-zero otherwise. With --diff: zero if there were no changes, non-zero otherwise. Raises: YapfError: if none of the supplied files were Python files. Nzyapf {}rz-cannot use -l/--lines with more than one filez=cannot use --in-place or --diff flags when reading from stdincSsg|] }|qS)rstrip).0liner r ,lib/python3.9/site-packages/yapf/__init__.py dzmain.. z)filename style_configlinesverifyz%s:%szz.Input filenames did not match any python files)rno_local_stylein_place print_diffrparallelquietverbose)1 _BuildParser parse_argsversionprintformat __version__rZ style_help _PrintHelprlenfileserror _GetLinesrdiffsysstdinclosedappendrZ raw_inputEOFErrorKeyboardInterruptrrGetDefaultStyleForDirosgetcwdZ removeBOMrZ FormatCodeZunicodejoinrr TokenErrorr YapfErrorargsWriteReformattedCodeZGetExcludePatternsForDirZGetCommandLineFiles recursiveZexclude FormatFilesrrr) argvparserr4rrZoriginal_sourcesourceZreformatted_source_eZ!exclude_patterns_from_ignore_filer$changedr r rmain.s~      8     r>cCs|jdur |js tt|_tt|jtdt t D]v\}}| D]}td|rjdpld|ddqZt |}t|tst|trdtt|}t|d|ddtqJdS) zPrints the help menu.Nz[style]# )sepz, =)rrrr.r/r0ZSetGlobalStyleZCreateStyleFromConfigrsortedZHelpitems splitlinesZGet isinstancesetlistr1mapstrlower)r4optionZ docstringr Z option_valuer r rr"s  r"Fc  sd} |rddl} ddl} t| t|} | j| Pf dd|D}| j|D]}| |O} qfWdq1s0Yn&|D] }| t | O} q| S)aFFormat a list of files. Arguments: filenames: (list of unicode) A list of files to reformat. lines: (list of tuples of integers) A list of tuples of lines, [start, end], that we want to format. The lines are 1-based indexed. This argument overrides the 'args.lines'. It can be used by third-party code (e.g., IDEs) when reformatting a snippet of code. style_config: (string) Style name or file path. no_local_style: (string) If style_config is None don't search for directory-local style configuration. in_place: (bool) Modify the files in place. print_diff: (bool) Instead of returning the reformatted source, return a diff that turns the formatted source into reformatter source. verify: (bool) True if reformatted code should be verified for syntax. parallel: (bool) True if should format multiple files in parallel. quiet: (bool) True if should output nothing. verbose: (bool) True if should print out filenames while processing. Returns: True if the source code changed in any of the files being formatted. FrNcs(g|] }t| qSr )Zsubmit _FormatFile)r r executorrrrrrrrrr rrs   zFormatFiles..) multiprocessingconcurrent.futuresmin cpu_countr#ZfuturesProcessPoolExecutorZ as_completedresultrN) filenamesrrrrrrrrrr=rQZ concurrentworkersZfuture_formatsZfuturerr rOrr7s .  r7c Cs|r|std||dur2|s2ttj|}z$tj||||||tj d\} } } Wntt j y} z0t d|| jdd| jdfWYd} ~ n4d} ~ 0ty} z|| _WYd} ~ n d} ~ 00|s|s| rt|| | || S)zFormat an individual file.zReformatting %sN)rrrrrloggerz%s:%s:%sr r)rrr.r/pathdirnamerZ FormatFileloggingZwarningrr2rr3r4 SyntaxErrorrr5) rrrrrrrrrZreformatted_codeencodingZ has_changer<r r rrNs4    8  rNcCsng}|D]`}ttt|dd}|ddkrr8rr3stderrwriterK)r<r r rrun_mainhs ru__main__)NFFFFFFF)NFFFFFF)__doc__Z __future__rrmr\r/r(Z lib2to3.pgen2rZ yapf.yapflibrrrrrr!r>r"r7rNr&rru__name__r r r rsH       ^ 6 &X