pPj^ZdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl mZmZddlmZddlmZddlmZmZdd lmZmZmZdd lmZmZmZmZdd l m!Z!m"Z"m#Z#dd l$m%Z%Gd dejLZ'ejPdddZ)e)jUde)jWdede)jWdddde%zde)jYZ-e-j]ddxZ/Z0e0jWd d!e1d"#e0jWd$d%e2d&#e0jWd'd(d)e1d*+e-j]d,d-xZ3Z0e0jWd d!e1d"#e0jWd$d%e2d&#e0jWd'd(d)e1d*+e-j]d.d/xZ4Z0e0jWd0d1d2d34e0jWd5d6e-j]d7d8xZ5Z0e-j]d9d:xZ6Z0e-j]d;e'd<=xZ7Z0e0jWd0d1d2d34e0jWd>d?d2ejp4e0jWd@dAe0jWdBdCe0jWdDd!dEdFGe0jWdHd2dI4e-j]dJe'dK=xZ9Z0e0jWd0d1d2d34e0jWd@dAe0jWdBdCe0jWd$d%e2d&#e0jWd d!e1d"#e0jWdHd2dI4e0jWd'dLd)e1dM+e-j]dNe'dO=xZ:Z0e0jWd0d1d2d34e0jWd>d?d2ejp4e0jWd@dAe0jWdBdCe0jWdHd2dI4e0jWdPd!d)dQGe0jWdRd!d)dSGe-j]dTdUxZ;Z0e0jWdVdWe1dX#e0jWdYd2dZ4e0jWd[d2d\4e0jWd]d2d^4e0jWd_eZ0e-j]ddxZ?Z0e0jWd$d%e2d&#e0jWd d!e1d"#e0jWd'd(d)e1d*+e0jWdejpe-j]dxZ@Z0[0ddZAddZBe ejgdfZDddZEeEe/ddZFeEe3ddZGdddZHdddZIdddZJeEe9ddZKeEe4ddZLeEe5ddZMeEe6ddZNeEe7ddZOeEe:ddZPeEe;ddZQeEe=ddZR ddZSddZTeEe>ddZUeEe?ddZVeEe@ddZWdd ddZXddZYddZZddZ[ddZ\y)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)CallableMapping)AnyNoReturn) b64decode)RECURSION_LIMIT)alivekill)DEFAULT_STATUS_FILEreceivesend) BadStatus IPCClient IPCException read_status)check_python_versionget_terminal_widthshould_force_color) __version__c eZdZdfd ZxZS)AugmentedHelpFormatterc *t|d|dd|y)N)progmax_help_position)super__init__)selfrkwargs __class__s V/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/mypy/dmypy/client.pyrzAugmentedHelpFormatter.__init__!s CdbCFC)rstrr!rreturnNone)__name__ __module__ __qualname__r __classcell__)r"s@r#rr sDDr$rdmypyzClient for mypy daemon mode@)r descriptionfromfile_prefix_charsactionz --status-filez&status file to retrieve daemon details)defaulthelpz-Vz --versionversionz %(prog)s z&Show program's version number and exit)r1r4r3startz Start daemon)r3z --log-fileFILEz#Direct daemon stdout/stderr to FILE)metavartyper3z --timeoutTIMEOUTz$Server shutdown timeout (in seconds)flagsFLAG*z$Regular mypy flags (precede with --))r7nargsr8r3restartz/Restart daemon (stop or kill followed by start)statuszShow daemon statusz-vz --verbose store_truezPrint detailed status)r1r3z--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_classr3z-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)r7r=r3z--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))r8r3z --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 INSPECTIONr8)r8attrs definitionzWhat kind of inspection to run)r7r8r2choicesr3countzFIncrease verbosity of the type string representation (can be repeated))r1r2r3z--limitNUMzQReturn at most NUM innermost expressions (if position is given); 0 means no limit)r7r8r2r3z--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-datar3ctdtjttj |}|j stjy |j |y#t$r"}t|jdYd}~yd}~wt$r,tjtjdYywxYw)zThe code is top-down.r,rN)rsyssetrecursionlimitr parser parse_argsr1 print_usagerfailargs Exception traceback print_excexit)argvr`errs r#mainrgs!/*   T "D ;;  KK   !       ! HHQK  sA.. C 7B5C  C cdt|tjtjdy)N)filerY)printrZstderrrd)msgs r#r_r_s #CJJHHQKr$cdfd }|S)z3Decorator to tie an action function to a subparser.c,j||S)Nr0) set_defaults)func subparsers r#registerzaction..register%sd+ r$)rpActionFunctionr&rsr)rqrrs` r#r1r1"s Or$cz t|jtdt |y#t$rYwxYw)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_filer_r start_serverr`s r#do_startry/s<&4##$ $%    s . ::ct|y)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_serverrxs r# do_restartr|Es 4r$cR t|t||y#t$rYwxYw)z?Restart daemon (it may or may not be running; but not hanging).N)do_stoprrw)r` allow_sourcess r#r{r{Os.  }%    s  &&cddlm}m}||j|}|||j|j |j rtjdt|jy)z8Start the server from command arguments and wait for it.r) daemonizeprocess_start_options)timeoutlog_filerYN) mypy.dmypy_serverrrr:rvrrrZrdwait_for_server)r`rrr start_optionss r#rwrwYsPC)$**mDM 0 0$,,QUQ^Q^_  D$$%r$ctj|z}tj|kr# t|}t |t dyt dy#t$rtjdYfwxYw)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)timerrsleep check_statusrjr_)rvrendtimedatas r#rrdso iikG#G ))+  {+D T  01  JJsO  s AA?>A?c t|js t|dtj}t |jdt |j |j}d|vrPtd|dt|dt |jdt |j |j}tj}||z |d<t||j|j|jy) 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.) TrrF)r4r` export_typesr>z Restarting: roundtrip_timeN) is_runningrvrwrrequestrr:rrjr{ check_outputverbose junit_xmlperf_stats_filer`t0responset1s r#do_runrxs d&& 'T. B   ZZ&& HH Xi0123t40    **   B!#bH 4<<9M9MNr$cLt|j}|jr t|t |t |jd|j d}|jsd|vr t|d|vr tdtjddtdy ) 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) rrvr show_statsrrrr_rZrerj)r`r?rs r# do_statusrs )) *F ||6 (8P8PZ[H ||w(*8( I#((ST+V[ \] $%r$ct|jdd}d|vr,t|tdtj ddy t dy ) z!Stop daemon via a 'stop' request.rArrrrrrzDaemon stoppedN)rrvrr_rZrerjr`rs r#r~r~sMt'';H(8 I#((ST+V[ \] r$ct|j\}} t|tdy#t$r}t t |Yd}~yd}~wwxYw)z!Kill daemon process with SIGKILL.z Daemon killedN)rurvr rjOSErrorr_r%)r`pid_rfs r#do_killrsM(( )FC S  o  SXs 1 AAActj}t|jd|j|j}tj}||z |d<t ||j |j|jy)z(Ask the daemon to check a list of files.rB)rDrrN) rrrvrDrrrrrrs r#do_checkrsf Bt'' QUQbQbcH B!#bH 4<<9M9MNr$ctj}|j |j9t|jd|j |j|j}n"t|jd|j }tj}||z |d<t ||j|j|jy)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. NrH)rremoveupdate)rr) rrrrrvrrrrrrs r# do_recheckrs B {{$++"9    **;;;;  4++YTEVEVW B!#bH 4<<9M9MNr$c t|jd|j|j|j|j |j |j|j|j }t|dddy)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. rI)rJjson callsites no_errorsno_anyflex_any use_fixme max_guessesFNrrr) rrvrJrrrrrrrrrs r# do_suggestrsd  YY....{{..$$ H5D$Or$c t|jd|j|j|j|j |j |j|j|j|j }t|dddy)z.Ask daemon to print the type of an expression.rM) showrN verbositylimit include_span include_kindinclude_object_attrs union_attrs force_reloadFNr) rrvrrNrrrrrrrrrs r# do_inspectrss  YY,,jj&&&&!66$$&& H5D$Or$c :tjdk(rtjdd|vrt|d |d|d|d}}}t jjt jjt jjt jj|r t||r>dd l m }||zj}||d t||rd |ini||d |d |r=|j!di} t#|d5} t%j&| | d d d rt j(|y y #t$rtdt |YBwxYw#1swYIxYw)zkPrint the output from a check or recheck command. Call sys.exit() unless the status code is zero. ntroutrfr?z Response: r)write_junit_xmlrNpython_versionplatformstatsw)osnamesystemr_KeyErrorr%rZstdoutwriteflushrkr mypy.utilr splitlinesboolgetopenrdumprd) rrrrrrf status_codermessages telemetryfs r#rr"so ww$ " ( Xg + (%(8BT+SJJSJJJJSJJ8-#I))+ % & I (T8 b  % & Z  LL"- /3 ' $1 IIi # $ 5 + z#h- )*+. $ $sE*2F* F FFc &t|jD]u\}}|dvrEt|dd}t|dkDr|dddt|dz d}t |dd |Ot |dd t |t rd |zn|d wy) N)rrfrrk2(z ... z more characters24z: z%.3fz>10)sorteditemsreprlenrj isinstancefloat)rkeyvalues r#rrOsX^^-.\ U 4 4K"%E5zB ":,eCJrM?:JK e, -  sj6NFUNTY$YZ[\r$cFtt|jddy)z&Hang for 100 seconds, as a debug hack.rVrrN)rjrrvrxs r#do_hangr[s '$""FA 67r$c@ddlm}m}|jrvt |jddxt _t _t j j}tj|dtj|d|jrGddl m }tjt!|j}|}|j#|}n||j$d }|||j&|j( j+y ) z!Serve requests in the foreground.r)Serverrar) bufferingrY)OptionsFrrN)rrrrrrZrrkfilenordup2 options_data mypy.optionsrpickleloadsr apply_changesr:rvrserve)r`rrfdr options_dict options_objoptionss r# do_daemonras@ }}"&t}}cQ"GG SZ ZZ    A A (||Id.?.?$@A i ++L9' %H 7D$$dll;AACr$c,tjy)z'Print full help (same as dmypy --help).N)r\ print_helprxs r#do_helprys r$rc i}t|}||d<tjjxs t |d<t |d<t |\}} t||5}t||d} | st|}t|jdd} |jdd} | rtjj| |jdd} | rtjj| | sddd|S#1swY|SxYw#ttf$r} d t!| icYd} ~ Sd} ~ wwxYw) 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. commandis_ttyterminal_widthFfinalrNrkr)dictrZrisattyrrrurrr rpoprrkrrr%) rvrrkwdsrr`rrclientr rrkrfs r#rrs- "H :DDOZZ&&(@,>,@DN/1D %GAt# tW % -  E"6?X\\'59:"h5JJ$$V,!h5JJ$$V,  -( O) -( O \ "#S""#s= D(BDDDDDE. E:EEc.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)rvrs r#rurus { #D  r$cd|vr td|d}t|ts tdt|s tdd|vr td|d}t|ts td||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)rrintr r%)rrrs r#rrs D<== u+C c3 122 :)**$HII,-O os +?@@  r$c: t|y#t$rYywxYw)z&Check if the server is running cleanlyFT)rur)rvs r#rrs(;  s  c<ttjddy)Nr)rgrZrerr$r# console_entryrs!"r$)rez list[str]r&r')rlr%r&r)rqzargparse.ArgumentParserr&z*Callable[[ActionFunction], ActionFunction])r`argparse.Namespacer&r')F)r`rrrr&r')g@)rvr%rrr&r') rdict[str, Any]rrr str | Nonerrr&r')rzMapping[str, object]r&r') rvr%rr%rz int | Nonerobjectr&r)rvr%r&tuple[int, str])rrr&r)rvr%r&r)r&r')]__doc__ __future__rargparserrrrZrrbcollections.abcrrtypingrr librt.base64r mypy.defaultsr mypy.dmypy_osr r mypy.dmypy_utilr r rmypy.ipcrrrrrrrr mypy.versionrRawDescriptionHelpFormatterrArgumentParserr\ro add_argumentadd_subparsers subparsers add_parser start_parserpr%rrestart_parser status_parser stop_parser kill_parser check_parserSUPPRESS run_parserrecheck_parsersuggest_parserrinspect_parser hang_parser daemon_parser help_parserrgr_ Namespacersr1ryr|r{rwrrrr~rrrrrrrrrrrrurrrrr$r#r=s~  # - ")%>>DDRR$ DXAAD ! ;SV 4 07_  + % 1  " " $ ((~(FF q|V#B++"+0:+KQ++\ ,r$