`N`* dZddlZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZdd lmZdd lmZddlmZddlmZ ddZ d dZdZd!dZdZdZdZdZdZdZdZd"dZdS)#acEntry points for YAPF. The main APIs that YAPF exposes to drive the reformatting. FormatFile(): reformat a file. FormatCode(): reformat a string of code. These APIs have some common arguments: style_config: (string) Either a style name or a path to a file that contains formatting style settings. If None is specified, use the default style as set in style.DEFAULT_STYLE_FACTORY 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. It can be used by third-party code (e.g., IDEs) when reformatting a snippet of code rather than a whole file. 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. N)parse)blank_line_calculator)comment_splicer)continuation_splicer)file_resources)identify_container) py3compat)pytree_unwrapper) pytree_utils) reformatter) split_penalty)style)subtype_assignerFct|r|rtdt||\}}} t||||||\} } | drJ| dd}|d|D|z} |r$|r|| krtj|| | |d| | fS| | | fS)aFormat a single Python file and return the formatted code. Arguments: filename: (unicode) The file to reformat. style_config: (string) Either a style name or a path to a file that contains formatting style settings. If None is specified, use the default style as set in style.DEFAULT_STYLE_FACTORY 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. It can be used by third-party code (e.g., IDEs) when reformatting a snippet of code rather than a whole file. 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. in_place: (bool) If True, write the reformatted code back to the file. logger: (io streamer) A stream to output logging. Returns: Tuple of (reformatted_code, encoding, changed). reformatted_code is None if the file is successfully written to (having used in_place). reformatted_code is a diff if print_diff is True. Raises: IOError: raised if there was an error reading the file. ValueError: raised if in_place and print_diff are both specified. z)Cannot pass both in_place and print_diff.) style_configfilenamelines print_diffverify c3K|]}|VdSN.0lines 5lib/python3.11/site-packages/yapf/yapflib/yapf_api.py zFormatFile..gs"%=%=td%=%=%=%=%=%=N) _CheckPythonVersion ValueErrorReadFile FormatCoderstripsplitjoinrWriteReformattedCode) rrrrrin_placeloggeroriginal_sourcenewlineencodingreformatted_sourcechangeds r FormatFiler/7s'B B*B @ A AA'/&'A'A$/7H *   !!!gt$$H  % %d + + 1 1$ 7 7E %=%=u%=%=%===G #>?.@@@)(4F*2H>>> 7 "" Xw ..r clttjtj||ds|dz } t j|}n*#tj$r}|dz|j z|_ d}~wwxYwtj |tj |tj|t!j|t%j|t)j|t-j|}|D]} | t3|}t5||t7jt;|||} || kr|rdn| dfSt=|| |} |r| | dkfS| dfS)a)Format a string of Python code. This provides an alternative entry point to YAPF. Arguments: unformatted_source: (unicode) The code to format. filename: (unicode) The name of the file being reformatted. style_config: (string) Either a style name or a path to a file that contains formatting style settings. If None is specified, use the default style as set in style.DEFAULT_STYLE_FACTORY 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. It can be used by third-party code (e.g., IDEs) when reformatting a snippet of code rather than a whole file. 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. Returns: Tuple of (reformatted_source, changed). reformatted_source conforms to the desired formatting style. changed is True if the source changed. rz: NF)rT) r rSetGlobalStyleCreateStyleFromConfigendswithr ParseCodeToTreer ParseErrormsgrSpliceCommentsrSpliceContinuationsrAssignSubtypesrIdentifyContainersr ComputeSplitPenaltiesrCalculateBlankLinesr UnwrapPyTreeCalculateFormattingInformation_LineRangesToSet_MarkLinesToFormatr Reformat_SplitSemicolons_GetUnifiedDiffstrip) unformatted_sourcerrrrrtreeeuwlinesuwlr- code_diffs rr#r#qs8u2<@@AAA  $ $T * *$  '(: ; ;DD     tOae #AE   &&&*4000!$''''---%d++++D111  )$ / /' ))c&&(((( 5 ! !%We$$$"+w00--- 322!3U ::,xAAA). ioo''2- -- T !!sA%%B 4BB cd}tjddkr'tjddkrt|dStjddkr%tjddkrt|dSdS)Nz-yapf is only supported for Python 2.7 or 3.4+r)sys version_info RuntimeError)errmsgs rr r s :&aA Q    a Q    rc tj|}tj|d|d5}|}dddn #1swxYwYtj|}dd|Ddz}|||fS#t$r}|r ||d}~wt$r}|r|d|||d}~wwxYw)aRead the contents of the file. An optional logger can be specified to emit messages to your favorite logging stream. If specified, then no exception is raised. This is external so that it can be used by third-party applications. Arguments: filename: (unicode) The name of the file. logger: (function) A function or lambda that takes a string and emits it. Returns: The contents of filename. Raises: IOError: raised if there was an error reading the file. rr2)moder,r+Nrc3@K|]}|dVdS)z N)r$rs rrzReadFile..s.==tt{{6**======rz@Could not parse %s! Consider excluding this file with --exclude.) r FileEncodingr open_with_encoding readlines LineEndingr&IOErrorUnicodeDecodeError)rr)r,fdr line_endingsourceerrs rr"r"s]" *844H  %sXr ; ; ;>@llnne!+E22K YY==u=== = = DF ; ((      fSkkk      f O  fSkkk  sF,BA BABA=B CB-- C:CCcbg}|D])}||*|Sr)extendSplit)rJresuwlines rrDrDs8 #fJJv||~~ *rz^#.*\byapf:\s*disable\bz^#.*\byapf:\s*enable\bc|dSt}t|D]+\}}|t||dz,|S)z#Return a set of lines in the range.NrO)setsortedupdaterange) line_rangesline_setlowhighs rrArAs[ 4 UU(+&&**ic4 OOE#tax(()))) /rc|r@|D]=}|t|j|jjdz |_>d}|t |kr}||}|jr t|jj r|dz }|t |kr||}|jrt|jj rmtj t|jj dd tjsnd|_|dz }|t |knHtj t|jj tjrd|_|dz }|t |k{dSdS)z1Skip sections of code that we shouldn't reformat.rOrrTN) intersectionrnlinenolastdisablelen is_comment _DisableYAPFfirstvaluerF _EnableYAPFresearchDISABLE_PATTERNr% IGNORECASE)rJrriindexs rrBrBs 888 --  v{1A5 6 6888fnn % G  U^F   fl(..00 1 1   c'll""5>&   ;v|/A/G/G/I/I#J#J 9_#\/5577==dCCBGMMOO],, &. 1*%c'll"" ?FK$5$;$;$=$=r} M Mfn QJE! G rcBtjt|ddtjpOtjt|ddtjSNrrrt)rrrr%rFrrs rr{r{s| )OJJt$$Q'--// @ @ A )OJJt$$R(..00"- A ABrcBtjt|ddtjpOtjt|ddtjSr)rrENABLE_PATTERNr%rFrrs rr~r~#s| )NJJt$$Q'--// @ @ A )NJJt$$R(..00"- A ABrcodec |}|}dtj||||ddddzS)zGet a unified diff of the changes. Arguments: before: (unicode) The original source code. after: (unicode) The reformatted source code. filename: (unicode) The code's filename. Returns: The unified diff text. rz (original)z (reformatted)r2)lineterm) splitlinesr&difflib unified_diff)beforeafterrs rrErE*sp     &     %           r)NNFFFN)r0NNFFr)r)__doc__rrrS lib2to3.pgen2r yapf.yapflibrrrrrr r r r r rrr/r#r r"rDrrrArBr{r~rErrrrs* ......((((((------''''''++++++""""""))))))%%%%%%$$$$$$&&&&&&))))))! 7/7/7/7/v$  A"A"A"A"H!!!% % % % P   -*   :BBBBBBr