a g g+@sUdZddlZddlZddlmZddlmZmZmZddl m Z ej dkrZddl m Z n ddlm Z ddlmZd Ze ed <ed ed ejZe ed <edZe ed<edejZe ed<eeeeedddZeedddZeeedddZeeedddZeedddZeddd d!Zeedd"d#Z ed$d%eeed&d'd(Z!eedd)d*Z"e dd+d,d-Z#ed.d%ee$d/d0d1Z%ee$d2d3d4Z&ee$e$d5d6d7Z'dS)8zN Simple formatting on strings. Further string formatting code is in trans.py. N) lru_cache)ListMatchPattern)Leaf))Final) WIDTH_TABLEZfurbFURBSTRING_PREFIX_CHARSz^([z]*)(.*)$STRING_PREFIX_REz \s*\t+\s*(\S)FIRST_NON_WHITESPACE_REz(?P\\+)(?P(u(?P[a-fA-F0-9]{4}))|(U(?P[a-fA-F0-9]{8}))|(x(?P[a-fA-F0-9]{2}))|(N\{(?P[a-zA-Z0-9 \-]{2,})\}))UNICODE_ESCAPE_RE)regex replacementoriginalreturncCs|||||S)zReplace `regex` with `replacement` twice on `original`. This is used by string normalization to perform replaces on overlapping matches. )sub)rrrrP/var/www/html/llm_bihealth/app/venv/lib/python3.9/site-packages/black/strings.py sub_twice#sr)stringrcCs|t}|dddvS)zS Returns: True iff @string starts with three quotation characters. Nr>"""''')lstripr )rZ raw_stringrrrhas_triple_quotes,s r)srcCs\g}|D]J}t|}|rL|d}||d|||dq ||q |S)zh Splits string into lines and expands only leading tabs (following the normal Python rules) N) splitlinesr matchstartappend expandtabs)rlineslinerZfirst_non_whitespace_idxrrr lines_with_leading_tabs_expanded5s     r%) docstringprefixrc Cs|sdSt|}tj}|ddD]&}|}|r"t|t|t|}q"|dg}|tjkrt|d}t|ddD]>\}}||d} | s||kr| || q~| dq~d |S)Nrr ) r%sysmaxsizerminlenstrip enumeraterstripr!join) r&r'r#indentr$strippedZtrimmedZ last_line_idxiZ stripped_linerrr fix_docstringKs"    r6cCs6t|d}d}||tvr2|||7}|d7}q|S)z Pre-conditions: * assert_is_leaf_string(@string) Returns: @string's prefix (e.g. '', 'r', 'f', or 'rf'). r(rr)assert_is_leaf_stringr )rr'Z prefix_idxrrrget_string_prefixcs   r8cCs|d}|d}d||fvr,t||}n t||}d|krRt|dksbnJ|d|ddvs|J|dt|d |ttsJt|d |d ttd d S) aH Checks the pre-condition that @string has the format that you would expect of `leaf.value` where `leaf` is some Leaf such that `leaf.type == token.STRING`. A more precise description of the pre-conditions that are checked are listed below. Pre-conditions: * @string starts with either ', ", ', or " where `set()` is some subset of `set(STRING_PREFIX_CHARS)`. * @string ends with a quote character (' or "). Raises: AssertionError(...) if the pre-conditions listed above are not satisfied. "'rrz0 is missing a starting quote character (' or ").)r:r9z/ is missing an ending quote character (' or ").Nz is NOT a subset of .)findmaxr-r.setissubsetr )rZ dquote_idxZ squote_idxZ quote_idxrrrr7vs&      r7cCst|}|dus Jd||d}|dddddd d d }t|d krxd |d krx|ddd}||d S)z#Make all string prefixes lowercase.Nzfailed to match string rFfBbUr(ur)rrr;)r rgroupreplacer.lower)rrZ orig_prefix new_prefixrrrnormalize_string_prefixs   rL@)r,)patternrcCs t|S)N)recompile)rNrrr_cached_compilesrQcCs|t}|dddkr|S|dddkr8d}d}n|ddkrNd}d}nd}d}||}|dkrl|S|d|}td |}td |}td |}||t|t| } d |vr|| r|S| } nXt|d || } | | kr| } ||| |}t|d || } t|d || } d|vrtt d| tj } | D]} dt | vrV|SqV|dkr| dddkr| ddd} | d} | d}|| kr|S|| kr|dkr|S||| |S)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. Nrrrrr9r:r;z(([^\\]|^)(\\\\)*)z([^\\]|^)\\((?:\\\\)*)rGz\1\2z\1\\rBz (?:(?.replace)rYr8rJrrWrOrr)r^textr'rIrrr"normalize_unicode_escape_sequencess  rfi)charrc Cst}t|}t|d}d}|d}||\}}}||krD|d}n"||krV|d}n|dkrbdS|S||krpq~||d}q$dS)aReturn the width of a single character as it would be displayed in a terminal or editor (which respects Unicode East Asian Width). Full width characters are counted as 2, while half width characters are counted as 1. Also control characters are counted as 0. rrr))r ordr.) rgtable codepointZhighestZlowestidxZstart_codepointZ end_codepointwidthrrr char_widths   rm)line_strrcCs|rt|Sttt|S)aReturn the width of `line_str` as it would be displayed in a terminal or editor (which respects Unicode East Asian Width). You could utilize this function to determine, for example, if a string is too wide to display in a terminal or editor. )isasciir.summaprm)rnrrr str_width7srr)rn max_widthrcCsBd}t|D],\}}t|}|||kr0|S||7}q t|S)zCount the number of characters in `line_str` that would fit in a terminal or editor of `max_width` (which respects Unicode East Asian Width). r)r0rmr.)rnrsZ total_widthr5rgrlrrrcount_chars_in_widthDs  rt)(__doc__rOr+ functoolsrtypingrrrZblib2to3.pytreer version_infoZtyping_extensionsr Zblack._width_tabler r __annotations__rPDOTALLr r rVrrWrboolrr%r6r8r7rLrQr]rfintrmrrrtrrrrsB          #J