L-PhdZddlmZddlZddlZddlmZddlmZej GddZ Gdd Z ddZ dS)z*Find functions and classes in Python code.) annotationsN)cast) CodeRegionc2eZdZUdZded<ded<ded<dS)Contextz0The nested named context of a function or class.strnamekindzset[int]linesN)__name__ __module__ __qualname____doc____annotations__P/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/coverage/regions.pyrrs1:: III IIIOOOOOrrcJeZdZdZddZddZddZdd Zdd ZddZ ddZ dS) RegionFinderzAn ast visitor that will find and track regions of code. Functions and classes are tracked by name. Results are in the .regions attribute. returnNonec"g|_g|_dSN)regionscontextselfs r__init__zRegionFinder.__init__s)+ &( rsourcercT|tj|dS)zCParse `source` and walk the ast to populate the .regions attribute.N) handle_nodeastparse)rrs r parse_sourcezRegionFinder.parse_source#s& 6**+++++rcJdd|jDS)z6Get the current fully qualified name we're processing..c3$K|] }|jV dSr)r ).0cs r z,RegionFinder.fq_node_name..)s$551555555r)joinrrs r fq_node_namezRegionFinder.fq_node_name's%xx55 555555rnodeast.ASTc t|tjtjfr||dSt|tjr||dS||dS)zRecursively handle any node.N) isinstancer" FunctionDefAsyncFunctionDefhandle_FunctionDefClassDefhandle_ClassDefhandle_node_body)rr-s rr!zRegionFinder.handle_node+s dS_c.BC D D (  # #D ) ) ) ) ) cl + + (   & & & & &  ! !$ ' ' ' ' 'rcXt|ddD]}||dS)z9Recursively handle the nodes in this node's body, if any.bodyrN)getattrr!)rr- body_nodes rr6zRegionFinder.handle_node_body4s> vr22 ( (I   Y ' ' ' ' ( (r&ast.FunctionDef | ast.AsyncFunctionDefc tt|jdjt t |jdjdz}|jr1|jdjdkr|jdxj |zc_ t|jD]}|jdkr|xj |zc_ n |j t|j d||j td||j||||jdS)z Called for `def` or `async def`.rclassfunctionr r startr N)setranger8linenorint end_linenorr r reversedappendrr rrr,r6poprr-r ancestors rr3zRegionFinder.handle_FunctionDef9sPE$)A,-tC29Q/R/RUV/VWWXX < ,DL,1W<< L  " "e + " " ..  H} **%'+ GDIz5AABBB  &&((k        d### r ast.ClassDefct}|jt|jd||jt d||j|| ||j t|jD]}|j dkr|xj |zc_ dS)zCalled for `class`.r?rAN)rCrrIrr rrr,rEr6rJrHr r rKs rr5zRegionFinder.handle_ClassDefPs %% GDIw>>???  &&((k        d###  .. ( (H}''%' ( (rN)rr)rrrr)rr)r-r.rr)r-r;rr)r-rMrr) r r rrrr$r,r!r6r3r5rrrrrs )))),,,,6666(((((((( .((((((rrrrrlist[CodeRegion]cVt}|||jS)a]Find function and class regions in source code. Analyzes the code in `source`, and returns a list of :class:`CodeRegion` objects describing functions and classes as regions of the code:: [ CodeRegion(kind="function", name="func1", start=8, lines={10, 11, 12}), CodeRegion(kind="function", name="MyClass.method", start=30, lines={34, 35, 36}), CodeRegion(kind="class", name="MyClass", start=25, lines={34, 35, 36}), ] The line numbers will include comments and blank lines. Later processing will need to ignore those lines as needed. Nested functions and classes are excluded from their enclosing region. No line should be reported as being part of more than one function, or more than one class. Lines in methods are reported as being in a function and in a class. )rr$r)rrfs r code_regionsrRgs'* BOOF :r)rrrrO) r __future__rr" dataclassestypingrcoverage.pluginr dataclassrrrRrrrrXs10"""""" &&&&&& L(L(L(L(L(L(L(L(^r