o ҷh-@sDdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlZe rBddlmZmZddlmZddlZd Zd Zd ZejGd d d Zd=ddZd=ddZd>ddZd?ddZd@d"d#Zd@d$d%ZdAd&d'Z dBd(d)Z!dCd-d.Z"dDd3d4Z#dEd5d6Z$d7Z%d8Z&d9Z'd:Z(d;Z)e*d]sz+create_passthrough_args..*=z**z({})z, )argsvarargappendr0 kwonlyargskwargformatjoin)r. call_argsr0rrrcreate_passthrough_argsUs r=filesourcetuple[bool, str]cCsNddl}tjtjddd|jdg|ddd }|jdkr"d 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! ) rB subprocessrunsys executabler returncodestderrstdout)r>r?rBresultrrr run_blackgs   rRc CsTddl}tjtjdddddd|jd g |d d d }|jdkr%d 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. rNrAruffcheckz--fixz--unsafe-fixesrCrDTrErFFzFailed to run ruff! ) rSrJrKrLrMr rNrOrP)r>r?rSrQrrrrun_ruffs&    rUcCs6ttfD]}|||\}}|st|tdq|S)zFormat the specified file using black and ruff. Returns: Formatted source code. Raises: ImportError: If either is not installed. SystemExit: If either failed. )rRrUprintrLexit)r>r?fnsuccessrrr run_linterss  r[c Cstg}||j|jr*d|jvr|dd|jvr |d|d|jdd|g}tj|j}g}t |D]|}|j j dj d ksJJ|j j d=||j |j D]}t|tjri|jd krid }nqXd }td g|_ t|}t|dur|jdd=n|jdd=tj|dd} |r| dd} tt|tjrdnd|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" rrselfcontextmanagerTFenable_ki_protectionNrVz ) indent_withz ->Iteratorz->AbstractContextManagerz await  z __all__ = z )HEADERr7rrr;astor code_to_ast parse_filer r-r5r0namer%rr r&r'r= get_docstringbody to_sourcereplaceTEMPLATEr:r"r rsortinsert) r>header generatedr? method_namesmethoddecis_cmnew_argsfunctemplatesnippetrrrgen_public_wrappers_sourcesR               rw new_filesdict[str, str]boolcCsF|D]\}}tj|sdSt|jdd}||kr dSqdS)NFutf-8)rIT)itemsosr existsr read_text)rxnew_path new_source old_sourcerrrmatches_disk_files s rfilesIterable[File]do_testNonec Csi}|D]'}td|jt|}t||}tj|j\}}tj|t|}|||<qt|}|rE|s?tdt ddStddS| D]!\}}t |dddd } | |Wdn1sewYqItd |sxt ddSdS) Nz Scanning:z2Generated sources are outdated. Please regenerate.rVz!Generated sources are up to date.wr{ )rInewlinez!Regenerated sources successfully.)rWr rwr[r}splitr;PREFIXrrLrXr|openwrite) rrrxr>rdirnamebasenamer matches_diskfprrrprocesss.     rc Cstjdd}|jddddd|}t}|dsJ|d }t|d d td 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)r)argparseArgumentParser add_argument parse_argsrcwdr~r IMPORTS_RUNIMPORTS_INSTRUMENTIMPORTS_WINDOWS IMPORTS_EPOLLIMPORTS_KQUEUErtest)parser parsed_args source_rootcoreto_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 a from typing import TYPE_CHECKING if TYPE_CHECKING: import select from collections.abc import Callable from contextlib import AbstractContextManager from .. import _core from .._file_io import _HasFileNo from ._traps import Abort, RaiseCancelT afrom typing import TYPE_CHECKING if TYPE_CHECKING: from contextlib import AbstractContextManager from typing_extensions import Buffer from .._file_io import _HasFileNo from ._unbounded_queue import UnboundedQueue from ._windows_cffi import Handle, CData __main__)rrrr)r*rrr+)r.r/rr )r>r r?r rr@)r>r r?r rr )r>r rr )rxryrrz)rrrrzrr)rr)+__doc__ __future__rrr r}rJrLpathlibrtextwraprtypingrrcollections.abcrrtyping_extensionsrrbrrarjdefiner r#r)r-r=rRrUr[rwrrrrrrrrrrrrrsN           &  I .