a GN^#$@sUddlZddlZddlmZmZddlmZddlmZddlm Z ddl m Z m Z m Z ddlZddlZddlZddlmZmZddlZddlmZddlZddlZddlZddlZddlZddlZddlZdd lm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0dd l1m2Z2dd l3m4Z4m5Z5m6Z6ddl7Z7ddl8Z8dd l9m:Z:m;Z;dd lm?Z?m@Z@mAZAddlBmCZCmDZDddlEmFZFmGZGddlHmIZIddlJmKZKddlLmMZNdZOdZPdZQee2deNdZReSZTeSZUeSZVeWZXeWZYeWZZeWZ[eWZ\e/e@e?fZ]e!de"dge'dfZ^e_Z`eWZae-e`eafZbe#eebfZce e7jddddZee e7jddddZfeCgeReCjhZiGdddejZkGd d!d!elZmGd"d#d#enZoGd$d%d%e ZpGd&d'd'e ZqGd(d)d)e ZrerjserjterjuhZvGd*dde Zwerjxewjyherjzewj{ewjyherj|ewj{ewjyherj}ewj{ewj~ewjyherjsewj{ewjewjewj~ewjewjyherjtewj{ewjewjewj~ewjewjherjuewj{ewjewjewj~ewjewjewjewjhiZe#ere,ewfed+<e4Gd,d-d-Ze,erewed.d/d0Ze7je7je/eSeWedfe)eSd1d2d3Ze7jed4d5gd6d7e7jd8d9eSd:d;e7jddd?e7jd@dAe7dBdCerDdDdEddFdGe7jdHddIdJe7jdKddLdJe7jdMdNddOdJe7jdPddQdJe7jdRddSdJe7jdTddUdJe7jdVeSeQdWdd?e7jdXeSePdYdd?e7jdZd[dd\dJe7jd]d^dd_dJe7jeNde7jd`dae7jddddddbddce7jdde7jdeddeddedbdedfdge7je7je)eSeWe(ereeeeeeeeeSeSe-eSe)eSddhdidjZe-eSeee7jddkdldmZeeepednddodpdqZe,eeepednddrdsdtZe,eeepednejeddudvdwZepjdfeeeepe edxdydzZepjd{eepeed|d}d~ZeSeeeTdddZeSeeTdddZee-eTeUeVfdddZe,ere(eIdddZdzeSe&ere?dddZe?eSdddZe.dZGddde%eZe4GdddeeZeGjeGjeGjhZeijeijeijeijeijeijeijeijhZdZdeGje<ddhZeGjeGjeGjeGjeGjeGjhZeGjeGjeGjeGjeGjeGjeGjeGjeGjeGjeGjeGjeGjeGjhZeGjeGjhZeeGjhBZeijeijeijeijeijhZeijeijeijeijeijhZeijeijeijeijeijeijeijeijeijeijeijeijeijeijeijhZhdZdZdZdZdZdZdZeGjdeGjdeGjdeGjdeGjdeGjdeGjdeGjdeGjdeGjdeGjdeGjdeGjdeGjdiZdZe4GdddZe4GdddZe4GdddZe4GdddeeZeijeijeijhZeGjeGjeGjeGjeGjeGjiZeeZeeZeeBZeeGjehBZe@eeSdddZe)e]e)e@dddZ e?e]e)e]dddZ e@e]dddZ d{e@e)e@e[dddZ d|e@e)e@e[dddZ hd£ZhdãZe]e'e@dddńZe4GddDŽdǃZe ddɍeSee(edʜdd̄ZeSeSd͜ddτZd}eeWee"ewe'edМdd҄Zd~ee"ewe'edӜddՄZdeeWe"ewe"eZe'ed֜dd؄ZeeeddٜddۄZdedܜe(e@ee@eedݜdd߄Ze^e^dddZedee"ewe'edӜddZedee"ewe'edӜddZe@edddZde@eSedddZe@eddddZde@eddddZe@ddddZ e@ddddZ!eSeSdddZ"e?e,eSddddZ#e?ddddZ$e?edddZ%e@e'e]dddZ&e]e]edddZ'e]edddZ(e]eddd Z)e]e)e]dd d Z*dd e?e]edd ddZ+e]edddZ,e]edddZ-e]edddZ.e@e,eYedddZ/e@edddZ0e?edddZ1e]edddZ2e]e[ddd Z3e@ddd!d"Z4ee@ed#d$d%Z5e?e,ewdd&d'Z6e?e,erdd(d)Z7eeWe'e,eZd*d+d,Z8e?e,eSdd-d.Z9e ee=d/d0d1Z:eee*eSe*eSdne=e'ed2d3d4Z;e e&eSed5d6d7Ze:j>e;j>fdd9d:Z?e/ej>e:j>e;j>fe/ej>e:j>e;j>fdd;d<Z@eSeSdd=d>d?ZAeSeSedd@dAdBZBeSeSdCdDdEZCee'ddFdGdHZDeSeSeSeSeSdIdJdKZEe&ejFddLdMdNZGejddOdPdQZHe*eSeSeSeSdRdSdTZIeSe*eSdUdVdWZJe+ee'e-e\efdXdYdZZKdeee'e-e\e@eWfd[d\d]ZLdd^eeWeSed_d`daZMeedbdcddZNeeWed*dedfZOeedgdhdiZPeecdgdjdkZQeebdldmdnZRece&ee-e,ee,efdodpdqZSece&eeddrdsdtZTddFdudvZUddFdwdxZVeWdykreVdS(N)ExecutorProcessPoolExecutor)contextmanager)datetime)Enum) lru_cachepartialwraps)Managerfreeze_support)Path)AnyCallable CollectionDict GeneratorGenericIterableIteratorListOptionalPatternSequenceSetTupleTypeVarUnioncast)user_cache_dir) dataclassevolveFactory)ast3ast27)PathSpec)NodeLeaf type_repr)pygrampytree)drivertoken)Grammar) ParseError)versionXzV/(\.eggs|\.git|\.hg|\.mypy_cache|\.nox|\.tox|\.venv|\.svn|_build|buck-out|build|dist)/z\.pyi?$blackLineFeatureT)bolderrred)fgr4c@seZdZdZdS)NothingChangedz3Raised when reformatted code is the same as source.N__name__ __module__ __qualname____doc__r=r=$lib/python3.9/site-packages/black.pyr7Usr7c@seZdZdZdS) CannotSplitzBA readable split that fits the allotted line length is impossible.Nr8r=r=r=r>r?Ysr?c@seZdZdZdS) InvalidInputz7Raised when input source code fails all parse attempts.Nr8r=r=r=r>r@]sr@c@s2eZdZdZdZdZdZeeeddddZ dS) WriteBackr)checkdiffreturncCs|r|s|jS|r|jS|jSN)CHECKDIFFYES)clsrErFr=r=r>from_configurationgszWriteBack.from_configurationN) r9r:r;NOrKrJrI classmethodboolrMr=r=r=r>rAas rAc@seZdZdZdZdZdS)ChangedrrBrCN)r9r:r;rNCACHEDrKr=r=r=r>rQosrQc@s6eZdZdZdZdZdZdZdZdZ e dd d Z d S) TargetVersionrCrDrGcCs |tjuSrH)rSPY27selfr=r=r> is_python2~szTargetVersion.is_python2N) r9r:r;rZPY33PY34PY35PY36PY37PY38rPr]r=r=r=r>rSusrSc@s0eZdZdZdZdZdZdZdZdZ dZ d Z d S) r2rBrCrDrTrUrVrWrX N) r9r:r;UNICODE_LITERALS F_STRINGSNUMERIC_UNDERSCORESTRAILING_COMMA_IN_CALLTRAILING_COMMA_IN_DEFASYNC_IDENTIFIERSASYNC_KEYWORDSASSIGNMENT_EXPRESSIONSPOS_ONLY_ARGUMENTSr=r=r=r>r2sVERSION_TO_FEATURESc@sTeZdZUeeZeeed<e Z e ed<dZ e ed<dZe ed<eddd Zd S) FileModetarget_versions line_lengthTstring_normalizationFis_pyirYcCs^|jr*dddt|jdddD}nd}|t|jtt|jtt|jg}d|S) N,css|]}t|jVqdSrH)strvalue.0r.r=r=r> sz)FileMode.get_cache_key..cSs|jSrHrv)vr=r=r>z(FileMode.get_cache_key..)key-.)rpjoinsortedrurqintrrrs)r\Z version_strpartsr=r=r> get_cache_keys    zFileMode.get_cache_keyN)r9r:r;r!setrprrS__annotations__DEFAULT_LINE_LENGTHrqrrrrPrsrurr=r=r=r>ros    ro)rpfeaturerGcstfdd|DS)Nc3s|]}t|vVqdSrHrnrwrr=r>ryr}z#supports_feature..)all)rprr=rr>supports_featuresr)ctxparamrvrGc Cst|ttfrJd|sJt|jdd}|d}|rFt|}ndSz"t |}|didi}Wn>tj t fy}z t j |d|d WYd}~n d}~00|sdS|jduri|_|jd d |D|S) zInject Black configuration from "pyproject.toml" into defaults in `ctx`. Returns the path to a successfully found and read configuration file, None otherwise. zInvalid parameter type passedsrcr=pyproject.tomlNZtoolr0z"Error reading configuration file: )filenamehintcSs&i|]\}}|dddd|qS)z--r_)replacerxkr{r=r=r> r}z'read_pyproject_toml..) isinstancerrPfind_project_rootparamsgetis_filerutomlloadZTomlDecodeErrorOSErrorclickZ FileErrorZ default_mapupdateitems)rrrvrootpathZpyproject_tomlconfiger=r=r>read_pyproject_tomls,    rz-hz--help)Zhelp_option_names)Zcontext_settingsz-cz--codez&Format the code passed in as a string.)typehelpz-lz --line-lengthz&How many characters per line to allow.)rdefaultrZ show_defaultz-tz--target-versioncCsg|]}|jqSr=)namelower)rxr{r=r=r> r}rcCsdd|DS)NcSsg|]}t|qSr=)rSupper)rxvalr=r=r>rr}z..r=)cpr{r=r=r>r|r}r|z^Python versions that should be supported by Black's output. [default: per-file auto-detection])rcallbackZmultiplerz--py36zAllow using Python 3.6-only syntax on all input files. This will put trailing commas in function signatures and calls also after *args and **kwargs. Deprecated; use --target-version instead. [default: per-file auto-detection])Zis_flagrz--pyiztFormat all input files like typing stubs regardless of file extension (useful when piping source on standard input).z-Sz--skip-string-normalizationz*Don't normalize string quotes or prefixes.z--checkzDon't write the files back, just return the status. Return code 0 means nothing would change. Return code 1 means some files would be reformatted. Return code 123 means there was an internal error.z--diffzGDon't write the files back, just output a diff for each file on stdout.z --fast/--safez@If --fast given, skip temporary sanity checks. [default: --safe]z --includea%A regular expression that matches files and directories that should be included on recursive searches. An empty value means all files are included regardless of the name. Use forward slashes for directories on all platforms (Windows, too). Exclusions are calculated first, inclusions later.z --excludea A regular expression that matches files and directories that should be excluded on recursive searches. An empty value means no paths are excluded. Use forward slashes for directories on all platforms (Windows, too). Exclusions are calculated first, inclusions later.z-qz--quietzbDon't emit non-error messages to stderr. Errors are still emitted; silence those with 2>/dev/null.z-vz --verbosezaAlso emit messages to stderr about files that were not changed or were ignored due to --exclude=.r)existsZ file_okayZdir_okayreadableZ allow_dash)nargsris_eagerz--configFzRead configuration from PATH.)rrrr)rcoderqtarget_versionrErFfastpyipy36skip_string_normalizationquietverboseincludeexcluderrrGc CsNtj||d}|r4|r*td|dqLt|}n|rFtdt}nt}t|||| d}|r|| r|td|ddd d |d urtt ||d |d z t | }Wn,t j ytd| |dYn0z t | }Wn.t j ytd| |dYn0t || | d}t|}t}t|| | ||D]d}t|}|rv|t|||||t|n.|s|dkr||ntd|qBt|d kr| s| std|d t|dkrt|||||dnt|||||d| s| s>t|jr&dndtjt|dd||jd S)z"The uncompromising code formatter.)rErFz+Cannot use both --target-version and --py36rCz`--py36 is deprecated and will be removed in a future version. Use --target-version py36 instead.)rprqrsrrzUsing configuration from rFblue)r3r6Nmoderz.Invalid regular expression for include given: z.Invalid regular expression for exclude given: )rErrrzinvalid path: u?No Python files are present to be formatted. Nothing to do 😴rB)rr write_backrreport)sourcesrrrruOh no! 💥 💔 💥uAll done! ✨ 🍰 ✨T)r4)rArMr4exitr PY36_VERSIONSrooutprint format_strre_compile_maybe_verbosereerrorReportr path_emptyr is_dirrgen_python_files_in_dir get_gitignoreraddlen reformat_onepop reformat_many return_codersechoru)rrrqrrErFrrrrrrrrrrrversionsrZ include_regexZ exclude_regexrrrsrr=r=r>mains           r)rrrrrGcCs"|s|s |std|ddS)z; Exit if there is no `src` provided for formatting u$No Path provided. Nothing to do 😴rN)rr)rrrrr=r=r>rsrr)rrrrrrGc Csztj}|s2t|dkr2t|||drtj}ni}|tjkrnt|}| }||vrn||t |krntj }|tj urt ||||drtj}|tjur|tj us|tj ur|tjurt||g||||Wn6ty}z||t|WYd}~n d}~00dS)zReformat a single file under `src` without spawning child processes. `fast`, `write_back`, and `mode` options are passed to :func:`format_file_in_place` or :func:`format_stdin_to_stdout`. r)rrrN)rQrNrruformat_stdin_to_stdoutrKrArJ read_cacheresolveget_cache_inforRformat_file_in_placerI write_cachedone Exceptionfailed) rrrrrchangedcacheZres_srcexcr=r=r>rs0  r)rrrrrrGc Csvt}t}tjdkr$t|d}t|d}z0|t |||||||dWt || nt || 0dS)z4Reformat multiple files using a ProcessPoolExecutor.win32=) max_workers)rrrrrloopexecutorN) asyncioZget_event_loopos cpu_countsysplatformminrrun_until_completeschedule_formattingshutdown)rrrrrrZ worker_countrr=r=r>rs*     r)rrrrrrrrGcsi}tjkr@t}t||\}}t|D]} || tjq,|sHdSg} g} dtjkrlt} | fddt|D} | }z$ t j t| t jt|WntyYn0|rtj|tjdIdH\}}|D]}| |} |r| |q|r4|| t|q|rDtjntj}tjusntjurx|tjurx| | || |qq| rtj| ddIdH| rt|| dS)zRun formatting of `sources` in parallel using the provided `executor`. (Use ProcessPoolExecutors for actual parallelism.) `write_back`, `fast`, and `mode` options are passed to :func:`format_file_in_place`. Nc s*i|]"}tt||qSr=)rZ ensure_futureZrun_in_executorrrxrrrlockrrrr=r>rDsz'schedule_formatting..)Z return_whenTrZreturn_exceptions) rArJr filter_cachedrrrQrRr LockkeysZadd_signal_handlersignalSIGINTcancelSIGTERMNotImplementedErrorrwaitZFIRST_COMPLETEDr cancelledappend exceptionrruresultrKrNrIgatherr)rrrrrrrrcachedrrZsources_to_cacheZmanagertaskspendingrrtaskrr=rr>r#sV          r)rrrrrrGcCsl|jdkrt|dd}t|j}t|d"}t|\}}} Wdn1sX0Yzt |||d} Wnt yYdS0||j krt|d|| d } | | Wdn1s0Yn||j krht} |d |d } |d | d }t|| | |}|pt8tjtjj|| dd } | || Wdn1s^0YdS) zFormat file under `src` path. Return True if changed. If `write_back` is DIFF, write a diff to stdout. If it is YES, write reformatted code to the file. `mode` and `fast` options are passed to :func:`format_file_contents`. z.pyiTrsrbNrrFw)encodingnewline  +0000rr write_through)suffixr rZutcfromtimestampstatst_mtimeopen decode_bytesreadformat_file_contentsr7rKwriterJutcnowrF nullcontextio TextIOWrapperrstdoutbufferdetach)rrrrrthenbuf src_contentsrr dst_contentsfnowsrc_namedst_nameZ diff_contentsr=r=r>rjs6   0  *  (r)r)rrrrGc Cs2t}ttjj\}}}|}zzt|||d}WWtj tj j||dd}|t j krf| |n>|t jkrt} d|d} d| d} | t||| | |dSty:YWtj tj j||dd}|t j kr| |n@|t jkr.t} d|d} d| d} | t||| | |dS0Wtj tj j||dd}|t j krl| |n@|t jkrt} d|d} d| d} | t||| | |nxtj tj j||dd}|t j kr| |n@|t jkr$t} d|d} d| d} | t||| | |0dS) zFormat file on stdin. Return True if changed. If `write_back` is YES, write reformatted code back to stdout. If it is DIFF, write a diff to stdout. The `mode` argument is passed to :func:`format_file_contents`. rTrzSTDIN rzSTDOUT FN)rr#rrstdinr(r r!r%r&r'rArKr"rJrFr)r7) rrrr*rrrdstr.r/r0r1r=r=r>rsl                          r)r,rrrGcCsH|dkrtt||d}||kr(t|sDt||t|||d|S)zReformat contents a file and return new contents. If `fast` is False, additionally confirm that the reformatted code is valid by calling :func:`assert_equivalent` and :func:`assert_stable` on it. `mode` is passed to :func:`format_str`. rr)stripr7rassert_equivalent assert_stable)r,rrr-r=r=r>r!s   r!)r,rrGcst||j}g}t|}|jr*|jnt|t|td|vpNttj |j |j d}t |j d}t }d}fddtjtjhD} ||D]n} t|D]} |t|q|| \} }t| D]} |t|qt| |j| dD]} |t| qqd|S) zReformat a string and return new contents. `mode` determines formatting options, such as how many characters per line are allowed. Zunicode_literals)remove_u_prefixrsnormalize_stringsrrcsh|]}t|r|qSr=)r)rxrrr=r> s zformat_str..rqfeaturesr) lib2to3_parselstriprpget_future_importsdetect_target_versionsnormalize_fmt_off LineGeneratorrr2rersrrEmptyLineTrackerr1rhrivisitranger rumaybe_empty_lines split_linerqr)r,rZsrc_noder-Zfuture_importslineseltZ empty_lineafterZsplit_line_features current_linerbeforeliner=r9r>rs<       r)rrGcCst|}t|j\}}|s(d|dfSd|dddkr@dnd}|dt||}|||fWdS1s~0YdS)zReturn a tuple of (decoded_contents, encoding, newline). `newline` is either CRLF or LF but `decoded_contents` is decoded with universal newlines (i.e. only contains LF). r s rNz )r%BytesIOtokenizedetect_encodingreadlineseekr&r )rZsrcbufrrHrZtiowr=r=r>rs   r)rprGcCsn|stjtjtjtjgStdd|Dr6tjtjgSg}t|tjsR| tjt|tj sj| tj|S)Ncss|]}|VqdSrH)r]rwr=r=r>ryr}zget_grammars..) r(ZBpython_grammar_no_print_statement_no_exec_statement_async_keywordsZ3python_grammar_no_print_statement_no_exec_statementZ!python_grammar_no_print_statementZpython_grammarrrr2rjr rk)rpZgrammarsr=r=r> get_grammars s$    rUr=)src_txtrprGc Cs|dddkr|d7}tt|D]}t|tj}z||d}WqWq$ty}zd|jd\}}| }z||d} Wnt yd} Yn0t d|d|d | } WYd}~q$d}~00q$| dt |t rttj|g}|S) z4Given a string with source, return the lib2to3 Node.rNrNTrBzzCannot parse: :: )rUrr*ZDriverr)ZconvertZ parse_stringr-context splitlines IndexErrorr@rr&r%syms file_input) rVrpZgrammardrvr ZpelinenocolumnrHZ faulty_linerr=r=r>r=0s&    2 r=noderGcCs t|}|S)z7Given a lib2to3 node, return its string representation.ru)rbrr=r=r>lib2to3_unparseKsrdTc@s8eZdZdZeeedddZeeedddZdS)VisitorzBBasic lib2to3 visitor that yields things of type `T` on `visit()`.raccsD|jdkrtj|j}n t|j}t|d||j|EdHdS)azMain method to visit `node` and its children. It tries to find a `visit_*()` method for the given `node.type`, like `visit_simple_stmt` for Node objects or `visit_INDENT` for Leaf objects. If no dedicated `visit_*()` method is found, chooses `visit_default()` instead. Then yields objects of type `T` from the selected visitor. Zvisit_N)rr+tok_namer'getattr visit_default)r\rbrr=r=r>rDWs  z Visitor.visitccs*t|tr&|jD]}||EdHqdS)CDefault `visit_*()` implementation. Recurses to children of `node`.N)rr%childrenrDr\rbchildr=r=r>rjgs  zVisitor.visit_defaultN) r9r:r;r<LNrrerDrjr=r=r=r>rfTsrfc@sLeZdZUdZeed<eeedddZ e e e e efdddd ZdS) DebugVisitorr tree_depthraccsdd|j}t|trt|j}t||dd|jd7_|jD]}||EdHqJ|jd8_t|d|dddn`tj |jt |j}t||d dd |j rtd|j d ddd td|j d dddS) N rCZyellowr6rB/F)r6r3r)r6nlZgreen)r6r3ru)rqrr%r'rrrlrDr+rhrruprefixrv)r\rbindentZ_typernr=r=r>rjrs   zDebugVisitor.visit_defaultN)rrGcCs*t}t|trt|}t||dS)zmPretty-print the lib2to3 AST of a given string of `code`. Convenience method for debugging. N)rprrur=listrD)rLrr{r=r=r>shows zDebugVisitor.show)r9r:r;rqrrrorrerjrOrrur&r%ryr=r=r=r>rpns  rpSTANDALONE_COMMENTandor>z-=z%=z&=z+==z@=z>>=z/=z//=z|=z^=z*=z**=z<<= rdrXrWrVrUrTrDrCrBc@s&eZdZUdZdZeed<eeZ e e e e fefed<eeZe eefed<dZeeed<eeZeeed<eeZeeed <edd d d Zed ddZd!eeedddZd"eedddZeed ddZeed ddZeed ddZ eed ddZ!eed dd Z"dS)#BracketTrackerz"Keeps track of brackets on a line.rdepth bracket_match delimitersNprevious_for_loop_depths_lambda_argument_depthsleafrGcCs|jtjkrdS|||||jtvrV|jd8_|j|j|jf}||_ |j|_ |jdkrt ||j }|r|j dur||j t|j <nt||j }|r||j t|<|jtvr||j|jt|jf<|jd7_||_ ||||dS)aMark `leaf` with bracket-related metadata. Keep track of delimiters. All leaves receive an int `bracket_depth` field that stores how deep within brackets a given leaf is. 0 means there are no enclosing brackets that started on this line. If a leaf is itself a closing bracket, it receives an `opening_bracket` field that it forms a pair with. This is a one-directional link to avoid reference cycles. If a leaf is a delimiter (a token on which Black can split the line if needed) and it's on depth 0, its `id()` is stored in the tracker's `delimiters` field. NrBr)rr+COMMENT'maybe_decrement_after_for_loop_variable&maybe_decrement_after_lambda_argumentsCLOSING_BRACKETSrrropening_bracket bracket_depthis_split_before_delimiterrridis_split_after_delimiterOPENING_BRACKETSBRACKET maybe_increment_lambda_arguments!maybe_increment_for_loop_variable)r\rrdelimr=r=r>mark s,         zBracketTracker.markrYcCs t|jS)zBReturn True if there is an yet unmatched open bracket on the line.)rPrr[r=r=r>any_open_brackets5sz BracketTracker.any_open_bracketsr=)rrGcstfdd|jDS)zReturn the highest priority of a delimiter found on the line. Values are consistent with what `is_split_*_delimiter()` return. Raises ValueError on no delimiters. c3s|]\}}|vr|VqdSrHr=rrr=r>ry?r}z8BracketTracker.max_delimiter_priority..)maxrr)r\rr=rr>max_delimiter_priority9sz%BracketTracker.max_delimiter_priority)priorityrGcs2|js dSp|tfdd|jDS)zReturn the number of delimiters with the given `priority`. If no `priority` is passed, defaults to max priority on the line. rc3s|]}|krdVqdS)rBNr=)rxrrr=r>ryJr}z?BracketTracker.delimiter_count_with_priority..)rrsumvalues)r\rr=rr>delimiter_count_with_priorityAs z,BracketTracker.delimiter_count_with_prioritycCs:|jtjkr6|jdkr6|jd7_|j|jdSdS)zIn a for loop, or comprehension, the variables are often unpacks. To avoid splitting on the comma in this situation, increase the depth of tokens between `for` and `in`. forrBTF)rr+NAMErvrrr r\rr=r=r>rLs z0BracketTracker.maybe_increment_for_loop_variablecCsL|jrH|jd|jkrH|jtjkrH|jdkrH|jd8_|jdSdS)z>See `maybe_increment_for_loop_variable` above for explanation.rinrBTF)rrrr+rrvrrr=r=r>rYs  z6BracketTracker.maybe_decrement_after_for_loop_variablecCs:|jtjkr6|jdkr6|jd7_|j|jdSdS)zIn a lambda expression, there might be more than one argument. To avoid splitting on the comma in this situation, increase the depth of tokens between `lambda` and `:`. lambdarBTF)rr+rrvrrr rr=r=r>rgs z/BracketTracker.maybe_increment_lambda_argumentscCsB|jr>|jd|jkr>|jtjkr>|jd8_|jdSdS)z=See `maybe_increment_lambda_arguments` above for explanation.rrBTF)rrrr+COLONrrr=r=r>rts  z5BracketTracker.maybe_decrement_after_lambda_argumentscCs|j|jdtjfS)z7Return the most recent opening square bracket (if any).rB)rrrr+RSQBr[r=r=r> get_open_lsqbszBracketTracker.get_open_lsqb)r=)r)#r9r:r;r<rrrr!dictrrrDepthNodeTyper&rLeafIDPriorityrrrxrrrrrPrrrrrrrrrr=r=r=r>rs   (    rc@seZdZUdZdZeed<eeZ e e ed<ee Z eee e fed<eeZeed<dZeed<dZeed <d=e ed d d d Zd>e ed d ddZeedddZeedddZeedddZeedddZeedddZeedddZeedddZeeddd Zeedd!d"Z e!j"feed#d$d%Z#edd&d'Z$edd(d)Z%edd*d+Z&e ed,d-d.Z'e ed/d0d1Z(e e e d2d3d4Z)d dd5d6Z*e ed2d7d8Z+e,dd9d:Z-edd;d<Z.d S)?r1z;Holds leaves and comments. Can be printed with `str(line)`.rrleavescommentsbracket_trackerFinside_bracketsshould_explodeN)r preformattedrGcCs|jtvpt|j}|s dStj|jkr>|jr>|jdd=|jrd|sd|j t || |d7_ |j sn|s|j |||||s|j|dS)agAdd a new `leaf` to the end of the line. Unless `preformatted` is True, the `leaf` will receive a new consistent whitespace prefix and metadata applied by :class:`BracketTracker`. Trailing commas are maybe removed, unpacked for loop variables are demoted from being delimiters. Inline comments are put aside. NrO)complex_subscript)rBRACKETSrPrvr4r+ris_class_paren_emptyrrv whitespaceis_complex_subscriptrrrmaybe_remove_trailing_commaappend_commentr )r\rrZ has_valuer=r=r>r s        z Line.appendcCsD|jjdkr2|jrtd|jr2|jtkr2td|j||ddS)zLike :func:`append()` but disallow invalid standalone comment structure. Raises ValueError when any `leaf` is appended after a standalone comment or when a standalone comment is not the first leaf on the line. rz$cannot append to standalone commentsz5cannot append standalone comments to a populated linerN)rr is_comment ValueErrorrrr{r )r\rrr=r=r> append_safes zLine.append_saferYcCst|jdko|jdjtkS)z"Is this line a standalone comment?rBr)rrrr{r[r=r=r>rszLine.is_commentcCst|o|jdjtjkS)zIs this line a decorator?r)rPrrr+ATr[r=r=r> is_decoratorszLine.is_decoratorcCst|ot|jdS)zIs this an import line?r)rP is_importrr[r=r=r>rszLine.is_importcCs*t|o(|jdjtjko(|jdjdkS)z Is this line a class definition?rclass)rPrrr+rrvr[r=r=r>is_classs z Line.is_classcCs&|jo$|jddddtdDkS)zEIs this line a class definition with a body consisting only of "..."?NcSsg|]}ttjdqS)rr&r+DOT)rxrr=r=r>rsz&Line.is_stub_class..rD)rrrEr[r=r=r> is_stub_classszLine.is_stub_classcCsd|jrt|jdkrdS|jdjtjkr:|jd}d}n|jd}d}|jtvsX|jr\dS|jtjkrD|j}t |jD]\}}||urzqqzdSd}|j|dj }|j|d|D]"}|j |kr|jtj kr|d7}q|dkr|j|djtj kS|dkr dS|j|djtvrD|d8}|j|}|jtjkrDdS|j|djtj kr`dSdS)zIs this line a collection literal with a trailing comma that's optional? Note that the trailing comma in a 1-tuple is not optional. rTFrrOrrBT) rrrr+rrrRPARr enumeraterCOMMA)r\ZcloserZ close_indexopenerZ _open_indexrZcommasZ comma_depthr=r=r>*is_collection_with_optional_trailing_commas@      z/Line.is_collection_with_optional_trailing_commacCsz|jd}Wnty"YdS0z|jd}WntyHd}Yn0|jtjkr`|jdkp|jtjko|duo|jtjko|jdkS)zBIs this a function definition? (Also returns True for async defs.)rFrBNdef)rr[rr+rrvASYNC)r\ first_leafZ second_leafr=r=r>is_defs      z Line.is_defcCs`t|o^t|jdko^|jo^|jdjtjko^|jdjdko^|jdjtjko^|jdjdkS)zzIs this a class with no base classes but using parentheses? Those are unnecessary and should be removed. rTrC(rD)) rPrrrrr+LPARrvrr[r=r=r>r/s zLine.is_class_paren_emptycCs,t|o*|jdjtjko*|jdjdS)z#Is the line a triple quoted string?r)"""''')rPrrr+STRINGrv startswithr[r=r=r>is_triple_quoted_string?s zLine.is_triple_quoted_string) depth_limitrGcCs*|jD]}|jtkr|j|krdSqdS)z)If so, needs to be split before emitting.TF)rrr{r)r\rrr=r=r>contains_standalone_commentsHs z!Line.contains_standalone_commentscCst}zR|jd}|t||jtjks>|jtjkrV|jsV|jd}|t|Wnt ylYdS0d}|j D]2\}}|D]$}t |r||vs|rdSd}qq|dS)NrrOFT) rrrrrr+rrrvr[rris_type_comment)r\Z ignored_ids last_leafZ comment_seenZleaf_idrcommentr=r=r>$contains_uncollapsable_type_commentsPs*      z)Line.contains_uncollapsable_type_commentscCs|js dStdd|jDd}tddt|jDd}||kr|jddD].}|jt|gD]}t|drfdSqfqPdS) NFcss|]}|jdkr|jVqdSrNr_rxlr=r=r>ryr}z9Line.contains_unsplittable_type_ignore..rcss|]}|jdkr|jVqdSrrrr=r=r>ryr}rOz ignoreT)rnextreversedrrrr)r\Z first_line last_linerbrr=r=r>!contains_unsplittable_type_ignoreqs  z&Line.contains_unsplittable_type_ignorecCs|jD]}t|rdSqdS)NTF)ris_multiline_stringrr=r=r>contains_multiline_stringss zLine.contains_multiline_strings)closingrGcCs|jr|jdjtjksdS|jr|jrt|jdkr|jdjtjkr|jtvr|jdjtjkr|jdjdkst|jdkr|jdjdkr|jdjd kst|jd kr|jd jdkr|jdjd kr|jtj ksdS| d S) z4Remove trailing comma if there is one and it's safe.rFrTimportrVirasrUT) rrr+rrrrrrvrremove_trailing_comma)r\rr=r=r>rsB      z Line.maybe_remove_trailing_comma)rrGcCs|jtkr|jrd|_dS|jtjkr.dS|jsDt|_d|_dS|jd}|jtjkr|j s|j rt t |j dkrt |st |jdkrt|_d|_dS|jd}|jt|g|dS)z0Add an inline or standalone comment to the line.rFrrDrCrOT)rr{rrrvr+rrrrvparentrrxrr setdefaultrr )r\rrr=r=r>rs<     zLine.append_commentrcCs|jt|gS)z;Generate comments that should appear directly after `leaf`.)rrrrr=r=r>comments_afterszLine.comments_aftercCs>|j}|jt|g}|jt|jdg|dS)z@Remove the trailing comma and moves the comments attached to it.rN)rrrrrextend)r\Ztrailing_commaZtrailing_comma_commentsr=r=r>rs  zLine.remove_trailing_commacCsj|j}|durdS|j}t|trL|jtjkr6dS|jtjkrLt ||}|duoht dd| DS)zAReturn True iff `leaf` is part of a slice with non-trivial exprs.NFcss|]}|jtvVqdSrH)rTEST_DESCENDANTSrxnr=r=r>rysz,Line.is_complex_subscript..) rr next_siblingrr%rr\ listmaker subscriptlist child_towardsany pre_order)r\rZ open_lsqbZsubscript_startr=r=r>rs     zLine.is_complex_subscriptcCsz|sdSd|j}t|j}t|}|j||j}|D]}|t|7}q__str__s  z Line.__str__cCst|jp |jS)z/Return True if the line has leaves or comments.)rPrrr[r=r=r>__bool__sz Line.__bool__)F)F)/r9r:r;r<rrrr!rxrrr&rrrrrrrrPrr rpropertyrrrrrrrrrrmaxsizerrrrrrrrrrurrr=r=r=r>r1sL    >!%%c@seZdZUdZdZeed<dZee ed<dZ e ed<e e Zee ed<e ee e fd d d Ze ee e fd d d Ze e ee e fdddZdS)rCa{Provides a stateful method that returns the number of potential extra empty lines needed before and after the currently processed line. Note: this tracker works on lines that haven't been split yet. It assumes the prefix of the first leaf consists of optional newlines. Those newlines are consumed by `maybe_empty_lines()` and included in the computation. FrsN previous_linerprevious_after previous_defs)rKrGcCs:||\}}|jdurdn||j}||_||_||fS)zReturn the number of extra empty lines before and after the `current_line`. This is for separating `def`, `async def` and `class` with extra empty lines (two on module-level). Nr)_maybe_empty_linesr r )r\rKrLrJr=r=r>rF"sz"EmptyLineTracker.maybe_empty_linescCsd}|jdkr|jrdnd}|jrJ|jd}|jd}t||}d|_nd}|j}|jr|jd|kr|j|jr|rdnd}qT|rdnd}qT|js|j s|j r| ||S|j r|j j r|j s||j jkr|pddfS|j r|j j r|jr|dfS|dfS)NrBrrCrNrr)rrsrrvcountrr rrrr#_maybe_empty_lines_for_class_or_defr rr)r\rKZ max_allowedrrLrr=r=r>r4sD        z#EmptyLineTracker._maybe_empty_lines)rKrLrGcCs|js|j|j|jdur"dS|jjr.dS|jj|jkrP|jjsL|jjrPdS|jjrr|jj|jkrr|dkrrdS|jr|jj|jkrd}q|js|jjr|j r|jj rd}qd}q|jr|jjsd}qd}nd}|jr|r|d8}|dfS)N)rrrrBrC) rr r rr rrrrsr)r\rKrLnewlinesr=r=r>r[sB   z4EmptyLineTracker._maybe_empty_lines_for_class_or_def)r9r:r;r<rsrPrr rr1r rr!rxr rrrFrrr=r=r=r>rCs   ( rCcsneZdZUdZdZeed<dZeed<ee Z e ed<dZ eed<d+e e e d d d Zee e d fd d Zee e d ddZee e d ddZeeeeee e dddZee e d ddZee e d ddZee e d ddZee e d ddZee e ddd Zee e dd!d"Zee e dd#d$Zee e d d%d&Zd'd(d)d*Z Z!S),rBzGenerates reformatted Line objects. Empty lines are not emitted. Note: destroys the tree it's visiting by mutating prefixes of its leaves in ways that will no longer stringify to valid Python code on the tree. FrsTr8rKr7r)rwrGccs<|js|jj|7_dS|j}t|j|d|_|VdS)zGenerate a line. If the line is empty, only emit if it makes sense. If the line is too long, split it first and then generate. If any lines were generated, set up a new current_line. Nr)rKrr1)r\rwZ complete_liner=r=r>rMs zLineGenerator.linerac#st|tr|jj}t|D]f}|r4|j|q|jtj kr\|j|| EdHq| EdH|j|| EdHqt ||d|j r|jtj krt||jdt||jtjkrt||jtvr|j|t|EdHdS)rkNr)r7)rr&rKrrgenerate_commentsr rr+rrMnormalize_prefixr8rnormalize_string_prefixr7normalize_string_quotesNUMBERnormalize_numeric_literal WHITESPACEsuperrj)r\rbrr __class__r=r>rjs(          zLineGenerator.visit_defaultccs$|dEdH||EdHdS)z/Increase indentation level, maybe yield a line.rBNrMrjr\rbr=r=r> visit_INDENTszLineGenerator.visit_INDENTccs2|EdH||EdH|dEdHdS)z/Decrease indentation level, maybe yield a line.Nrrrr=r=r> visit_DEDENTszLineGenerator.visit_DEDENT)rbkeywordsparensrGccsPt||d|jD]8}|jtjkr:|j|vr:|EdH||EdHqdS)aVisit a statement. This implementation is shared for `if`, `while`, `for`, `try`, `except`, `def`, `with`, `class`, `assert` and assignments. The relevant Python language `keywords` for a given statement will be NAME leaves within it. This methods puts those on a separate line. `parens` holds a set of string leaf values immediately after which invisible parens should be put. ) parens_afterN)normalize_invisible_parensrlrr+rrvrMrD)r\rbr"r#rnr=r=r> visit_stmts   zLineGenerator.visit_stmtccs:|jr&t|r&||jdEdHn||EdHdS)zVisit a suite.rCN)rs is_stub_suiterDrlrjrr=r=r> visit_suiteszLineGenerator.visit_suiteccs|jo|jjtv}|rh|jr6t|r6||EdHq|dEdH||EdH|dEdHn4|jr~|jr~t|js|EdH||EdHdS)z,Visit a statement without nested statements.NrBr)rr STATEMENTrs is_stub_bodyrjrMr')r\rbZ is_suite_liker=r=r>visit_simple_stmtszLineGenerator.visit_simple_stmtccsj|EdHt|j}|D]$}||EdH|jtjkrqBqt|}|jD]}||EdHqPdS)z-Visit `async def`, `async for`, `async with`.N)rMrrlrDrr+rr)r\rbrlrnZ internal_stmtr=r=r>visit_async_stmts   zLineGenerator.visit_async_stmtccs.|jD]"}|EdH||EdHqdS)zVisit decorators.N)rlrMrDrmr=r=r>visit_decoratorss zLineGenerator.visit_decoratorsrccs|EdHdS)zBRemove a semicolon and put the other statement on a separate line.N)rMrr=r=r> visit_SEMIszLineGenerator.visit_SEMIccs"||EdH|EdHdS)zAEnd of file. Process outstanding comments and end with a newline.N)rjrMrr=r=r>visit_ENDMARKERszLineGenerator.visit_ENDMARKERccs.|jjs|EdH||EdHdSrH)rKrrrMrjrr=r=r>visit_STANDALONE_COMMENT!s z&LineGenerator.visit_STANDALONE_COMMENTccs|j\}}|jtjkrtt|jdkrt|jdjtjkrtttjd}ttj d}| pXd}| |t tj |||g||EdHdS)z_Force parentheses between a unary op and a binary power: -2 ** 8 -> -(2 ** 8) rDrBrrrN)rlrr\powerrr+ DOUBLESTARr&rrremove insert_childr%atomrj)r\rb _operatoroperandlparrparindexr=r=r> visit_factor&s      zLineGenerator.visit_factorNrYcCs |j}t}t|dhddhd|_t|hdddhd|_t|ddhdhd|_t|d dhd d hd|_t|hd |d|_t|d h|d|_t|d h|d|_ t|dh|d|_ t|dh|d|_ t||t d|_ t|dhdhd|_t||dhd|_t||dhd|_|j|_|j|_dS)z,You are in a twisty little maze of passages.assertrt)r"r#>elseelififr?r>whiler=rr>finallytryr=exceptrCwithrrrGrdelN)r&rrZvisit_assert_stmtZ visit_if_stmtZvisit_while_stmtZvisit_for_stmtZvisit_try_stmtZvisit_except_clauseZvisit_with_stmtZ visit_funcdefZvisit_classdef ASSIGNMENTSZvisit_expr_stmtZvisit_return_stmtZvisit_import_fromZvisit_del_stmtr,Zvisit_async_funcdefr-Zvisit_decorated)r\r{Ør=r=r>__attrs_post_init__7s* z!LineGenerator.__attrs_post_init__)r)"r9r:r;r<rsrPrr8r!r1rKr7rrrMrorjr%r r!rrur&r(r+r,r-r&r.r/r0r;rH __classcell__r=r=rr>rBs*     rB)rrrGc CsJd}d}d}|j}|j}|j}|tvr*|S|tjkr8|S|dusNJd||tjkrr|jtjtj tj hvrr|S|j }|st |} | r| jt vr|S|tjkr| jtjkr|S| jtjkr|s|S|S| jtjkr| jr| jjtjtjtjtjhvr|S| jjtjkr| jSn| jtvr>t| ttBdr|Sn| jtjkrz| jr| jjtjtj hvr|rt|S|Snx| jr| jjtjkr| jtvr|S| jtjkr| jr| jjtjkr| j r| j jtjkr| j jdkr|Sn|jt vr|S|jtjtjhvr4|r,|jtjkrF|Sn|jtjkr^|rF|jtjkrF|Sn|jtjkr|sv|S|tjkr|jtjkr|Sn&|jtjkr|jS|jtjkrF|Sn|jtjkr|sFt |} | r| jtjkrF|SnN|jtj kr~|tj!ks|tj"kr"|S|sh|tj#krVt |} | rP| jtj$krf|Sn|tj%krz|Sn|jtjkrF|Sn|jtjkr|tjkr|S|st |} | r| jtj!kr|Sn|jtjhtBvrF|Snf|jtj&kr|S|jtj'kr<|r |St |} | r4| jtj(ks4| jtj#krF|Sn |jtj)krv|tj!krZ|S|rF|jtj!krF|Sn|jtjtj hvr|s|jdusJd|jjtj kr|S|S|sF|Sn||jtj*kr|rF|tj#krF|SnT|jtj+kr|rF|jtj,krF|Sn*|jtjtj-hvr|st |} | rP| jt vrT|S| j} | dushJ| jtjkr| jtjtj hvr|S| jtjkr| jtjkr|Sn|tjtj$tj.hvrF|Snz|jtj/kr4|tj#kr|r2|jtj#kr2|Sn2|tjkrF|d kr|S|rF|jtj#krF|Sn|jtj krF|S|S) zReturn whitespace prefix if needed for the given `leaf`. `complex_subscript` signals whether the given leaf is part of a subscription which has non-trivial arguments, like arithmetic expressions or function calls. rrr Nz/INTERNAL ERROR: hand-made leaf without parent: withinrzsubscripts are always parentedr)0rrrvALWAYS_NO_SPACEr+rrr\Z subscriptrZsliceop prev_siblingpreceding_leafrrEQUALarglistargument parameters varargslist typedargslistrvVARARGS_SPECIALS is_varargVARARGS_PARENTSUNPACKING_PARENTSfactorMATH_OPERATORS RIGHTSHIFT shift_exprrZtnametrailerrrrrLSQB decorator dotted_namerclassdefr5 dictsetmakerr2 star_exprr import_from) rrrNZSPACEZ DOUBLESPACEtrr{prevZprevpZ prevp_parentr=r=r>rXsH               "    rcCsP|rL|j}|rDt|tr|Szt|dWStyBYdS0|j}qdS)z3Return the first leaf that precedes `node`, if any.rN)rNrr&rxrr[r)rbrr=r=r>rO>s  rO)ancestor descendantrGcCs|}|r|j|kr|j}q|S)z:Return the child of `ancestor` that contains `descendant`.r)rhrirbr=r=r>rPsrrcCs`|j}|}|r\|j}|durq\|jdj|kr0q\|jtjkr>q\|jdurV|jjtvrVq\|}q |S)zReturn `leaf` or one of its ancestors that is the topmost container of it. By "container" we mean a node where `leaf` is the very first child. Nr)rvrrlrr\r]rNr)rZ same_prefix containerrr=r=r> container_ofXs rl)rrrGcCs|jtjkrtSdS)zReturn the priority of the `leaf` delimiter, given a line break after it. The delimiter priorities returned here are from those delimiters that would cause a line break after themselves. Higher numbers are higher priority. r)rr+rCOMMA_PRIORITYrrr=r=r>rqs rcCs8t|ttBdrdS|jtjkrP|jrP|jjtjtj hvrP|dusL|jt vrPt S|jt vr~|jr~|jjtj tjhvr~t|jS|jtvrtS|jtjkr|dur|jtjkrtS|jtjtjhvrdS|jdkr|jr|jjtjtjhvs|jtjkrt|jtr|jjdkrtS|jdkrF|jrF|jjtjtjhvrFtS|jdvrn|jrn|jjtjkrnt S|jdkr~tS|jd kr|jr|jjtj!tj"hvr|dur|jtjkr|jd kstS|jd kr|jr|jjtj!kr|dur|jtjkr|jdkstS|jt#vr4|jr4t$SdS) zReturn the priority of the `leaf` delimiter, given a line break before it. The delimiter priorities returned here are from those delimiters that would cause a line break before themselves. Higher numbers are higher priority. rKrNrasyncr?>r=r?isrnot)%rWrXrYrr+rrr\rerar DOT_PRIORITYr[rZrdMATH_PRIORITIES COMPARATORSCOMPARATOR_PRIORITYrSTRING_PRIORITYrrrvZcomp_forZ old_comp_forrrNr&COMPREHENSION_PRIORITYZcomp_ifZ old_comp_iftestTERNARY_PRIORITYZcomp_op comparisonLOGIC_OPERATORSLOGIC_PRIORITYrnr=r=r>rs        $      r>z # fmt:offz# yapf: disablez # fmt: off>z # fmt: onz# yapf: enablez# fmt:onccs:t|j|jtjkdD]}t|j|jd|jdVqdS)aClean the prefix of the `leaf` and generate comments from it, if any. Comments in lib2to3 are shoved into the whitespace prefix. This happens in `pgen2/driver.py:Driver.parse_tokens()`. This was a brilliant implementation move because it does away with modifying the grammar to include all the possible places in which comments can be placed. The sad consequence for us though is that comments don't "belong" anywhere. This is why this function generates simple parentless Leaf objects for comments. We simply don't know what the correct parent should be. No matter though, we can live without this. We really only need to differentiate between inline and standalone comments. The latter don't share the line with any code. Inline comments are emitted as regular token.COMMENT leaves. Standalone are emitted with a fake STANDALONE_COMMENT token identifier.  is_endmarkerrNrvN) list_commentsrvrr+ ENDMARKERr&rvr)rpcr=r=r>rsrc@s2eZdZUdZeed<eed<eed<eed<dS) ProtoCommentaDescribes a piece of syntax that is a comment. It's not a :class:`blib2to3.pytree.Leaf` so that: * it can be cached (`Leaf` objects should not be reused more than once as they store their lineno, column, prefix, and parent information); * `newlines` and `consumed` fields are kept separate from the `value`. This simplifies handling of special marker comments like ``# fmt: off/on``. rrvrconsumedN)r9r:r;r<rrrur=r=r=r>rs  ri)r )rvr~rGc Csg}|rd|vr|Sd}d}d}t|dD]\}}|t|d7}|}|sZ|d7}|dsx|dr.|d7}q.||kr|stj}nt}t |} | t || ||dd}q.|S)zNReturn a list of :class:`ProtoComment` objects parsed from the given `prefix`.#rrNrB\)rrvrr) rsplitrr>rendswithr+rr{ make_commentr r) rvr~r rZnlinesZ ignored_linesr:rMZ comment_typerr=r=r>r s4    r)contentrGcCsH|}|sdS|ddkr(|dd}|r@|ddvr@d|}d|S)aReturn a consistently formatted comment from the given `content` string. All comments (except for "##", "#!", "#:", '#'", "#%%") should have a single space between the hash sign and the content. If `content` didn't start with a hash sign, one is provided. rrrBNz !:#'%rr)rstrip)rr=r=r>r1 s  r)rMrqinnerr<rGc #s|jr|VdSt|d}|sR|jsR|jsRt||dsH|rR|VdS|jr`t g}n6t t t t t dfdd }|jrtt|g}n|g}|D]t}g}zD|||D]4} t| d|krtd|t| d|d qWntyYqYq0|EdHqq|VdS) aSplit a `line` into potentially many lines. They should fit in the allotted `line_length` but might not be able to. `inner` signifies that there were a pair of brackets somewhere around the current `line`, possibly transitively. This means we can fallback to splitting by delimiters if the LHS/RHS don't yield any results. `features` are syntactical features that may be used in the output. NrN)rqline_strrMr<rGc3s\t|D]8}tt|||d}t|ddr |EdHdSq t|d|dEdHdS)N)omitrrqrBr;)generate_trailers_to_omitrxright_hand_splitis_line_short_enough)rMr<rrHrr=r>rhsj s  zsplit_line..rhsz+Split function returned an unchanged resultT)rqrr<)rrur4rrrrrrleft_hand_splitr1rr2rrdelimiter_splitstandalone_comment_splitr?rrG) rMrqrr<rZ split_funcsr split_funcr rr=rr>rGD sL       rGrc csg}g}g}|}d}|jD]P}||urF|jtvrF|j|urF|rB|n|}||||ur|jtvr|}|}q|sxtdt|||}t|||dd} t|||} t|| | || | fD]} | r| VqdS)a/Split line into many lines, starting with the first matching bracket pair. Note: this usually looks weird, only use this for function definitions. Prefer RHS otherwise. This is why this function is not symmetrical with :func:`right_hand_split` which also handles optional parentheses. NNo brackets foundTis_body) rrrrr rr?bracket_split_build_line bracket_split_succeeded_or_raise) rMr< tail_leaves body_leaves head_leavescurrent_leavesZmatching_bracketrheadbodytailr r=r=r>r s6       r)rMrqr<rrGccsg}g}g}|}d}d} t|jD]V} ||urB| |urB|r>|n|}|| ||ur"| jtvr"t| |vr"| j}| } |}q"|r| r|std|||t |||} t |||dd} t |||} t | | | | j s|jt j kr|js| jt jkr| js|js| dst| |rt| h|}zt||||dEdHWdStyt| st| |dstdn| s| rtd Yn0t|t| | | | fD]}|r|VqdS) aXSplit line into many lines, starting with the last matching bracket pair. If the split was by optional parentheses, attempt splitting without them, too. `omit` is a collection of closing bracket IDs that shouldn't be considered for this split. Note: running this function modifies `bracket_depth` on the leaves of `line`. NrTrr)r<rrzr s|            r)rrrrGcCsBtt|}|s>|dkr&tdn|dkr>td|ddS)aRaise :exc:`CannotSplit` if the last left- or right-hand split failed. Do nothing otherwise. A left- or right-hand split is based on a pair of brackets. Content before (and including) the opening bracket is left on one line, content inside the brackets is put on a separate line, and finally content starting with and following the closing bracket is put on a separate line. Those are called `head`, `body`, and `tail`, respectively. If the split produced the same line (all content in `head`) or ended up with an empty `body` and the `tail` is just the closing bracket, then it's considered failed. rz)Splitting brackets produced the same linerDz,Splitting brackets on an empty body to save z characters is not worth itN)rrur4r?)rrrZtail_lenr=r=r>r s rr)roriginalrrrGc Cst|jd}|rd|_|jd7_|rt|ddd|joZ|jdkoZtdd|D }|jsf|rtt |dd d D]@}||j t krqz||j t j kr||dtt j d qqz|D]0}|j|dd ||D]}|j|dd qq|rt|||_|S) zReturn a new line with given `leaves` and respective comments from `original`. If `is_body` is True, the result line is one-indented inside brackets and as such has its first leaf's prefix normalized and a trailing comma added when expected. rTrBrrrcss|]}|jtjkVqdSrH)rr+rrr=r=r>ryA r}z+bracket_split_build_line..rrtr)r1rrrrrvrrrErrr{r+rinsertr&r rr) rrrrr Z no_commasir comment_afterr=r=r>r- s2   r)rrGcs,tdtttttdfdd }|S)zNormalize prefix of the first leaf in every line returned by `split_func`. This is a decorator over relevant split functions. r=rc3s,||D]}t|jddd|Vq dS)NrTr)rr)rMr<rrr=r> split_wrapper] sz0dont_increase_indentation..split_wrapper)r=)r r1rr2r)rrr=rr>dont_increase_indentationW s"rc #szjd}Wnty(tdYn0j}z|jt|hd}Wnty`tdYn0|tkr||dkrtdt j j dt j }d}ttt d fd d }jD]}||Ed H|D]} || Ed Hqt||j}|j|krLt|tjhd r&|o"tj|v}n&t|tjtjhd rL|oJtj|v}|jt|} | |krVt j j dqr|r|tkrʈjdjtj krʈjdjt!krʈ"ttj dVd S)zSplit according to delimiters of the highest priority. If the appropriate Features are given, the split will add trailing commas also in function signatures and calls that contain `*` and `**`. rz Line emptyrzNo delimiters foundrBz7Splitting a single attribute from its owner looks wrongrrTrc3sJzj|ddWn2tyDVtjjd|Yn0dSzEAppend `leaf` to current line or to new line if appending impossible.TrrNrrr1rrr rrKrMr=r>append_to_line s  z'delimiter_split..append_to_lineNrKrt)#rr[r?rrrrrrrr1rrrr r&rrrrrWr\rUr2rirQrRrhrrrmrr+rr{r ) rMr<rbtZdelimiter_priorityZ lowest_depthZtrailing_comma_saferrrZ leaf_priorityr=rr>rf sV     rc#sdstdtjjdtttdfdd }jD]0}||EdH|D]}||EdHq^qBr~VdS)z4Split standalone comments from the rest of the line.rz*Line does not have any standalone commentsrrc3sJzj|ddWn2tyDVtjjd|Yn0dSrrrrr=r>r s  z0standalone_comment_split..append_to_lineN) rr?r1rrr&rrr)rMr<rrrr=rr>r s  rcCsP|j}|j}|j}t|tjkoL|dkr6|r6|jtjkpL|dkoL|oL|jtjkS)z9Return True if the given leaf starts an import statement.rfrom) rrrvrPr+rr\Z import_namere)rrrfr{r=r=r>r s rr)rrrGcCs(|j}|j}|tjthvo&|d|S)zdReturn True if the given leaf is a special comment. Only returns true for type comments for now.z# type:)rrvr+rr{r)rrrfr{r=r=r>r sr)rrrGcCsV|sL|jd}d|dvrL|dd}t|dkr>|d8}d||_dSd|_dS) zLeave existing extra newlines if not `inside_brackets`. Remove everything else. Note: don't use backslashes for formatting or you'll lose your voting rights. rrrrrNrBNr)rvrrr)rrZsplZnl_countr=r=r>r s    r)rr7rGcCsdtd|jtj}|dus*Jd|j|d}|}|rL|dd}||d|_dS)zMake all string prefixes lowercase. If remove_u_prefix is given, also removes any u prefix from the string. Note: Mutates its argument. z^([furbFURB]*)(.*)$Nzfailed to match string rBurrC)rmatchrvDOTALLgrouprr)rr7rZ orig_prefixZ new_prefixr=r=r>r s  rcCs|jd}|dddkr dS|dddkr:d}d}n|ddkrPd}d}nd}d}|j|}|d krpdS|jd|}td |}td |}td |}|j|t|t| } d |vr|| rdS| } nZt|d || } | | kr | } ||| ||_t|d || } t|d|| } d|vrt d| tj } | D]} dt | vrfdSqf|dkr| d ddkr| dd d} | d} | d}|| krdS|| kr|dkrdS||| ||_dS)zPrefer double quotes but only if it doesn't cause more escaping. Adds or removes backslashes as appropriate. Doesn't parse and fix strings nested in f-strings (yet). Note: Mutates its argument. furbFURBNrDrrr"'rz(([^\\]|^)(\\\\)*)z([^\\]|^)\\((?:\\\\)*)rz\1\2z\1\\r.z (?:[^{]|^)\{ # start of the string or a non-{ followed by a single { ([^{].*?) # contents of the brackets except if begins with {{ \}(?:[^}]|$) # A } followed by end of the string or a non-} rz\") rvr>findrcompilercasefoldsearch sub_twicefindallVERBOSErur)rrvZ orig_quoteZ new_quoteZfirst_quote_posrvZunescaped_new_quoteZescaped_new_quoteZescaped_orig_quoterZnew_bodymatchesmZorig_escape_countZnew_escape_countr=r=r>r s^           rcCs|j}|drn|drL|dd|dd}}||}nd|vr|d\}}d}|dr|dd}d}n|d r|dd}t|}|d||}nD|d r|dd }|d }|d krd }t||}nt|}||_dS)zNormalizes numeric (float, int, and complex) literals. All letters used in the representation are normalized to lowercase (except in Python 2 long literals). )Z0oZ0b0xNrCrrrrB+)jrrrL)rvrrrrformat_float_or_int_stringr)rtextrLrJsignZnumberrr=r=r>rO s0         r)rrGcCs0d|vr |S|d\}}|p dd|p*dS)z"Formats a float string like "1.0".rr)r)rrLrJr=r=r>rs sr)rbr$rGcCs\t|jddD]}|jtvrdSqd}tt|jD]\}}|dkrbt|trb|j t j krbd}|r@t |rrq6|j t j krt||drt||ddnt|rt||ddn|j t jkr|j tjkrd|_d|jd _n4|j tjkr||ttjd|ttjdqXn$t|tr2t|s@t||ddt|toT|j|v}q6dS) aEMake existing optional parentheses invisible or create new ones. `parens_after` is a set of string leaf values immediately after which parens should be put. Standardizes on visible parentheses for single-element tuples, and keeps existing visible parentheses for other tuples and generator expressions. Fr}NrTrjvisiblerr)rrvrvFMT_OFFrrxrlrr%rr\testlist_star_expris_walrus_assignmentr5#maybe_make_parens_invisible_in_atomwrap_in_parentheses is_one_tuplerer+rSTARr4r&Z append_childrr)rbr$rZ check_lparr:rnr=r=r>r%| s>      r%cCsd}|rt|}qdS)zJConvert content between `# fmt: off`/`# fmt: on` into standalone comments.TN)convert_one_fmt_off_pair)rbZ try_againr=r=r>rA srAc Cs@|D]0}d}t|jddD]}|jtvr2|jtkrTt|}|rT|jtvrTq t t |}|sfq |d}|j }|j}||j d|_|jdd dd|D} | dr| dd } d} |D]} | } | dur| } q|dusJd | dusJd || tt| |d|d|jd d S|j }q qdS)zConvert content of a single `# fmt: off`/`# fmt: on` into a standalone comment. Returns True if a pair was converted. rFr}NrNrcss|]}t|VqdSrHrcrr=r=r>ry r}z+convert_one_fmt_off_pair..rz(INTERNAL ERROR: fmt: on/off handling (1)z(INTERNAL ERROR: fmt: on/off handling (2)rT)rrrvrvrrr{rOrrxgenerate_ignored_nodesrrrrr3r4r&r) rbrZprevious_consumedrrgZ ignored_nodesrrrvZ hidden_valueZ first_idxZignoredr:r=r=r>r sJ      rccslt|}|durh|jtjkrhd}t|jddD]"}|jtvrBd}q.|jtvr.d}q.|rZdS|V|j }qdS)zvStarting from the container of `leaf`, generate all leaves until `# fmt: on`. Stops at the end of the block. NFr}T) rlrr+rrrvrvFMT_ONrr)rrkZ is_fmt_onrr=r=r>r s  r)rbrrGcCs|jtjksr s0       rcCspt|ts|jtjkrdS|jd|jd}}t|ton|jtjkon|jdkont|ton|jtj kon|jdkS)zGiven a `LN`, determines whether it's an atom `node` with invisible parens. Useful in dedupe-ing and normalizing parens. Frrr) rr&rr\r5rlr+rrvr)rbrrr=r=r>r% s   rcCs>|jtjkor7 s  rcCs>t|jdkrdS|j\}}}|jtjkr6|jtjks:dS|S)zqReturns `wrapped` if `node` is of the shape ( wrapped ). Parenthesis can be optional. Returns None otherwiserDN)rrlrr+rr)rbr8wrappedr9r=r=r>unwrap_singleton_parenthesisA s  rr)rrnrrGcCshttj|rdnd}ttj|r"dnd}|j}d|_|p>d}ttj|||g}||_| ||dS)zWrap `child` in parentheses. This replaces `child` with an atom holding the parentheses and the old child. That requires moving the prefix. If `visible` is False, the leaves will be valueless (and thus invisible). rrrrN) r&r+rrrvr3r%r\r5r4)rrnrr8r9rvr: new_childr=r=r>rO s rcCsv|jtjkrLt|}|dus(|jtjkr,dSt|jdkoJ|jdjtjkS|jt vott|jdkot|jdjtjkS)zMReturn True if `node` holds a tuple with one element, with or without parens.NFrCrB) rr\r5r testlist_gexprrlr+rIMPLICIT_TUPLE)rbZgexpr=r=r>ra s    rcCst|}|duo|jtjkS)z7Return True iff `node` is of the shape ( test := test )N)rrr\Znamedexpr_test)rbrr=r=r>rq srcCs||jtjkrdS|jtjkr*|jdkr*dS|jtjkr:dSt|jdkrLdS|j\}}}|jtj krx|jtj krxt |SdS)zAReturn True if `node` holds a `yield` or `yield from` expression.TyieldFrD) rr\Z yield_exprr+rrvr5rrlrrr)rbr8exprr9r=r=r>rw s   r)rrLrGcCs@|jtvs|jsdS|j}|jtjkr6|js0dS|j}|j|vS)aFReturn True if `leaf` is a star or double star in a vararg or kwarg. If `within` includes VARARGS_PARENTS, this applies to function signatures. If `within` includes UNPACKING_PARENTS, it applies to right hand-side extended iterable unpacking (PEP 3132) and additional unpacking generalizations (PEP 448). F)rrVrr\rd)rrLrr=r=r>rW s rWcCs$|jd}|dddvo"d|vS)zKReturn True if `leaf` is a multiline string that actually spans many lines.rNrD>rrrN)rvr>)rrvr=r=r>r s rcCsVt|jdksD|jdjtjksD|jdjtjksD|jdjtjkrHdSt|jdS)z2Return True if `node` is a suite with a stub body.rTrrBrDFrC)rrlrr+NEWLINEINDENTDEDENTr*rr=r=r>r' s r'cCsdt|tr|jtjkrdSt|jdkr,dS|jd}|jtjkobt|jdkobtdd|jDS)zCReturn True if `node` is a simple statement containing an ellipsis.FrCrrDcss|]}|ttjdkVqdS)rNr)rxrr=r=r>ry r}zis_stub_body..) rr%rr\ simple_stmtrrlr5r)rbrnr=r=r>r* s   r*cCs|jtjkrdS|jd}|jd}|jtjkr<|jtjks@dSt}|jddD]2}t|t rn| |qT| D]}| |qvqTz | WSt yYdS0dS)zReturn maximum delimiter priority inside `node`. This is specific to atoms with contents contained in a pair of parentheses. If `node` isn't an atom or there are no enclosing parentheses, returns 0. rrrBN)rr\r5rlr+rrrrr&rrrr)rbrrrrrr=r=r>r s         rcCs*|jtjkrd|_n|jtjkr&d|_dS)zMake sure parentheses are visible. They could be invisible as part of some statements (see :func:`normalize_invisible_parens` and :func:`visit_import_from`). rrN)rr+rrvrrr=r=r>r s  r)rMrrGc Cs|jr$|jjtjtjhvr$|jdvs(dSz8|jd}|jtjkrJt |hnt }|j j |d}Wnt tfyxYdS0|tkS)zFShould `line` immediately be split with `delimiter_split()` after RHS?z[{(Frr)rrr\r5rervrr+rrrrrr[rrm)rMrrr max_priorityr=r=r>r s rcCsVt}|D]@}|jtjkrD|jdd}|dvrB|tjq|jtj krhd|jvrf|tj q|jtj kr|j r|j jt jt jhvr|tjq|jtjkr|tjq|jt jt jhvr|jr|jdjtjkr|jt jkrtj}ntj}|jD]N}|jtvr|||jt jkr|jD]}|jtvr.||q.qq|S)a&Return a set of (relatively) new Python features used in this file. Currently looking for: - f-strings; - underscores in numeric literals; - trailing commas after * or ** in function signatures and calls; - positional only arguments in function signatures and lambdas; NrC>zf"ZRFzF'frzf'ZFRZrfzF"rr)rrrr+rrvrr2rfrrgSLASHrr\rUrQrm COLONEQUALrlrlrrirhSTARSrR)rbr<rZ value_headrZchZargchr=r=r>get_features_used s>            rcst|fddtDS)z5Detect the version to target based on the nodes used.csh|]}t|kr|qSr=rrwr<r=r>r:6 sz)detect_target_versions..)rrSrr=rr>r@3 s r@)rMrqrGc cs t}|Vd|j}d}d}t}t|ddD]\}}} || 7}||krPq| t|jt|jk} |jtksv| r|q|r||urd}n|jtvr| t |q0|jtvr0|dkr|j |djt vr| t |q0|r | t || |||V|jr0|j}|}q0dS)a(Generate sets of closing bracket IDs that should be omitted in a RHS. Brackets can be omitted if the entire trailer up to and including a preceding closing bracket fits in one line. Yielded sets are cumulative (contain results of previous yields, too). First set is empty. rTNT)rrrB)rrenumerate_with_lengthrrvrvrr{rrrrrrclearr) rMrqrlengthrrZinner_bracketsr:r leaf_lengthZhas_inline_commentr=r=r>r; s<     rcst}ttttddfdfdd |jD]}|jtjkrBq|jd}t |t rt |jdkr|jt j kr|jdjt jkrq.qq.|jtjkr|jd}t |t r|jdkrq|t|jd dO}q.qq.|S) z/Return a set of __future__ imports in the file.N)rlrGc3s|D]}t|tr(|jtjkr|jVq|jtjkrn|jd}t|tsPJd|jtjksdJd|jVq|jtj kr|jEdHqt dqdS)NrzInvalid syntax parsing imports) rr&rr+rrvr\Zimport_as_namerlZimport_as_namesAssertionError)rlrnZ orig_nameget_imports_from_childrenr=r>rq s       z5get_future_imports..get_imports_from_childrenrrCrBZ __future__rD)rrrorrurlrr\rrr&rr+rrrerv)rbZimportsrnZ first_childZ module_namer=rr>r?m s,"        r?)rrGcCsP|d}g}|rD|}|}Wdn1s:0Ytd|S)z9 Return a PathSpec matching gitignore content if present.z .gitignoreNZ gitwildmatch)rr readlinesr$Z from_lines)r gitignorerHZgfr=r=r>r s  &r)rrrrrrrGc csN|sJd||D](}||r@||dqzd||}Wnrty}z*||d|WYd}~qWYd}~n:d}~0ty| r||d|YqYn0| r|d7}| |} | r| dr||dq| r,t ||||||EdHq|r| |} | r|VqdS) aGenerate all files under `path` whose paths are not excluded by the `exclude` regex, but are included by the `include` regex. Symbolic links pointing outside of the `root` directory are ignored. `report` is where output about exclusions goes. z/INTERNAL ERROR: `root` must be absolute but is z#matches the .gitignore file contentrtzcannot be read because Nz'is a symbolic link that points outside rz(matches the --exclude regular expression) is_absoluteiterdirZ match_fileas_posix path_ignoredr relative_torr is_symlinkrrrrr) rrrrrrrnZnormalized_pathrZ exclude_matchZ include_matchr=r=r>r s>         r)srcsrGcCs~|stdStdd|D}|r2|d}|jD]@}|drP|S|drd|S|dr8|Sq8|S)a)Return a directory containing .git, .hg, or pyproject.toml. That directory can be one of the directories passed in `srcs` or their common parent. If no directory in the tree contains a marker that would specify it's the project root, the root of the file system is returned. rtcss|]}t|VqdSrH)r rrr=r=r>ry r}z$find_project_root..z fake-filez.gitz.hgr)r rrrparentsr)rZ common_baseZ directoryr=r=r>r s       rc@seZdZUdZdZeed<dZeed<dZeed<dZ e ed<dZ e ed<dZ e ed <e ed d d d Ze ed dddZe ed dddZee dddZedddZd S)rzDProvides a reformatting counter. Can be rendered with `str(report)`.FrErrr change_count same_count failure_countN)rrrGcCs|tjurF|jrdnd}|js$|js6t|d||jd7_n@|jrx|tjurb|d}n |d}t|dd|jd7_d S) zGIncrement the counter for successful reformatting. Write out a message.zwould reformat reformattedrrrBz" already well formatted, good job.z( wasn't modified on disk since last run.Fr3N) rQrKrErrrrrNr)r\rrr msgr=r=r>r s      z Report.done)rmessagerGcCs&td|d||jd7_dS)zCIncrement the counter for failed reformatting. Write out a message.zerror: cannot format rXrBN)r4r)r\rr r=r=r>rsz Report.failed)rr rGcCs |jrt|d|dddS)Nz ignored: Fr )rr)r\rr r=r=r>r!szReport.path_ignoredrYcCs|jr dS|jr|jrdSdS)aReturn the exit code that the app should use. This considers the current state of changed files and failures: - if there were any failures, return 123; - if any files were changed and --check is being used, return 1; - otherwise return 0. {rBr)rrrEr[r=r=r>r%s  zReport.return_codecCs|jrd}d}d}n d}d}d}g}|jrb|jdkr8dnd }|tj|jd |d |d d |jr|jdkrvdnd }||jd |d ||jr|jdkrdnd }|tj|jd |d |ddd|dS)zcRender a color report of the current state. Use `click.unstyle` to remove colors. zwould be reformattedzwould be left unchangedzwould fail to reformatr zleft unchangedzfailed to reformatrBrrz filerrTr r5rsz, r)rErr rZstylerrr)r\r Z unchangedrrrr=r=r>r8s,zReport.__str__)r9r:r;r<rErPrrrrrrrr rQrrurrr rrr=r=r=r>rs       c Csd}tjdkrZttjdddD]6}ztj||d|fdWStyTYq Yq 0q n8dD]2}ztj|||dWStyYq^Yq^0q^t|S) Nz rDrXrBrTrrD)feature_version)rWrV)r version_inforEastparse SyntaxErrorr"r#)rr minor_versionrr=r=r> parse_astVs    rcCstt|tjtjtjtjtjfr,tj|jdSt|tjtjtjfrPtj|j dSt|tj tj frptj|j dS|S)z,Map ast nodes deprecated in 3.8 to Constant.rz) rrZStrr"r#ZBytesZConstantrZNumrZ NameConstantrvrr=r=r>_fixup_ast_constantsisr)rr3rGc sdttjtjtjftttdfdd z t|}Wn2t yj}zt d|WYd}~n d}~00z t|}WnRt y}z:t d t |j|}t d|d |dWYd}~n d}~00d |}d |}||krt t||d d }t d |ddS)z:Raise AssertionError if `src` and `dst` aren't equivalent.r)rbrrGc 3st|}d||jjdVt|jD]`}tjtjf}tj dkrR|t jf7}t ||rbqzt ||}Wnt yYq*Yn0d|d|dVt |tr>|D]}|dkrt |t jtjtjfrt |t jtjtjfr|jD]}||dEdHqqt |t jtjtjfr||dEdHqq*t |t jtjtjfrl||dEdHq*d|d|d |jjVq*d|d |jjVdS) z=Simple visitor generating strings to compare ASTs by content.rJrrrBr~targetsrCNz, # z) # /)rrr9r_fieldsr"Z TypeIgnorer#rrrrriAttributeErrorrxZDeletereltsAST)rbrZfieldZtype_ignore_classesrvitem_vr=r>r|s:        "zassert_equivalent.._vzScannot use --safe with this file; failed to parse source file. AST error message: Nrz-INTERNAL ERROR: Black produced invalid code: ze. Please report a bug on https://github.com/psf/black/issues. This invalid output might be helpful: rNrr3zINTERNAL ERROR: Black produced code that is not equivalent to the source. Please report a bug on https://github.com/psf/black/issues. This diff might be helpful: )r)rrrr"r#rrrurrr dump_to_filer traceback format_tb __traceback__rF)rr3Zsrc_astrZdst_astlogZ src_ast_strZ dst_ast_strr=rr>r5ys>,-   r5)rr3rrGcCsFt||d}||krBtt||ddt||dd}td|ddS)zDRaise AssertionError if `dst` reformats differently the second time.rsourcez first passz second passzINTERNAL ERROR: Black produced different code on the second pass of the formatter. Please report a bug on https://github.com/psf/black/issues. This diff might be helpful: N)rrrFr)rr3rZnewdstr#r=r=r>r6s   r6)outputrGcGshtjdddddd>}|D](}|||r|ddkr|dqWd n1sX0Y|jS) z;Dump `output` to a temporary file. Return path to the file.rZblk_z.logFutf8)rrvrdeleterrrNN)tempfileNamedTemporaryFiler"r)r%r.rHr=r=r>rs  *rrYccs dVdS)zWReturn an empty context manager. To be used like `nullcontext` in Python 3.7. Nr=r=r=r=r>r$sr$)aba_nameb_namerGc CsJddl}dd|dD}dd|dD}d|j||||dd S) z9Return a unified diff string between strings `a` and `b`.rNcSsg|] }|dqSrNr=rxrMr=r=r>rr}zdiff..rNcSsg|] }|dqSr.r=r/r=r=r>rr}rrU)ZfromfileZtofiler)difflibrrZ unified_diff)r*r+r,r-r0Za_linesZb_linesr=r=r>rFs rF)rrGcCstd|D] }|q dS)zFasyncio signal handler that cancels all `tasks` and reports to stderr.zAborted!N)r4r)rrr=r=r>rsr)rrGc Csztjdddkrtj}ntjj}dd||D}|s^Wtd}|tj| dS|D] }| qb| tj ||ddWtd}|tj| n td}|tj| 0dS) zFCancel all pending tasks on `loop`, wait for them, and close the loop.NrC)rDrWcSsg|]}|s|qSr=)r)rxrr=r=r>rr}zshutdown..zconcurrent.futuresTr) rrr all_tasksTaskloggingZ getLoggerZsetLevelZCRITICALcloserrr )rr1Z to_cancelZ cf_loggerrr=r=r>rs,       r)regex replacementrrGcCs|||||S)zReplace `regex` with `replacement` twice on `original`. This is used by string normalization to perform replaces on overlapping matches. )sub)r5r6rr=r=r>rsr)r5rGcCsd|vrd|}t|}|S)zdCompile a regular expression string in `regex`. If it contains newlines, use verbose mode. rNz(?x))rr)r5Zcompiledr=r=r>r!s r)sequencerGccs0t|d}t|D]}||fV|d8}qdS)z;Like `reversed(enumerate(sequence))` if that were possible.rBN)rr)r8r:elementr=r=r>enumerate_reversed,s   r:)rMrrGccsttttgttttff|r&tnt}||j D]V\}}t |j t |j }d|j vrbdS| |D]}|t |j 7}ql|||fVq6dS)z|Return an enumeration of leaves with their length. Stops prematurely on multiline strings and standalone comments. rNN)rrrr&rrIndexr:rrrrvrvr)rMropr:rrrr=r=r>r4s  r)r)rMrqrrGcCs0|st|d}t||ko.d|vo.| S)zReturn True if `line` is no longer than `line_length`. Uses the provided `line_str` rendering, if any, otherwise computes a new one. rN)rur4rr)rMrqrr=r=r>rJs r)rMrGcCs|j}t|dkrdS|djtjkr|djtjkrd}d}|d}|dddD]}|jtvr||jtvrrdS|d7}nP|jtjkr|d7}n:|jtjkr|jtjks|jtvsdSn|jtvrdS|dkrT|dkrTdSqTdS) zReturn False if the line cannot be split *for sure*. This is not an exhaustive search but a cheap heuristic that we can use to avoid some unfortunate formattings (mostly around wrapping unsplittable code in unnecessary parentheses). rCFrrBrrONT) rrrr+rrrrr)rMrZ call_count dot_countrrr=r=r>rXs,         rcCs|j}|jsdS|}||dkr*dS|tkr6dSt|jdksLJd|jd}|jd}|jd}|jd}|jtvr|jt vrd}d |j } t |D]H\} } } | jt vr| j |urd}|r| | 7} | |krq| jtvrd}qt|j| dkrdS|jt jks@|jt jks@|jt jkr|jr|jjtjkr|jtvrPdSt|r^dSd |j } d} t |D]H\} } } | | 7} | |j ur| s| |krdSn| jtvrtd} qtdS) aDoes `line` have a shape safe to reformat without optional parens around it? Returns True for only a subset of potentially nice looking formattings but the point is to not return false positives that end up producing lines that are too long. TrBFrCzStranded delimiterrrOrrT)rrrrrrrrrrrrrrr+rRBRACErrr\r^r)rMrqrrrsecondZ penultimaterZ remainderrZ_indexrrZseen_other_bracketsr=r=r>r|sf               r)rrGcCstd|dS)Nzcache.z.pickle) CACHE_DIRrrr=r=r>get_cache_filesrAc Cs|t|}|siS|dJ}zt|}Wn*tjtfyXiYWdS0Wdn1sn0Y|S)zRead the cache if it exists and is well formed. If it is not well formed, the call to write_cache later should resolve the issue. rN)rArrpicklerZUnpicklingErrorr)r cache_fileZfobjrr=r=r>rs 6r)rrGcCs|}|j|jfS)zKReturn the information used to check if a file is already formatted or not.)rrst_size)rrr=r=r>rsr)rrrGcCsPtt}}|D]4}|}||t|kr<||q||q||fS)zSplit an iterable of paths in `sources` into two sets. The first contains paths of files that modified on disk or are not in the cache. The other contains paths to non-modified files. )rrrrr)rrZtodorrr=r=r>rs  r)rrrrGcCst|}zztjdddi|dd|D}tjt|jdd }tj||ddWd n1sh0Yt |j |Wnt yYn0d S) zUpdate the cache file.T)rexist_okcSsi|]}|t|qSr=)rrrr=r=r>rr}zwrite_cache..F)dirr'rT)ZprotocolN) rAr@mkdirr(r)rurrBdumprrrr)rrrrCZ new_cacher.r=r=r>rs. rcCsXzddlm}ddlm}Wnty0YdS0||fD]}t|dr:dd|_q:dS)aMake Click not crash. On certain misconfigured environments, Python 3 selects the ASCII encoding as the default which restricts paths that it can access during the lifetime of the application. Click refuses to work in this scenario by raising a RuntimeError. In case of Black the likelihood that non-ASCII characters are going to be used in file paths is minimal since it's Python source code. Moreover, this crash was spurious on Python 3.7 thanks to PEP 538 and PEP 540. r)core) _unicodefunN_verify_python3_envcSsdSrHr=r=r=r=r>r|!r}zpatch_click..)rrIrJModuleNotFoundErrorhasattrrK)rIrJmoduler=r=r> patch_clicks     rOcCstttdSrH)r rOrr=r=r=r> patched_main$srP__main__)r=)N)N)Fr=)r=)r=r=)r=)r=)r)F)F(Xrrconcurrent.futuresrr contextlibrrenumr functoolsrrr r%rr3Zmultiprocessingr r rpathlibr rBr5rrrr(rQr typingr rrrrrrrrrrrrrrrrZappdirsrattrrr r!rrZ typed_astr"r#Zpathspecr$Zblib2to3.pytreer%r&r'Zblib2to3r(r)Zblib2to3.pgen2r*r+Zblib2to3.pgen2.grammarr,Zblib2to3.pgen2.parser-Z_black_versionr. __version__rZDEFAULT_EXCLUDESZDEFAULT_INCLUDESr@ruZ FileContentZEncodingZNewLinerrrrrr;roZ SplitFuncfloatZ TimestampZFileSize CacheInfoZCacherrr4Z initializeZpython_symbolsr\ UserWarningr7rr?rr@rArQrSrarbrcrr2rZrjr^rer_r`rhrfrgrirkrlrmrnrrorPrZContextZ ParameterrZcommandroptionZChoiceZversion_optionrRZ pass_contextrrrrZAbstractEventLooprrNrrr!rbytesrrUr=rdrerfrprrrrZif_stmtZ while_stmtZfor_stmtZtry_stmtZ except_clauseZ with_stmtZfuncdefrbr)r{rhr{LESSGREATEREQEQUALNOTEQUAL LESSEQUAL GREATEREQUALrtVBAR CIRCUMFLEXAMPER LEFTSHIFTr\PLUSMINUSrr DOUBLESLASHPERCENTrTILDEr2r[rrVrQr^rUrTrXr5rcrrrrYrxZlambdefZor_testZand_testZnot_testrzrdrZxor_exprZand_exprr]Z arith_exprZtermr1rrFrwrmryr|rvrursrrrr1rCrBZtestlistZexprlistrrrr_rLBRACEr>rrrrrrrrrMrrOrrlrrrrrrrrrGrrrrrrrrrrrrrrr%rArrrrrrrrrrrWrr'r*rrrrr@rr?rrrrrrrr5r6rr$rFr2rrrrr:rrrrrArrrrrOrPr9r=r=r=r>s     L           & $          0a  "  K / " )'#     tHg`  $R 'Y *"EL$ 36$ $  1"22 9T,N    ( $$Y.