U
|
Z ±d ã @ s° d Z ddlZddlZddlZddlZddlmZ ddlZddlm Z dZ
|
dZ dZ dZ d d
|
Zd d Zd d Zdd ZeeedddZdd Zdd Zdd Zedkr¬e dS )zG
|
Code generation script for class methods
|
to be exported as public API
|
é N)ÚPath)ÚindentZ
|
_generatedaT # ***********************************************************
|
# ******* WARNING: AUTOGENERATED! ALL EDITS WILL BE LOST ******
|
# *************************************************************
|
from ._run import GLOBAL_RUN_CONTEXT, _NO_SEND
|
from ._ki import LOCALS_KEY_KI_PROTECTION_ENABLED
|
from ._instrumentation import Instrument
|
|
# fmt: off
|
z
|
# fmt: on
|
z¯locals()[LOCALS_KEY_KI_PROTECTION_ENABLED] = True
|
try:
|
return{}GLOBAL_RUN_CONTEXT.{}.{}
|
except AttributeError:
|
raise RuntimeError("must be called from async context")
|
c C s t | tjst | tjr dS dS )zHCheck if the AST node is either a function
|
or an async function
|
TF)Ú
|
isinstanceÚastZ FunctionDefÚAsyncFunctionDef)Únode© r úNd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_tools/gen_exports.pyÚ is_function' s r
|
c C s8 t | s dS | jD ] }t|tjr|jdkr dS qdS )z-Check if the AST node has a _public decoratorFZ_publicT)r
|
Údecorator_listr r ÚNameÚid)r Ú decoratorr r r Ú is_public0 s
|
r c c s" t | ¡D ]}t|r
|
|V q
|
dS )z¢Return a list of methods marked as public.
|
The function walks the given tree and extracts
|
all objects that are functions which are marked
|
public.
|
N)r Úwalkr )Útreer r r r Úget_public_methods: s r c C s g }| j j D ]}| |j¡ q | j jr:| d| j jj ¡ | j jD ]}| |jd |j ¡ qB| j jrz| d| j jj ¡ d d |¡¡S )a Given a function definition, create a string that represents taking all
|
the arguments from the function, and passing them through to another
|
invocation of the same function.
|
|
Example input: ast.parse("def f(a, *, b): ...")
|
Example output: "(a, b=b)"
|
Ú*ú=z**z({})z, )ÚargsÚappendÚargÚvarargÚ
|
kwonlyargsÚkwargÚformatÚjoin)ZfuncdefZ call_argsr r r r Úcreate_passthrough_argsE s r )Ú source_pathÚ lookup_pathÚreturnc C sÔ t g}tj | ¡}t|D ]¤}|jjd jdks4t|jjd= g |_t |}t
|
|¡dkrh|j dd
= n |j dd
= tj |dd}t t|t
|
jrdnd||j| ¡}|t|d }| |¡ q| t¡ d |¡S )
|
zZScan the given .py file for @_public decorators, and generate wrapper
|
functions.
|
|
r ÚselfNé z )Z indent_withz await ú z
|
|
)ÚHEADERÚastorZ code_to_astZ
|
parse_filer r r ÚAssertionErrorr r r Z get_docstringÚbodyZ to_sourceÚTEMPLATEr r r Únamer r ÚFOOTERr ) r r Ú generatedÚsourceÚmethodÚnew_argsÚfuncÚtemplateZsnippetr r r Úgen_public_wrappers_sourceY s(
|
ý
|
r1 c
|
C sX | ¡ D ]J\}}tj |¡s" dS t|ddd}| ¡ }W 5 Q R X ||kr dS qdS )NFÚrúutf-8©ÚencodingT)ÚitemsÚosÚpathÚexistsÚopenÚread)Ú new_filesÚnew_pathÚ
|
new_sourceZold_fileZ
|
old_sourcer r r Úmatches_disk_files
s r? c
|
|
C s¼ i }| D ]F\}}t d| t||}tj |¡\}}tj |t| ¡}|||< q|rzt|spt d t d¡ q¸t d n>|
|
¡ D ],\}}t |ddd} | |¡ W 5 Q R X qt d d S ) Nz Scanning:z2Generated sources are outdated. Please regenerate.r" z!Generated sources are up to date.Úwr3 r4 z!Regenerated sources successfully.) Úprintr1 r7 r8 Úsplitr ÚPREFIXr? ÚsysÚexitr6 r: Úwrite)
|
Zsources_and_lookupsÚdo_testr< r r r> ÚdirnameÚbasenamer= Úfr r r Úprocess s
|
|
|
|
rK c C s t jdd} | jddddd | ¡ }t ¡ }|d ¡ s>t|d }|d
|
d f|d d f|d df|d df|d dfg}t||j d d S )Nz,Generate python code for public api wrappers)Ú descriptionz--testz-tÚ
|
store_truez test if code is still up to date)ÚactionÚhelpÚLICENSEz
|
trio/_corez_run.pyÚrunnerz_instrumentation.pyzrunner.instrumentsz_io_windows.pyzrunner.io_managerz _io_epoll.pyz _io_kqueue.py)rG )
|
ÚargparseÚArgumentParserÚ add_argumentÚ
|
parse_argsr Úcwdr9 r& rK Útest)ÚparserZ parsed_argsZ source_rootÚcoreZto_wrapr r r Úmain§ s( ÿ ÿ
|
|
|
|
|
ûrZ Ú__main__)Ú__doc__rR r r% r7 Úpathlibr rD Útextwrapr rC r$ r* r( r
|
r r r Ústrr1 r? rK rZ Ú__name__r r r r Ú<module> s*
|
|
,
|
