U g.@sdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlZe rddlmZmZddlmZddlZd Zd Zd ZejGd d d ZdddddZdddddZdddddZdddddZd dddd d!Zd dddd"d#Zd dddd$d%Z d dd&d'd(Z!d)d*d+d,d-Z"d.d*d/d0d1d2Z#d/d3d4d5Z$d6Z%d7Z&d8Z'd9Z(d:Z)e*d;kre$dS)^sz+create_passthrough_args..*=z**z({})z, )argsvarargappendr( kwonlyargskwargformatjoin)r'Z call_argsr(rrrcreate_passthrough_argsVs r4ztuple[bool, str])filesourcercCsNddl}tjtjddd|jdg|ddd }|jdkrDd d |jfSd|jfS) a Run black on the specified file. Returns: Tuple of success and result string. ex.: (False, "Failed to run black! error: cannot format ...") (True, "") Raises: ImportError: If black is not installed. rN-mblack--stdin-filename-Tutf8inputcapture_outputencodingFzFailed to run black! ) r8 subprocessrunsys executabler returncodestderrstdout)r5r6r8resultrrr run_blackhs  rHc CsTddl}tjtjdddddd|jd g |d d d }|jdkrJd d|jfSd |jfS)aRun ruff on the specified file. Returns: Tuple of success and result string. ex.: (False, "Failed to run ruff! error: Failed to parse ...") (True, "") Raises: ImportError: If ruff is not installed. rNr7ruffcheckz--fixz--unsafe-fixesr9r:Tr;r<FzFailed to run ruff! ) rIr@rArBrCr rDrErF)r5r6rIrGrrrrun_ruffs&   rKcCspt||\}}|s$t|tdt||\}}|sHt|tdt||\}}|slt|td|S)zFormat the specified file using black and ruff. Returns: Formatted source code. Raises: ImportError: If either is not installed. SystemExit: If either failed. )rHprintrBexitrK)r5r6successresponserrr run_linterss    rQ)r5rc Cstg}|jr||j|jrZd|jkr2|dd|jkrF|d|d|jdd|g}tj|j}g}t |D]}|j j dj d kst |j j d=||j |jD]"}t|tjr|jd krd }qqd }g|_t|}t|d kr |jd d =n |jdd =tj|dd} |r6| dd} tt|tjrLdnd|j|j |} | t| d} || q||dd|d|S)zZScan the given .py file for @_public decorators, and generate wrapper functions. rz!from typing import TYPE_CHECKING z import sysz import sys z, assert not TYPE_CHECKING or sys.platform=="z" r rselfcontextmanagerTFNrLz )Z indent_withz ->Iteratorz->ContextManagerz await  z __all__ = z )HEADERrr/rr3astorZ code_to_astZ parse_filer r&r-r(AssertionErrornamerrrr r!r4Z get_docstringbodyZ to_sourcereplaceTEMPLATEr2rr rsortinsert) r5header generatedr6Z method_namesmethoddecZis_cmnew_argsfunctemplateZsnippetrrrgen_public_wrappers_sourcesR              rezdict[str, str]bool) new_filesrc CsV|D]H\}}tj|s"dSt|dd}|}W5QRX||krdSqdS)NFutf-8)r?T)itemsosr existsopenread)rgnew_path new_sourceZold_fileZ old_sourcerrrmatches_disk_filess rpzIterable[File]None)filesdo_testrc Csi}|D]N}td|jt|}t||}tj|j\}}tj|t|}|||<qt|}|r|s|tdt dqtdnN| D].\}}t |dddd} | |W5QRXqtd |st ddS) Nz Scanning:z2Generated sources are outdated. Please regenerate.rLz!Generated sources are up to date.wrh )r?newlinez!Regenerated sources successfully.)rMr rerQrjsplitr3PREFIXrprBrNrirlwrite) rrrsrgr5rodirnamebasenamernZ matches_diskfrrrprocess s(     r})rc Cstjdd}|jddddd|}t}|ds>t|d }t|d d t d t|d dt d t|dddt dt|dddt dt|dddt dg}t||jddS)Nz,Generate python code for public api wrappers) descriptionz--testz-t store_truez test if code is still up to date)actionhelpLICENSEzsrc/trio/_corez_run.pyrunner)rz_instrumentation.pyzrunner.instrumentsz_io_windows.pyzrunner.io_managerwin32)rrz _io_epoll.pylinuxz _io_kqueue.pydarwin)rs)argparseArgumentParser add_argument parse_argsrcwdrkrWr IMPORTS_RUNIMPORTS_INSTRUMENTIMPORTS_WINDOWS IMPORTS_EPOLLIMPORTS_KQUEUEr}test)parserZ parsed_argsZ source_rootcoreZto_wraprrrmain<sNraNfrom collections.abc import Awaitable, Callable from typing import Any, TYPE_CHECKING from outcome import Outcome import contextvars from ._run import _NO_SEND, RunStatistics, Task from ._entry_queue import TrioToken from .._abc import Clock if TYPE_CHECKING: from typing_extensions import Unpack from ._run import PosArgT z)from ._instrumentation import Instrument zZfrom typing import TYPE_CHECKING if TYPE_CHECKING: from .._file_io import _HasFileNo zfrom typing import Callable, ContextManager, TYPE_CHECKING if TYPE_CHECKING: import select from .. import _core from ._traps import Abort, RaiseCancelT from .._file_io import _HasFileNo zfrom typing import TYPE_CHECKING, ContextManager if TYPE_CHECKING: from .._file_io import _HasFileNo from ._windows_cffi import Handle, CData from typing_extensions import Buffer from ._unbounded_queue import UnboundedQueue __main__)+__doc__ __future__rrrrjr@rBpathlibrtextwraprtypingrrcollections.abcrrZtyping_extensionsrrVrxrUr[Zdefiner rr#r&r4rHrKrQrerpr}rrrrrrrrrrrsJ        &K .