o DDif @sUddlZddlZddlmZddlmZddlmZddl m Z m Z m Z m Z mZmZmZddlmZddlmZmZddlmZdd lmZmZmZdd lmZdd lmZej Z!ee"d <ej#Z$ee"d <ej%Z&ee"d<ej'Z(ee"d<dZ)dZ*e+dej,Z-e e.e"d<eddGdddZ/dee.e0fde.fddZ1de.de.fddZ2de ede.fd d!Z3de.de4fd"d#Z5de ede e.fd$d%Z6dee.e0fd&edee.e.ffd'd(Z7dee.e0fd&ed)e4d*e4de/f d+d,Z8dS)-N) dataclass)BytesIO)detect_encoding) FrozenSetIterableIteratorPatternSetTupleUnion) NEWLINE_RE)PythonTokenTypes TokenType) split_lines) AutoConfig ParserConfigPartialParserConfig)Token)tokenize_lines_INDENT_NAME_NEWLINE_STRING z z \\(\r\n?|\n)_CONTINUATION_RET)frozenc@s"eZdZUeed<eeed<dS)ConfigDetectionResultconfigtokensN)__name__ __module__ __qualname__r__annotations__rrr#r#U/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/_parser/detect_config.pyrs rsourcereturncCs t|trdStt|jdS)z Detects the encoding from the presence of a UTF-8 BOM or an encoding cookie as specified in PEP 263. If given a string (instead of bytes) the encoding is assumed to be utf-8. zutf-8r) isinstancestrpy_tokenize_detect_encodingrreadline)r%r#r#r$_detect_encoding's r+ source_strcCs t|}|dur|dStS)zN Finds the first newline, and uses that value as the default newline. Nr)r searchgroup_FALLBACK_DEFAULT_NEWLINE)r,matchr#r#r$_detect_default_newline4s r1rcCsHz tdd|D}Wn tytYSw|j}|dus"Jd|S)zY Finds the first INDENT token, and uses that as the value of the default indent. css|] }|jtur|VqdS)N)typer).0tr#r#r$ Csz!_detect_indent..Nz,INDENT tokens must contain a relative_indent)next StopIteration_FALLBACK_DEFAULT_INDENTrelative_indent)r first_indentfirst_indent_strr#r#r$_detect_indent>s r<cCsDt|dks t|dsdSt|ddp t|dd S)NrF)lenr fullmatchr)r,r#r#r$_detect_trailing_newlineKs rBcCst}d}|D]t}|dkr|jttfvrq|dkr!|jdkr!d}q|dkr-|jdkr-d}q|dkr9|jdkr9d}q|dkrE|jdkrEd }q|dkrU|jtkrU||jq|d kra|jtkrad}q|dkrk|jd vrkq|dkrw|jtkrwd}qt|St|S) zs Finds __future__ imports in their proper locations. See `https://www.python.org/dev/peps/pep-0236/`_ rfrom __future__importasz(),)setr2rrstringradd frozenset)rfuture_importsstatetokr#r#r$_detect_future_importsUs.rRpartialcCs<|j}t|tr t|n|}t|tr|n||}||fS)zA Returns an (original encoding, converted source) tuple. )encodingr'rr+r(decode)r%rSpartial_encodingrTr,r#r#r$convert_to_utf8us rWdetect_trailing_newlinedetect_default_newlinec Cs|j}t||d\}}|j}t|tr|rt|ntn|}|o"t|} |r+| s+||7}t|dd} t || |} |j } t| trKt | \} } t | }n| }|j}t|trat | \} } t| }n|}tt| |||| ||d| dS)zf Computes a ParserConfig given the current source code to be parsed and a partial config. )rST)keepends)linesrTdefault_indentdefault_newlinehas_trailing_newlineversionrO)rr)parsed_python_versionrWr]r'rr1r/rBrrr\ itertoolsteer<rOrRrr)r%rSrXrYpython_versionrTr,partial_default_newliner]r^r[rpartial_default_indent tokens_dupr\partial_future_importsrOr#r#r$ detect_configsN         rh)9rare dataclassesriortokenizerr)typingrrrrr r r libcst._nodes.whitespacer !libcst._parser.parso.python.tokenr rlibcst._parser.parso.utilsrlibcst._parser.types.configrrrlibcst._parser.types.tokenrlibcst._parser.wrapped_tokenizerINDENTrr"NAMErNEWLINErSTRINGrr/r8compileUNICODErr(rbytesr+r1r<boolrBrRrWrhr#r#r#r$sX    $