o 2gN~@sUdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddlmZmZmZmZmZmZmZmZmZddlmZded<zddlZWn ey`dZYnwz ddlZddlZd ZWn eyxd ZYnwed Z e j!d kre"e #d dZ$ded<ne %d dZ&e"e&j'dZ$Wdn1swYe(dZ)ded<dZ*ded<dZ+ded<dZ,ded<dZ-ded<dZ.ded<e/dZ0ded<ddd%d&Z1dd'd(Z2dd+d,Z3dd1d2Z4dd4d5Z5dd8d9Z6dd=d>Z7dd@dAZ8GdBdCdCe9Z:ddEdFZ;ddKdLZdWZ?dedX<dYZ@dedZ<d[ZAded\<d]ZBded^<d_ZCded`<ddhdiZDddkdlZEGdmdndnZFddpdqZGddvdwZHiZIdxedy<dd}d~ZJ ddddZKdddZLe jMdks}e jMdkrdddZNndddZNddddZOdddZPdddZQdddZRdddZSdddZTdddZUddddZVdddZWdddZXdddZYGdddZZdddZ[dddZ\dddZ]e j^Z_dddZ`ddd„ZadddńZbddddȄZcdddʄZddS)z3Utility functions with no non-trivial dependencies.) annotationsN) resources) IOAnyCallable ContainerFinalIterableSequenceSizedTypeVar)LiteralrorjsonTFT) 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-A/home/garg/my-data/venv/lib/python3.10/site-packages/mypy/util.py is_dunderAs r/cCst| o|do|dSN_)r/r+r,r%r-r-r. is_sunderMsr3mod_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)r4outr-r-r.split_module_namesQs  r;modules Iterable[str]target str | NonecCst||}|dur dS|dSNr) split_target)r<r>resultr-r-r. module_prefix^s rCtuple[str, str] | NonecCsRg} ||vr|d|fS|dd}t|dkrdS|d}|d|dq)NTr6r7r)joinr8leninsert)r<r> remaining componentsr-r-r.rAes  rAobjobjectcCs,|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'. Nnilr6z'>)r&typesplitrstrip)rJtr-r-r. short_typeqs rRtextbytestuple[str, int]cCsZt|}|r'|drdnd}|dd}|ds!|dkr#d}||fSd}|d fS) z*PEP-263 for detecting Python file encodingr7rascii)z iso-latin-1-zlatin-1-z iso-latin-1zlatin-1utf8rM)rmatchgroupdecoder+)rSrBlineencodingdefault_encodingr-r-r.find_python_encoding|s 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' rVrM)repr)r`r-r-r.bytes_to_human_readable_reprs rbc@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.rcsrcsourcec 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. srXrN)r+r_r[ LookupErrorrcr&)rhr]r1 source_text lookuperrr-r-r.decode_python_encodings   rlpathreadCallable[[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)OSErrorrl splitlinesrc)rmrnrh source_linesr-r-r. read_py_files   rtr\max_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. rVr7rN...rrF)r\rurwrxoffsetr-r-r.trim_source_lines   $"r|list[tuple[int, str]]cCs\d}||vrgS|d}g}t|D]\}}||r+||d|t|dfq|S)Nz# mypy:  r7)rO enumerater+r9rF)rhPREFIXlinesresultsir\r-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}n:|D]5\} } | durR|tj|d| | |dj||| dd 7}q3|tj|d| d |d |d |d 7}q3|t 7}|S) Nr)escaper7)errorsfailurestimetests)rverrr~z"mypy-py{ver}-{platform} {filename})rrfilename)rSrrr%rzmypy-py-) xml.sax.saxutilsrrFrformatritemsrrEr) rrrrrrrrxmlrmessagesr-r-r._generate_junit_contents s@       rNonec Csrt|||||}tjtj|}tj|ddt|d}||dWddS1s2wYdS)NT)exist_okwbutf-8) rosrmdirnameabspathmakedirsopenwriteencode) rrrrmrrrxml_dirsfr-r-r.write_junit_xml;s  "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.rVs zIdMapper.__init__orKrvcCs.||jvr|j|j|<|jd7_|j|S)Nr7r)rrr-r-r.idZs   z IdMapper.idN)r)r)rrKr)rv)rdrerfrgrrr-r-r-r.rMs rfullnamecCs|dddS)zBDrop the final component of a qualified name (e.g. ('x.y' -> 'x').r6r7r)r8)rr-r-r. get_prefixasr cur_mod_idrelativeis_cur_package_init_filetuple[str, bool]cCsp|dkr|dfS|d}|}|r|d8}t||k}|dkr)d|d| }||r3d||fSd|fS)NrTr6r7)rOrFrE)rrr>rpartsrelokr-r-r.correct_relative_importfs  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.sz'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_descriptorsxs 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"     rdircCs |tjs |tj7}||S)ayGiven two paths, return if path is a sub-path of dir. Moral equivalent of: Path(dir) in Path(path).parents Similar to the pathlib version: - Treats paths case-sensitively - Does not fully handle unnormalised paths (e.g. paths with "..") - Does not handle a mix of absolute and relative paths Unlike the pathlib version: - Fast - On Windows, assumes input has been slash normalised - Handles even fewer unnormalised paths (e.g. paths with "." and "//") As a result, callers should ensure that inputs have had os.path.abspath called on them (note that os.path.abspath will normalise) )r,rsepr+)rmrr-r-r.is_sub_path_normabss   rlinuxdarwincCs0|ds|s |S|dr||S|d|S)N/)r+r,)rmr`r-r-r. os_path_joins   rapcCstj||Sr)rrmrE)rrr-r-r.rsstatuscCs"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.')rPr2r-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 -redefinitionrVr7)r&)r%rr_namerr-r-r.get_unique_redefinition_names rprogramcCs,tjdddkrtdj|ddSdS)zAReport issues with the Python used to run mypy, dmypy, or stubgenNrV)rzXRunning {name} with Python 3.7 or lower is not supported; please upgrade to 3.8 or newerr2)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)rOrr-r-r. szcount_stats..cSr): note:r-rr-r-r.rrrz)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 ")r9)r next_wordres allow_breakcr-r-r. split_wordss    rcCsttddptjptS)zIGet current terminal width if possible, otherwise return the default one.MYPY_FORCE_TERMINAL_WIDTH0)rvrgetenvshutilget_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 rr7r r~)rpoprFr9rE) rrurrwords 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. )hashlibsha1 hexdigestrr-r-r. hash_digest;sr$cupcCsHtjdkr Jdd|dddg}t|dd d }|S) z.Reproduce a gray color in ANSI escape sequencewin32Fz"curses is not available on WindowsrNrMmrr7r)rrrEr[cursestparmr)r% set_colorgrayr-r-r.parse_gray_colorEs r,cCs>tdtdd}ztt|WStyt|YSw)NMYPY_FORCE_COLOR FORCE_COLORr)rrr(rv ValueError)env_varr-r-r.should_force_colorNs   r1c@seZdZdZ d8d9d d Zd:d dZd:ddZd:ddZ   d;dd!d"Z d?d$d%Z d@d'd(Z dAd*d+Z dBdCd0d1Z dd,d2dDd6d7ZdS)EFancyFormatterzcApply color and bold font to terminal output. This currently only works on Linux and Mac. Ff_outIO[str]f_errhide_error_codesr( hide_successr)rcCs||_||_tjdvrd|_dSts |r|s d|_dStjdkr,| |_ntjdkr8| |_n| |_|jsP|j |j |j |j dd|_dSdS)N)rrr& emscriptenTr&r8r)redgreenblueyellownone)r6r7rr dummy_termr1isattyinitialize_win_colorsinitialize_vt100_colorsinitialize_unix_colorsREDGREENBLUEYELLOWcolors)rr3r5r6r7r-r-r.r\s*     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)r&r8zzzzzzzzT) rrBOLDUNDERrErDrCrFNORMALDIMrr-r-r.rAxsz&FancyFormatter.initialize_vt100_colorscCstjdksJtjdkr?t}|jtks|jtkrdSddl}|jj }d}d}d}d}| | |||B|B| d SdS) rHr&FrNr7rVriT) rrgetwindowsversionmajorrbuildrctypeswindllkernel32SetConsoleMode GetStdHandlerA)rwinverrPrRENABLE_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) rHr&Fz /dev/nullrb)fdNboldsmulsetafr%sgr0T)rrCURSES_ENABLEDrfilenoioUnsupportedOperationrr( setuptermerrortigetstrr[rKrIrJr,rLr) COLOR_BLUErE COLOR_GREENrD COLOR_REDrC COLOR_YELLOWrF)rr[rr\underr*set_eseqnormalr-r-r.rBs@           z%FancyFormatter.initialize_unix_colorsrSr&color1Literal['red', 'green', 'blue', 'yellow', 'none']r\ underlinedimcCsN|jr|S|r |j}nd}|r||j7}|r||j7}||j|||jS)z2Apply simple color and style (underlined or bold).r)r>rIrJrLrGrK)rrSrnr\rprqstartr-r-r.styles   zFancyFormatter.styleNrr5fixed_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:r7maxsplitzerror: )rr ^N~rry) rcopyrrOrrFr+rindexrindexr|r)rrrtwidth new_messagesrrelocr marker_line marker_columncolumnmarkerru source_liner{new_marker_liner-r-r.fit_in_terminals0       zFancyFormatter.fit_in_terminalrecCs 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.rrvr7rwr9Tr\[rMNrr<rznote:r;r ryr=)rq)rOr6rshighlight_quote_groupsrfindunderline_linkr+r)rrerrcodeposcode 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. r rVrrr=Tr)countrOrrs)rrrr:rpartr-r-r.r#s  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*Nr=T)rp)researchrrendrs)rrrYrrrr-r-r.r4s 0zFancyFormatter.underline_linkT n_sourcesrv use_colorcCs6|jrdSd|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. rzSuccess: no issues found in source filer:Tr)r7plural_srs)rrrrr-r-r.format_success@s zFancyFormatter.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)r9Tr)rrs)rrrrrrrr-r-r. format_errorNs$  zFancyFormatter.format_errorF) r3r4r5r4r6r(r7r(r)rr)r()FFF) rSr&rnror\r(rpr(rqr(r)r&r)rr5rtrur)r5)rer&r)r&)rr&r)r&)rr&r)r&)T)rrvrr(r)r&) rrvrrvrrvrr(rr(r)r&)rdrerfrgrrAr@rBrsrrrrrrr-r-r-r.r2Vs(   )  %  r2 typeshed_dirfilecCsD|dur|nt}ztj|tj|f|kWSty!YdSw)NF)rrrm commonpathrr/)rrr-r-r.is_typeshed_filebs  rcCs0|dsdStddtjtj|DS)Nz.pyiFcss|]}|dVqdS)z-stubsN)r,)r componentr-r-r. nsz'is_stub_package_file..)r,anyrrmrOr)rr-r-r.is_stub_package_filejs "rcCs|duo|dkSr0r-r2r-r-r.unnamed_functionqsrt0cCstt|dS)Ni)rvrperf_counter_ns)rr-r-r. time_spent_usxsrs int | SizedcCs&t|tr|nt|}|dkrdSdS)Nr7rr) isinstancervrF)rrr-r-r.r|srdocstrcCshdtdt|}|dr.d|ddvr(|ddkr(d|dddSd |d Sd |d S) zKReturns docstring correctly encapsulated in a single or double quoted form.r~z (?<=[^\\])\\nrr r7rMz"""z''z"")rErrOrar+)r docstr_reprr-r-r.quote_docstrings    rdebugcCsbtdur|rtj|tjtjBdStj|tjdS|r&tj|ddddStj|ddddS)N)optionrVT)indent sort_keysr),r)r separators)rdumps OPT_INDENT_2 OPT_SORT_KEYSjsonr)rJrr-r-r. json_dumpssrcCstdur t|St|Sr)rloadsrr#r-r-r. json_loadss  rr)r%r&r'r(r)r()r%r&r)r()r4r&r)r5)r<r=r>r&r)r?)r<r=r>r&r)rD)rJrKr)r&)rSrTr)rU)r`rTr)r&)rhrTr)r&)rmr&rnror)rp) r\r&rurvrwrvrxrvr)rU)rhr&r)r}) rrrr(rrrr&rr&r)r&)rrrr(rrrmr&rr&rr&r)r)rr&r)r&) rr&rrvr>r&rr(r)r)rrr)r)Fr-) rrKrrKrr(rrr)r)rmr&rr&r)r()rmr&r`r&r)r&)rr&rr&r)r&)r)rrvr)r)r%r&r)r&)r%r&rrr)r&)rr&r)r)rr5r)r)rr&r)r5)r)rv) rr&rurvrrvrrvr)r&)rrTr)r&)r%rTr)r&r)rr?rr&r)r()rr&r)r()r%r?r)r()rrvr)rv)rrr)r&)rr&r)r&)rJrKrr(r)rT)rrTr)r)erg __future__rr rbrrrrrr importlibrimportlib_resourcestypingrrrrrr r r r typing_extensionsr __annotations__r ImportErrorr(_cursesr`rrr&filesrrm _resourceparentcompilerrrrrr frozensetr$r/r3r;rCrArRr_rb Exceptionrcrlrtr|rrrrrrrrrrrrrrrrrrrrrrrrrr$r,r1r2rrrrtime_refrrrrrr-r-r-r.s  ,                        2            $