X-PhzadZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZmZmZddlmZmZddlmZmZmZddlmZmZdd lmZdd lmZmZm Z dd l!m"Z"Gd d ej#Z$ej%dddZ&ej'dkrde&_(e&)de&*dede&*dddde"zde&+Z,e,-dd xZ.Z/e/*d!d"e0d#$e/*d%d&e1d'$e/*d(d)d*e0d+,e,-d-d. xZ2Z/e/*d!d"e0d#$e/*d%d&e1d'$e/*d(d)d*e0d+,e,-d/d0 xZ3Z/e/*d1d2d3d45e/*d6d7 e,-d8d9 xZ4Z/e,-d:d; xZ5Z/e,-dxZ6Z/e/*d1d2d3d45e/*d?d@d3ej75e/*dAdB e/*dCdD e/*dEd"dFdGHe/*dId3dJ5e,-dKe$dL>xZ8Z/e/*d1d2d3d45e/*dAdB e/*dCdD e/*d%d&e1d'$e/*d!d"e0d#$e/*dId3dJ5e/*d(dMd*e0dN,e,-dOe$dP>xZ9Z/e/*d1d2d3d45e/*d?d@d3ej75e/*dAdB e/*dCdD e/*dId3dJ5e/*dQd"d*dRHe/*dSd"d*dTHe,-dUdV xZ:Z/e/*dWdXe0dY$e/*dZd3d[5e/*d\d3d]5e/*d^d3d_5e/*d`e;dabe/*dcd3dd5e/*dedfe0dg$e/*dhe1dibe,-djdk xZZ/e/*d%d&e1d'$e/*d!d"e0d#$e/*d(d)d*e0d+,e/*dej7 e,-dxZ?Z/[/Gdde@ZAddZBddZCeejDgdfZEddZFeFe.ddZGeFe2ddZHdddZIdddZJdddZKeFe8ddZLeFe3ddZMeFe4ddZNeFe5ddZOeFe6ddZPeFe9ddZQeFe:ddZReFe<ddZSddZTddZUeFe=ddZVeFe>ddZWeFe?ddZXdddd„ZYddĄZZddƄZ[ddȄZ\ddɄZ]ddʄZ^dS)zClient for mypy daemon mode. This manages a daemon process which keeps useful state in memory rather than having to read it back from disk on each run. ) annotationsN)Mapping)AnyCallableNoReturn)alivekill)DEFAULT_STATUS_FILEreceivesend) IPCClient IPCException)RECURSION_LIMIT)check_python_versionget_terminal_widthshould_force_color) __version__c eZdZdfd ZxZS) AugmentedHelpFormatterprogstrkwargsrreturnNonec @tjd|dd|dS)N)rmax_help_position)super__init__)selfrr __class__s Q/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/mypy/dmypy/client.pyr zAugmentedHelpFormatter.__init__ s-CdbCCFCCCCC)rrrrrr)__name__ __module__ __qualname__r __classcell__)r"s@r#rrsGDDDDDDDDDDr$rdmypyzClient for mypy daemon mode@)r descriptionfromfile_prefix_chars)Tactionz --status-filez&status file to retrieve daemon details)defaulthelpz-Vz --versionversionz %(prog)s z&Show program's version number and exit)r0r3r2startz Start daemon)r2z --log-fileFILEz#Direct daemon stdout/stderr to FILE)metavartyper2z --timeoutTIMEOUTz$Server shutdown timeout (in seconds)flagsFLAG*z$Regular mypy flags (precede with --))r6nargsr7r2restartz/Restart daemon (stop or kill followed by start)statuszShow daemon statusz-vz --verbose store_truezPrint detailed status)r0r2z--fswatcher-dump-filez0Collect information about the current file statestopz)Stop daemon (asks it politely to go away)r zKill daemon (kills the process)checkz"Check some files (requires daemon))formatter_classr2z-qz--quietz --junit-xmlz!Write junit.xml to the given filez--perf-stats-filez/write performance information to the given filefiles+zFile (or directory) to check)r6r<r2z--export-typeszLStore types of all expressions in a shared location (useful for inspections)runz2Check some files, [re]starting daemon if necessaryARGz.Regular mypy flags and files (precede with --)recheckzRRe-check the previous list of files, with optional modifications (requires daemon)z--updatezGFiles in the run to add or check again (default: all from previous run)z--removezFiles to remove from the runsuggestz>Suggest a signature or show call sites for a specific functionfunctionFUNCTIONz9Function specified as '[package.]module.[class.]function'z--jsonz:Produce json that pyannotate can use to apply a suggestionz --no-errorsz-Only produce suggestions that cause no errorsz--no-anyz/Only produce suggestions that don't contain Anyz --flex-anyzJAllow anys in types if they go above a certain score (scores are from 0-1))r7r2z --callsitesz+Find callsites instead of suggesting a typez --use-fixmeNAMEzCA dummy name to use instead of Any for types that can't be inferredz --max-guesseszBSet the maximum number of types to try for a function (default 64)inspectz+Locate and statically inspect expression(s)locationLOCATIONzLocation specified as path/to/file.py:line:column[:end_line:end_column]. If position is given (i.e. only line and column), this will return all enclosing expressionsz--show INSPECTIONr7)r7attrs definitionzWhat kind of inspection to run)r6r7r1choicesr2countzFIncrease verbosity of the type string representation (can be repeated))r0r1r2z--limitNUMzQReturn at most NUM innermost expressions (if position is given); 0 means no limit)r6r7r1r2z--include-spanz]Prepend each inspection result with the span of corresponding expression (e.g. 1:2:3:4:"int")z--include-kindz^Prepend each inspection result with the kind of corresponding expression (e.g. NameExpr:"int")z--include-object-attrsz4Include attributes of "object" in "attrs" inspectionz --union-attrszgInclude attributes valid for some of possible expression types (by default an intersection is returned)z--force-reloadz?Re-parse and re-type-check file before inspection (may be slow)hangzHang for 100 secondsdaemonzRun daemon in foregroundz--options-datar2ceZdZdZdS) BadStatuszException raised when there is something wrong with the status file. For example: - No status file found - Status file malformed - Process whose pid is in the status file does not exist N)r%r&r'__doc__rr$r#rXrXsr$rXargv list[str]rrctdtjtt|}|jstdS ||dS#t$r%}t|j dYd}~dSd}~wt$r+tj tjdYdSwxYw)zThe code is top-down.r)rN)rsyssetrecursionlimitrparser parse_argsr0 print_usagerXfailargs Exception traceback print_excexit)rZrderrs r#mainrjs!!!/***   T " "D ;   KK         !               ! ! ! HQKKKKKK  s&A== CB''4CCmsgrrcdt|tjtjddS)N)filer])printr^stderrrh)rks r#rcrc&s( #CJHQKKKKKr$ subparserargparse.ArgumentParser*Callable[[ActionFunction], ActionFunction]cdfd }|S)z3Decorator to tie an action function to a subparser.funcActionFunctionrc4||S)Nr/) set_defaults)rtrps r#registerzaction..register1sd+++ r$)rtrurrur)rprxs` r#r0r0.s( Or$rdargparse.Namespacec t|jtdn#t$rYnwxYwt |dS)a Start daemon (it must not already be running). This is where mypy flags are set from the command line. Setting flags is a bit awkward; you have to use e.g.: dmypy start -- --strict since we don't want to duplicate mypy's huge list of flags. zDaemon is still aliveN) get_status status_filercrX start_serverrds r#do_startr;sd&4#$$$ $%%%%      s & 33c$t|dS)zRestart daemon (it may or may not be running; but not hanging). We first try to stop it politely if it's running. This also sets mypy flags from the command line (see do_start()). N)restart_serverr~s r# do_restartrQs4r$F allow_sourcesboolch t|n#t$rYnwxYwt||dS)z?Restart daemon (it may or may not be running; but not hanging).N)do_stoprXr})rdrs r#rr[sL       }%%%%%s  cddlm}m}||j|}|||j|j|jrtjdt|jdS)z8Start the server from command arguments and wait for it.r) daemonizeprocess_start_options)timeoutlog_filer]N) mypy.dmypy_serverrrr9r|rrr^rhwait_for_server)rdrrr start_optionss r#r}r}esyCBBBBBBB))$*mDDMy 0$,QUQ^___  D$%%%%%r$@r|rfloatc(tj|z}tj|krU t|}n$#t$rtjdYHwxYwt |t ddSt ddS)zTWait until the server is up. Exit if it doesn't happen within the timeout. g?zDaemon startedNz%Timed out waiting for daemon to start)time read_statusrXsleep check_statusrnrc)r|rendtimedatas r#rrps ikkG#G )++   {++DD    JsOOO H  T  011111s?A A ct|jst|dtj}t |jdt |j|j}d|vrQtd|dt|dt |jdt |j|j}tj}||z |d<t||j |j |j dS) aDo a check, starting (or restarting) the daemon as necessary Restarts the daemon if the running daemon reports that it is required (due to a configuration change, for example). Setting flags is a bit awkward; you have to use e.g.: dmypy run -- --strict a.py b.py ... since we don't want to duplicate mypy's huge list of flags. (The -- is only necessary if flags are specified.) TrrE)r3rd export_typesr=z Restarting: roundtrip_timeN) is_runningr|r}rrequestrr9rrnr check_outputverbose junit_xmlperf_stats_filerdt0responset1s r#do_runrs d& ' '/T.... B   Z& HH 2Xi022333t40000   *     B!#bH 4<9MNNNNNr$cRt|j}|jrt|t |t |jd|jd}|jsd|vrt|d|vr#tdtj ddtdd S) zOPrint daemon status. This verifies that it is responsive to requests. r>)fswatcher_dump_filererror:Daemon may be busy processing; if this persists, consider r killzDaemon is up and runningN) rr|r show_statsrrrrcr^rZrn)rdr>rs r# do_statusrs ) * *F |6 (8PZ[H |w(**8( \#(ST+ \ \ \]]] $%%%%%r$ct|jdd}d|vr4t|tdtjddd St dd S) z!Stop daemon via a 'stop' request.r@rrrrrrzDaemon stoppedN)rr|rrcr^rZrnrdrs r#rrspt';;;H(8 \#(ST+ \ \ \]]]]] r$ct|j\}} t|tddS#t$r'}t t |Yd}~dSd}~wwxYw)z!Kill daemon process with SIGKILL.z Daemon killedN)r{r|r rnOSErrorrcr)rdpid_ris r#do_killrs|( ) )FC S  o  SXXs9 A*A%%A*ctj}t|jd|j|j}tj}||z |d<t ||j|j|jdS)z(Ask the daemon to check a list of files.rA)rCrrN) rrr|rCrrrrrrs r#do_checkrsi Bt' QUQbcccH B!#bH 4<9MNNNNNr$cJtj}|j|j)t|jd|j|j|j}nt|jd|j}tj}||z |d<t ||j|j|j dS)aAsk the daemon to recheck the previous list of files, with optional modifications. If at least one of --remove or --update is given, the server will update the list of files to check accordingly and assume that any other files are unchanged. If none of these flags are given, the server will call stat() on each file last checked to determine its status. Files given in --update ought to exist. Files given in --remove need not exist; if they don't they will be ignored. The lists may be empty but oughtn't contain duplicates or overlap. NOTE: The list of files is lost when the daemon is restarted. NrG)rremoveupdate)rr) rrrrr|rrrrrrs r# do_recheckrs B {$+"9   *;;    4+YTEVWWW B!#bH 4<9MNNNNNr$c t|jd|j|j|j|j|j|j|j|j  }t|ddddS)zAsk the daemon for a suggested signature. This just prints whatever the daemon reports as output. For now it may be closer to a list of call sites. rH)rIjson callsites no_errorsno_anyflex_any use_fixme max_guessesFNrrr) rr|rIrrrrrrrrrs r# do_suggestrsi  Y..{.$   H5D$OOOOOOr$c t|jd|j|j|j|j|j|j|j|j |j  }t|ddddS)z.Ask daemon to print the type of an expression.rL) showrM verbositylimit include_span include_kindinclude_object_attrs union_attrs force_reloadFNr) rr|rrMrrrrrrrrrs r# do_inspectrsr  Y,j&&!6$&   H5D$OOOOOOr$rdict[str, Any]rr str | Nonerc bd|vrt|d |d|d|d}}}n/#t$r"tdt|YnwxYwtj|tjtj|tj|rt||rRddl m }||z }||dt||rd |ini||d |d |rS| d i} t|d 5} tj| | d d d n #1swxYwY|rtj|d Sd S)zkPrint the output from a check or recheck command. Call sys.exit() unless the status code is zero. routrir>z Response: r)write_junit_xmlrNpython_versionplatformstatsw)rcKeyErrorrr^stdoutwriteflushror mypy.utilr splitlinesrgetopenrdumprh) rrrrrri status_codermessages telemetryfs r#rr.s!( Xg + (%(8BT+S +++ )#h-- ) )*****+JSJJSJ8  ------#I))++ % & II ( 0T8  b  % & Z    $LL"-- /3 ' ' $1 Ii # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ s!4)A A *F  FFMapping[str, object]c |t|D]\}}|dvr`t|dd}t|dkr |dddt|dz d}t |dd |it |dd t |t rd |zn|d dS) N)rrirro2(z ... z more characters24z: z%.3fz>10)sorteditemsreprlenrn isinstancer)rkeyvalues r#rrWsX^^--..\\ U 4 4 4KK"%E5zzB ":KKCJJrMKKK ee, - - -  ssssj6N6N$YFUNNTY$YZ[[[[\\r$cNtt|jdddS)z&Hang for 100 seconds, as a debug hack.rUrrN)rnrr|r~s r#do_hangrcs* '$"FA 6 6 677777r$c6ddlm}m}|jrut |jddxt _t _t j}tj |dtj |d|j rQddl m }tjt!j|j }|}||}n||jd }|||j|j d S) z!Serve requests in the foreground.r)Serverrar) bufferingr])OptionsFrrN)rrrrrr^rrofilenoosdup2 options_data mypy.optionsrpickleloadsbase64 b64decode apply_changesr9r|rserve)rdrrfdr options_dict options_objoptionss r# do_daemonr is@??????? }"&t}cQ"G"G"GG SZ Z    A A I((((((|F$4T5F$G$GHH gii ++L99'' %HHH F7D$dl;;;AACCCCCr$c8tdS)z'Print full help (same as dmypy --help).N)r` print_helpr~s r#do_helprs r$rcommand int | Nonekwdsobjectc i}t|}||d<tjp t |d<t |d<t |\}} t||5}t||d} | st|}t| dd} | dd} | rtj | | dd} | rtj | | dddn #1swxYwYn/#ttf$r} d t!| icYd} ~ Sd} ~ wwxYw|S) aSend a request to the daemon. Return the JSON dict with the response. Raise BadStatus if there is something wrong with the status file or if the process whose pid is in the status file has died. Return {'error': } if an IPC operation or receive() raised OSError. This covers cases such as connection refused or closed prematurely as well as invalid JSON received. ris_ttyterminal_widthFfinalrNror)dictr^risattyrrr{r r r rpoprrorrr) r|rrrrrdrnameclientrrroris r#rrs "H ::DDOZ&&((@,>,@,@DN/11D %%GAt# tW % % -    E -"6??X\\'599::"h55-J$$V,,,!h55-J$$V,,, -  - - - - - - - - - - - - - - - \ "###S"""""""# OsC*E :B7D=1 E =EE EE E5E0*E50E5tuple[int, str]c>t|}t|S)zRead status file and check if the process is alive. Return (pid, connection_name) on success. Raise BadStatus if something's wrong. )rr)r|rs r#r{r{s { # #D   r$rcBd|vrtd|d}t|tstdt|stdd|vrtd|d}t|tstd||fS)z}Check if the process is alive. Return (pid, connection_name) on success. Raise BadStatus if something's wrong. rz"Invalid status file (no pid field)zpid field is not an intzDaemon has diedconnection_namez.Invalid status file (no connection_name field)z%connection_name field is not a string)rXrintrr)rrrs r#rrs D<=== u+C c3  31222 ::+)***$$HIII,-O os + +A?@@@  r$dict[str, object]cjtj|stdt |5} t j|}n"#t$r}td|d}~wwxYw dddn #1swxYwYt|tstd|S)zRead status file. Raise BadStatus if the status file doesn't exist or contains invalid JSON or the JSON is not a dict. zNo status file foundz Malformed status file (not JSON)Nz Invalid status file (not a dict)) rpathisfilerXrrloadrerr)r|rres r#rrs 7>>+ & &0./// k  Ga G9Q<??Q F G GGGGGGGGGGGGGGG dD ! !<:;;; Ks5BAB A4A//A44BB BcJ t|n#t$rYdSwxYwdS)z&Check if the server is running cleanlyFT)r{rX)r|s r#rrs?; uu 4s   cHttjdddS)Nr)rjr^rZrr$r# console_entryr)s !""r$)rZr[rr)rkrrr)rprqrrr)rdryrr)F)rdryrrrr)r)r|rrrrr) rrrrrrrrrr)rrrr) r|rrrrrrrrr)r|rrr)rrrr)r|rrr!)r|rrr)rr)_rY __future__rargparserrrrr^rrfcollections.abcrtypingrrr mypy.dmypy_osrr mypy.dmypy_utilr r r mypy.ipcr r mypy.mainrrrrr mypy.versionrRawDescriptionHelpFormatterrArgumentParserr` version_infocolorrw add_argumentadd_subparsers subparsers add_parser start_parserprr restart_parser status_parser stop_parser kill_parser check_parserSUPPRESS run_parserrecheck_parsersuggest_parserrinspect_parser hang_parser daemon_parser help_parserrerXrjrc Namespacerur0rrrr}rrrrrrrrrrrrr rrr{rrrr)rr$r#rKsR #""""" ######**********%%%%%%%%>>>>>>>>>>,,,,,,,,%%%%%%RRRRRRRRRR$$$$$$ DDDDDXADDD ! ;SV   wFL4 07_  + % 1  " " $ $ ((~(FFF q|V#B++++++\    ,$r$