U :xgE<@sUddlmZddlZddlZddlZddlZddlZddlZddlZ ddl m Z ddl m Z ddlmZeejejhZeejejhZeedreejeeZ[dd d d d d dhZddddZddddddZddddddZddddddZddddd Zd!dd"d#Z Gd$d%d%Z!Gd&d'd'e!Z"Gd(d)d)e!Z#e"e#d*Z$d+e%d,<z e&d-Wn"e'k re$d.e$d/<YnXe$d0e$d/<ddd1d2Z(d:d3d4d4d5d6dd7d8d9Z)dS);) annotationsN)chain)PurePath)_log real_prefix __pycache__z.gitz.hgz.toxz.noxz .pytest_cachez .mypy_cachezt.Iterator[str]returnccsfttjD]R}t|dd}|dks|tr2qtj |sZ|}tj |}||kr2qq2|VqdS)z;Find the filesystem paths associated with imported modules.__file__N) listsysmodulesvaluesgetattr startswith_ignore_alwaysospathisfiledirname)modulenameoldr6/tmp/pip-unpacked-wheel-5ifiwx3z/werkzeug/_reloader.py_iter_module_paths(s   rzset[str]None)pathsexclude_patternsr cCs |D]}|t||qdSN)difference_updatefnmatchfilter)rrpatternrrr_remove_by_pattern<sr%zt.Iterable[str]) extra_filesrr c Cst}tttj|D]}tj|}tj|r>||qtj |di}t |D]\}}}| t s|tj |tkr|qXd}|D]&} | drd}|tj|| q|s|tj |s|qX|||<qXq|tt|||S)axFind paths for the stat reloader to watch. Returns imported module files, Python files under non-system paths. Extra files and Python files under extra directories can also be scanned. System paths have to be excluded for efficiency. Non-system paths, such as a project root or ``sys.path.insert``, should be the paths of interest to the user anyway. TF)z.pyz.pyc)setrr r rrabspathraddrwalkr_stat_ignore_scanbasename_ignore_common_dirsclearendswithjoinupdaterr%) r&rrrZ parent_has_pyrootdirsfilesZhas_pyrrrr_find_stat_pathsAs4        r5cCszt}tttj|D]2}tj|}tj|r>tj|}| |qt D]}| tj|qPt ||t |S)zFind paths for the stat reloader to watch. Looks at the same sources as the stat reloader, but watches everything under directories instead of individual files. ) r'rr r rrr(rrr)rr%_find_common_roots)r&rr3rrrr_find_watchdog_pathsws      r7)rr cspi}tdd|DtddD]&}|}|D]}||i}q(|qtddddfd d |d S) Ncss|]}t|jVqdSr )rparts.0xrrr sz%_find_common_roots..T)keyreversez t.Mapping[str, dict[str, t.Any]]ztuple[str, ...]r)noderr cs@|D]\}}|||fq|s<|r<tjj|dSr )itemsr)rrr0)r?rprefixchild_walkrvrrrDsz!_find_common_roots.._walkr)sortedlen setdefaultr.r')rr2chunksr?chunkrrCrr6s  r6z list[str]cCstjdkr tjftjddStjg}tjd}tjdd}tjd}t|dddkstjdkr|j dkrtj |stj |d rtj |}tjdkr tj |stj |d r|d 7}tj tjdd kr tj |dd kr |d||nftj |rdtt|j }tj tj |d}|dkrh|d |7}n|}|d |d f|||S) zpDetermine how the script was executed, and return the args needed to execute it again in a new process. ) rNr__main__ __package__ntz.exe.z-m)r version_info executableZ orig_argvargvrrrrrNrexistsr(splitextpopappendrtcaststrr,extendlstrip)rEZ py_scriptargsrMZ py_modulerrrr_get_args_for_reloadingsD          r_c@seZdZdZdddddddd Zdd d d Zd dZdd ddZdd ddZdd ddZ dddddZ dddddZ dS) ReloaderLooprPNrt.Iterable[str] | None int | floatr)r&rintervalr cCs,dd|p dD|_t|pd|_||_dS)NcSsh|]}tj|qSr)rrr(r9rrr sz(ReloaderLoop.__init__..r)r&r'rrc)selfr&rrcrrr__init__szReloaderLoop.__init__r cCs ||S)zgDo any setup, then run one step of the watch to populate the initial filesystem state. )run_steprerrr __enter__szReloaderLoop.__enter__cCsdS)z4Clean up any resources associated with the reloader.Nrreexc_typeexc_valexc_tbrrr__exit__szReloaderLoop.__exit__cCs|t|jqdS)zfContinually run the watch step, sleeping for the configured interval after each step. N)rgtimesleeprcrhrrrrunszReloaderLoop.runcCsdS)z}Run one step for watching the filesystem. Called once to set up initial state, then repeatedly to update it. NrrhrrrrgszReloaderLoop.run_stepintcCsLtdd|jt}tj}d|d<tj||dd}|dkr|SqdS) z|Spawn a new Python interpreter with the same arguments as the current one, but running the reloader thread. infoz * Restarting with trueWERKZEUG_RUN_MAINF)env close_fdsrKN)rrr_renvironcopy subprocesscall)rer^Z new_environZ exit_coderrrrestart_with_reloader s z"ReloaderLoop.restart_with_reloaderr[filenamer cCs||tddSNrK) log_reloadr exitrer~rrrtrigger_reloads zReloaderLoop.trigger_reload str | bytescCs"tj|}tdd|ddS)Nrsz * Detected change in z , reloading)rrr(rrrrrrs zReloaderLoop.log_reload)NNr) __name__ __module__ __qualname__rrfrirnrqrgr|rrrrrrr`s r`cs4eZdZdZddfdd ZddddZZS) StatReloaderLoopstatr`r csi|_tSr )mtimessuperrirh __class__rrri$szStatReloaderLoop.__enter__rc Cstt|j|jD]`}zt|j}Wntk r:YqYnX|j|}|dkr\||j|<q||kr| |qdSr ) r5r&rrrst_mtimeOSErrorrgetr)rermtimeZold_timerrrrg(s   zStatReloaderLoop.run_step)rrrrrirg __classcell__rrrrr!srcsleZdZddddfdd Zddddd Zd d fd d ZddZdd ddZdd ddZZ S)WatchdogReloaderLoopzt.Anyr)r^kwargsr csddlmddlmddlmddlmddlmddlmddlm}dd lm }t j |||j Gfd d d |}|j }|d r|dd }d|d|_||_dd|jD}|d|ddtD|jd|_d|_dS)Nr)EVENT_TYPE_CLOSED)EVENT_TYPE_CREATED)EVENT_TYPE_DELETED)EVENT_TYPE_MODIFIED)EVENT_TYPE_MOVED)FileModifiedEvent)PatternMatchingEventHandler)Observercs(eZdZddfdd ZdS)z3WatchdogReloaderLoop.__init__..EventHandlerr)eventcs&|jhkrdS|jdSr )Z event_typeZsrc_path)rer)rrrrrrrr on_any_eventHsz@WatchdogReloaderLoop.__init__..EventHandler.on_any_eventN)rrrrrrrrrrrrrr EventHandlerGsrobserveriz watchdog ()cSsg|]}tj|s|qSr)rrisdir)r:prrr as z1WatchdogReloaderLoop.__init__..*.py*.pyc*.zipcSsg|]}d|dqS)z*/z/*r)r:drrrres)patternsignore_patternsF)rrr)Zwatchdog.eventsrrrrrrrZwatchdog.observersrrrfrrlowerr/rrr&r-r event_handler should_reload)rer^rrrrZ reloader_nameZextra_patternsrrrrf:s0             zWatchdogReloaderLoop.__init__rr}cCsd|_||dS)NT)rrrrrrrksz#WatchdogReloaderLoop.trigger_reloadr`r csi|_|jtSr )watchesrstartrrirhrrrrirs zWatchdogReloaderLoop.__enter__cCs|j|jdSr )rstopr0rjrrrrnws zWatchdogReloaderLoop.__exit__cCs*|js|t|jqtddSr)rrgrorprcr rrhrrrrq{szWatchdogReloaderLoop.runc Cst|j}t|j|jD]V}||jkrdz|jj|j|dd|j|<Wntk rbd|j|<YnX| |q|D]&}|j |d}|dk rt|j |qtdS)NT) recursive) r'rr7r&rrschedulerrdiscardrWZ unschedule)reZ to_deleterZwatchrrrrgs    zWatchdogReloaderLoop.run_step) rrrrfrrirnrqrgrrrrrr9s 1r)rwatchdogzdict[str, type[ReloaderLoop]]reloader_loopszwatchdog.observersrautorcCs~tjdkstjsdSz ddl}Wntk r:YdSX|tj}|d|j@sz|d|jO<|tj|j|dS)zuEnsure that echo mode is enabled. Some tools such as PDB disable it which causes usability issues after a reload.NrrK) r stdinisattytermios ImportError tcgetattrECHO tcsetattrTCSANOW)r attributesrrrensure_echo_ons  rzt.Callable[[], None]rarbr[) main_funcr&rrc reloader_typer c Csddl}||jddt||||d}z^tjddkrzttj|dd }d |_ || | W5QRXnt |Wntk rYnXdS) zz#run_with_reloader..)r&rrcrurtr)targetr^T)signalSIGTERMrrrxrr threadingThreaddaemonrrqr rr|KeyboardInterrupt)rr&rrcrrZreloaderrYrrrrun_with_reloaders$r)NNrr)* __future__rr"rrzr rrotypingrY itertoolsrpathlibr _internalrtuple base_prefixbase_exec_prefixrrA exec_prefixhasattrr)rr+r-rr%r5r7r6r_r`rrr__annotations__ __import__rrrrrrrs`       6;=b