o 7fLv@sxUdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddlmZmZmZmZmZmZmZmZddlmZz ddlZddlZdZWn ey`dZYnwed Ze jd krxee d d Z!d e"d<ne #d dZ$ee$j%d Z!Wdn1swYe&dZ'd e"d<dZ(d e"d<dZ)d e"d<dZ*d e"d<dZ+d e"d<dZ,d e"d<e-dZ.d e"d<ddd#d$Z/dd%d&Z0dd)d*Z1dd/d0Z2dd2d3Z3dd6d7Z4dd;d<Z5dd>d?Z6Gd@dAdAe7Z8ddCdDZ9ddIdJZ:ddPdQZ;ddSdTZd e"dX<dYZ?d e"dZ<d[Z@d e"d\<d]ZAd e"d^<ddfdgZBddidjZCGdkdldlZDddndoZEddtduZFiZGdve"dw<dd{d|ZH }ddddZIdddZJddddZKdddZLdddZMdddZNdddZOdddZPdddZQddddZRdddZSdddZTdddZUGdddZVdddZWdddZXdddZYe jZZ[dddZ\dddZ]dddZ^dS)z3Utility functions with no non-trivial dependencies.) annotationsN) resources)IOCallable ContainerFinalIterableSequenceSizedTypeVar)LiteralTFT) mypytypeshedr TYPESHED_DIRzpy.typeds=([ \t\v]*#.*(\r\n?|\n))??[ \t\v]*#.*coding[:=][ \t]*([-\w.]+) ENCODING_REDEFAULT_SOURCE_OFFSETPDEFAULT_COLUMNS MINIMUM_WIDTH MINIMUM_WINDOWS_MAJOR_VT100iZ)MINIMUM_WINDOWS_BUILD_VT100)__init____new____call____init_subclass____class_getitem__SPECIAL_DUNDERSnamestrexclude_specialboolreturncCs$|r|tvrdS|do|dS)zReturns whether name is a dunder name. Args: exclude_special: Whether to return False for a couple special dunder methods. F__)r" startswithendswith)r#r%r+B/home/ertert/spirit/venv/lib/python3.10/site-packages/mypy/util.py is_dunder;s r-cCst| o|do|dSN_)r-r)r*r#r+r+r, is_sunderGsr1mod_name list[str]cCs4|g}d|vr|ddd}||d|vs|S)zReturn the module and all parent module names. So, if `mod_name` is 'a.b.c', this function will return ['a.b.c', 'a.b', and 'a']. .r)rsplitappend)r2outr+r+r,split_module_namesKs  r9modules Iterable[str]target str | NonecCst||}|dur dS|dSNr) split_target)r:r<resultr+r+r, module_prefixXs rAtuple[str, str] | NonecCsRg} ||vr|d|fS|dd}t|dkrdS|d}|d|dq)NTr4r5r)joinr6leninsert)r:r< remaining componentsr+r+r,r?_s  r?objobjectcCs,|durdStt|}|dddS)zReturn the last component of the type name of an object. If obj is None, return 'nil'. For example, if obj is 1, return 'int'. Nnilr4z'>)r$typesplitrstrip)rHtr+r+r, short_typeks rPtextbytestuple[str, int]cCsZt|}|r'|drdnd}|dd}|ds!|dkr#d}||fSd}|d fS) z*PEP-263 for detecting Python file encodingr5rascii)z iso-latin-1-zlatin-1-z iso-latin-1zlatin-1utf8rK)rmatchgroupdecoder))rQr@lineencodingdefault_encodingr+r+r,find_python_encodingvs r]bcCst|ddS)a2Converts bytes into some human-readable representation. Unprintable bytes such as the nul byte are escaped. For example: >>> b = bytes([102, 111, 111, 10, 0]) >>> s = bytes_to_human_readable_repr(b) >>> print(s) foo >>> print(repr(s)) 'foo\n\x00' rTrK)repr)r^r+r+r,bytes_to_human_readable_reprs r`c@seZdZdZdS) DecodeErrorzException raised when a file cannot be decoded due to an unknown encoding type. Essentially a wrapper for the LookupError raised by `bytearray.decode` N)__name__ __module__ __qualname____doc__r+r+r+r,rasrasourcec Cs`|drd}|dd}nt|\}}z||}W|Sty/}ztt||d}~ww)zmRead the Python file with while obeying PEP-263 encoding detection. Returns the source as a string. srVrN)r)r]rY LookupErrorrar$)rfr[r/ source_text lookuperrr+r+r,decode_python_encodings   rjpathreadCallable[[str], bytes]list[str] | NonecCsJz||}Wn tyYdSwz t|}W|Sty$YdSw)zaTry reading a Python file as list of source lines. Return None if something goes wrong. N)OSErrorrj splitlinesra)rkrlrf source_linesr+r+r, read_py_files   rrrZmax_lenintcol min_widthcCs|d|dkrd|d}t||kr|dfS|||kr(|d|ddfS|t||dkrL|||d}d||||dd|dfSd|| dt||dfS)aTrim a line of source code to fit into max_len. Show 'min_width' characters on each side of 'col' (an error location). If either start or end is trimmed, this is indicated by adding '...' there. A typical result looks like this: ...some_variable = function_to_call(one_arg, other_arg) or... Return the trimmed string and the column offset to to adjust error location. rTr5rN...rrD)rZrsrurvoffsetr+r+r,trim_source_lines   $"rzlist[tuple[int, str]]cCs\d}||vrgS|d}g}t|D]\}}||r+||d|t|dfq|S)Nz# mypy:  r5)rM enumerater)r7rD)rfPREFIXlinesresultsirZr+r+r,get_mypy_commentss  rz JUNIT_HEADER_TEMPLATEz {text} JUNIT_TESTCASE_FAIL_TEMPLATEz {text} JUNIT_ERROR_TEMPLATEzr JUNIT_TESTCASE_PASS_TEMPLATEz JUNIT_FOOTERdtfloatseriousmessages_by_filedict[str | None, list[str]]versionplatformc Csddlm}|rd}t|}nt|}d}tj|||t|pdd}|s/|tj|||d7}n9|D]4\} } | durR|tj|d| | |dj||| dd 7}q3|tj|d| d |d j||d d 7}q3|t 7}|S) Nr)escaper5)errorsfailurestimetests)rverrr|z"mypy-py{ver}-{platform} {filename})rrfilename)rQrrr#rzmypy-py{ver}-{platform})rr) xml.sax.saxutilsrrDrformatritemsrrCr) rrrrrrrrxmlrmessagesr+r+r,_generate_junit_contentss@        rNonec Csrt|||||}tjtj|}tj|ddt|d}||dWddS1s2wYdS)NT)exist_okwbutf-8) rosrkdirnameabspathmakedirsopenwriteencode) rrrrkrrrxml_dirsfr+r+r,write_junit_xml5s  "rc@s$eZdZdZd ddZd d d Zd S)IdMapperzGenerate integer ids for objects. Unlike id(), these start from 0 and increment by 1, and ids won't get reused across the life-time of IdMapper. Assume objects don't redefine __eq__ or __hash__. r'rcCsi|_d|_dSr>id_mapnext_idselfr+r+r,rPs zIdMapper.__init__orIrtcCs.||jvr|j|j|<|jd7_|j|S)Nr5r)rrr+r+r,idTs   z IdMapper.idN)r'r)rrIr'rt)rbrcrdrerrr+r+r+r,rGs rfullnamecCs|dddS)zBDrop the final component of a qualified name (e.g. ('x.y' -> 'x').r4r5r)r6)rr+r+r, get_prefix[sr cur_mod_idrelativeis_cur_package_init_filetuple[str, bool]cCsp|dkr|dfS|d}|}|r|d8}t||k}|dkr)d|d| }||r3d||fSd|fS)NrTr4r5)rMrDrC)rrr<rpartsrelokr+r+r,correct_relative_import`s  rz$Final[dict[type[object], list[str]]] fields_cachecls type[object] Sequence[str]cs>ddl|tvr|fdd}dd|Dt|<t|S)Nrcs|p |SN)isgetsetdescriptorismemberdescriptor)rinspectr+r,ysz'get_class_descriptors..cSs$g|]\}}|dkr|dkr|qS) __weakref____dict__r+).0xyr+r+r, {s$z)get_class_descriptors..)rr getmembers)rmembersr+rr,get_class_descriptorsrs rr+newold copy_dict skip_slotstuple[str, ...]c Cst|dr|rt|j|_n|j|_t|jD]+}||vrqzt||r.t||t||n t||r8t||WqtyBYqwdS)aCopy state of old node to the new node. This handles cases where there is __dict__ and/or attribute descriptors (either from slots or because the type is defined in a C extension module). Assume that both objects have the same __class__. rN) hasattrdictrr __class__setattrgetattrdelattrAttributeError)rrrrattrr+r+r,replace_object_states"     rpath1path2cCst|t|jvS)z8Given two paths, return if path1 is a sub-path of path2.)pathlibPathparents)rrr+r+r, is_sub_pathsrstatuscCs"tjtjt|dS)zKill the current process without fully cleaning up. This can be quite a bit faster than a normal exit() since objects are not freed. N)sysstdoutflushstderrr_exit)rr+r+r, hard_exits  rcCs |dS)z+Remove internal suffixes from a short name.')rNr0r+r+r,unmangles rexistingContainer[str]cCsL|d}||vr |Sd}|t||vr |d7}|t||vs|t|S)zGet a simple redefinition name not present among existing. For example, for name 'foo' we try 'foo-redefinition', 'foo-redefinition2', 'foo-redefinition3', etc. until we find one that is not in existing. z -redefinitionrTr5)r$)r#rr_namerr+r+r,get_unique_redefinition_names rprogramcCs,tjdddkrtdj|ddSdS)zAReport issues with the Python used to run mypy, dmypy, or stubgenNrT)rzXRunning {name} with Python 3.7 or lower is not supported; please upgrade to 3.8 or newerr0)r version_infoexitr)rr+r+r,check_python_versionsrrtuple[int, int, int]cCs@dd|D}dd|D}dd|D}t|t|t|fS)zDCount total number of errors, notes and error_files in message list.cSg|]}d|vr|qS): error:r+rer+r+r,rzcount_stats..cSsh|] }|ddqS):r)rMrr+r+r, szcount_stats..cSr): note:r+rr+r+r,rrrx)rr error_filesnotesr+r+r, count_statssrmsgcCsVd}g}d}|D]}|dkr|r||d}q|dkr| }||7}q|||S)z=Split line of text into words (but not within quoted groups).rT ")r7)r next_wordres allow_breakcr+r+r, split_wordss    r cCsttddptjptS)zIGet current terminal width if possible, otherwise return the default one.MYPY_FORCE_TERMINAL_WIDTH0)rtrgetenvshutilget_terminal_sizecolumnsrr+r+r+r,get_terminal_widths r first_offset num_indentc Cst|}|d}g}|r8|d}|r||n||}t|t|d|kr/|d|7}n|||}|s ||dd|} | |S)aWrap a long error message into few lines. Breaks will only happen between words, and never inside a quoted group (to avoid breaking types such as "Union[int, str]"). The 'first_offset' is the width before the start of first line. Pad every next line with 'num_indent' spaces. Every line will be at most 'max_len' characters, except if it is a single word or quoted group. For example: first_offset ------------------------ path/to/file: error: 58: Some very long error message that needs to be split in separate lines. "Long[Type, Names]" are never split. ^^^^-------------------------------------------------- num_indent max_len rr5rr|)r poprDr7rC) rrsrrwords next_linerr max_line_lenpaddingr+r+r, soft_wraps     rdatacCst|S)zCompute a hash digest of some data. We use a cryptographic hash because we want a low probability of accidental collision, but we don't really care about any of the cryptographic properties. )hashlibsha256 hexdigest)rr+r+r, hash_digests rcupcCsHtjdkr Jdd|dddg}t|dd d }|S) z.Reproduce a gray color in ANSI escape sequencewin32Fz"curses is not available on WindowsrNrKmrr5r)rrrCrYcursestparmr)r set_colorgrayr+r+r,parse_gray_colors r%cCs>tdtdd}ztt|WStyt|YSw)NMYPY_FORCE_COLOR FORCE_COLORr )rr r&rt ValueError)env_varr+r+r,should_force_color(s   r*c@seZdZdZd7d d Zd8d d Zd8d dZd8ddZ   d9d:ddZ d;dd&d'Z d?d)d*Z d@dAd/d0Z dd+d1dBd5d6ZdS)CFancyFormatterzcApply color and bold font to terminal output. This currently only works on Linux and Mac. f_outIO[str]f_errhide_error_codesr&r'rcCs||_tjdvr d|_dSts|r|sd|_dStjdkr)| |_ntjdkr5| |_n| |_|jsM|j |j |j |j dd|_ dSdS)N)linuxdarwinr emscriptenTrr2r)redgreenblueyellownone)r/rr dummy_termr*isattyinitialize_win_colorsinitialize_vt100_colorsinitialize_unix_colorsREDGREENBLUEYELLOWcolors)rr,r.r/r+r+r,r6s(     zFancyFormatter.__init__cCsBtjdvsJd|_d|_d|_d|_d|_d|_d|_d |_ d S) SReturn True if initialization was successful and we can use colors, False otherwise)rr2zzzzzzzzT) rrBOLDUNDERr?r>r=r@NORMALDIMrr+r+r,r;Nsz&FancyFormatter.initialize_vt100_colorscCstjdksJtjdkr?t}|jtks|jtkrdSddl}|jj }d}d}d}d}| | |||B|B| d SdS) rBrFrNr5rTriT) rrgetwindowsversionmajorrbuildrctypeswindllkernel32SetConsoleMode GetStdHandler;)rwinverrJrLENABLE_PROCESSED_OUTPUTENABLE_WRAP_AT_EOL_OUTPUT"ENABLE_VIRTUAL_TERMINAL_PROCESSINGSTD_OUTPUT_HANDLEr+r+r,r:\s.   z$FancyFormatter.initialize_win_colorsc Cs^tjdksts dSz7ztj}Wn'tjy8tdd}tj |dWdn1s1wYYnwtj |dWn tj yKYdSwt d}t d}t d }t d }t d }|ro|ro|ro|ro|sqdS| |_ | |_| |_t||_t|tj |_t|tj |_t|tj |_t|tj |_d S) rBrFz /dev/nullrb)fdNboldsmulsetafrsgr0T)rrCURSES_ENABLEDrfilenoioUnsupportedOperationrr! setuptermerrortigetstrrYrErCrDr%rFr" COLOR_BLUEr? COLOR_GREENr> COLOR_REDr= COLOR_YELLOWr@)rrUrrVunderr#set_eseqnormalr+r+r,r<zs@           z%FancyFormatter.initialize_unix_colorsFrQr$color1Literal['red', 'green', 'blue', 'yellow', 'none']rV underlinedimcCsN|jr|S|r |j}nd}|r||j7}|r||j7}||j|||jS)z2Apply simple color and style (underlined or bold).r)r8rCrDrFrArE)rrQrhrVrjrkstartr+r+r,styles   zFancyFormatter.styleNrr3fixed_terminal_width int | NonecCsZ|pt}|}t|D]\}}d|vr3|jddd\}}t||t|tdd}|d|||<|dtrd|vr|td }||d} | d} | t} d | vrZd} n | | | d d} |td } t || | t \}}dt|||<dt| || }t|t||krt| d kr|d t||d d }|||d<q |S)zHImprove readability by wrapping error messages and trimming source code.rerror:r5maxsplitzerror: )rr^N~rrw) rcopyr}rMrrDr)rindexrindexrzr)rrrnwidth new_messagesrr_locr marker_line marker_columncolumnmarkerrs source_linerynew_marker_liner+r+r,fit_in_terminals0       zFancyFormatter.fit_in_terminalr_cCs d|vrL|jddd\}}|jr||jdddd||S|d}|d kr5||d }|d |}nd }||jdddd||||d Sd |vrk|jddd\}}|||}||dd|S|dtrd|vr~|j|dddS||dS|S)zBColorize an output line by highlighting the status and error code.rrpr5rqr3TrV[rKNrr6rznote:r5rrsr7)rk)rMr/rmhighlight_quote_groupsrfindunderline_linkr)r)rr_r{rcodeposcode formattedr+r+r,colorizes6    zFancyFormatter.colorizercCsp|ddr |S|d}d}t|D]!\}}|ddkr'|||d7}q||jd|dddd7}q|S)zMake groups quoted with double quotes bold (including quotes). This is used to highlight types, attribute names etc. rrTrrr7Tr)countrMr}rm)rrrr8rpartr+r+r,rs  z%FancyFormatter.highlight_quote_groupsnotecCsTtd|}|s |S|}|}|d||j|||ddd||dS)zuUnderline a link in a note message (if any). This assumes there is at most one link in the message. z https?://\S*Nr7T)rj)researchrlendrm)rrrWrlrr+r+r,r s 0zFancyFormatter.underline_linkT n_sourcesrt use_colorcCs,d|dt|}|s|S|j|dddS)zFormat short summary in case of success. n_sources is total number of files passed directly on command line, i.e. excluding stubs and followed imports. zSuccess: no issues found in source filer4Trplural_srm)rrrrr+r+r,format_successszFancyFormatter.format_success)blockersrn_errorsn_filesrc Csdd|dt|d|dt|}|r|d7}n |d|dt|d7}|s*|S|j|d d d S) z)Format a short summary in case of errors.zFound z errorz in z filez$ (errors prevented further checking)z (checked r)r3Trr)rrrrrrrr+r+r, format_error!s$  zFancyFormatter.format_error)r,r-r.r-r/r&r'rr'r&)FFF) rQr$rhrirVr&rjr&rkr&r'r$r)rr3rnror'r3)r_r$r'r$)rr$r'r$)rr$r'r$)T)rrtrr&r'r$) rrtrrtrrtrr&rr&r'r$)rbrcrdrerr;r:r<rmrrrrrrr+r+r+r,r+0s&    )  %  r+ typeshed_dirfilecCsD|dur|nt}ztj|tj|f|kWSty!YdSw)NF)rrrk commonpathrr()rrr+r+r,is_typeshed_file5s  rcCs0|dsdStddtjtj|DS)Nz.pyiFcss|]}|dVqdS)z-stubsN)r*)r componentr+r+r, Asz'is_stub_package_file..)r*anyrrkrMr)rr+r+r,is_stub_package_file=s "rcCs|duo|dkSr.r+r0r+r+r,unnamed_functionDsrt0cCstt|dS)Ni)rtrperf_counter_ns)rr+r+r, time_spent_usKsrs int | SizedcCs&t|tr|nt|}|dkrdSdS)Nr5rr) isinstancertrD)rrr+r+r,rOsrdocstrcCshdtdt|}|dr.d|ddvr(|ddkr(d|dddSd |d Sd |d S) zKReturns docstring correctly encapsulated in a single or double quoted form.r|z (?<=[^\\])\\nrrr5rKz"""z''z"")rCrrMr_r))r docstr_reprr+r+r,quote_docstringWs    r)F)r#r$r%r&r'r&)r#r$r'r&)r2r$r'r3)r:r;r<r$r'r=)r:r;r<r$r'rB)rHrIr'r$)rQrRr'rS)r^rRr'r$)rfrRr'r$)rkr$rlrmr'rn) rZr$rsrtrurtrvrtr'rS)rfr$r'r{) rrrr&rrrr$rr$r'r$)rrrr&rrrkr$rr$rr$r'r)rr$r'r$) rr$rrtr<r$rr&r'r)rrr'r)Fr+) rrIrrIrr&rrr'r)rr$rr$r'r&)r)rrtr'r)r#r$r'r$)r#r$rrr'r$)rr$r'r)rr3r'r)rr$r'r3)r'rt) rr$rsrtrrtrrtr'r$)rrRr'r$)rrRr'r$r)rr=rr$r'r&)rr$r'r&)r#r=r'r&)rrtr'rt)rrr'r$)rr$r'r$)_re __future__rrr\rrrr rr importlibrimportlib_resourcestypingrrrrrr r r typing_extensionsr r!_cursesrZ ImportErrorr rr$filesr__annotations__rk _resourceparentcompilerrrrrr frozensetr"r-r1r9rAr?rPr]r` Exceptionrarjrrrzrrrrrrrrrrrrrrrrrrrrr rrrr%r*r+rrrrtime_refrrrr+r+r+r,s  (                      2          $