oPj~dZddlmZddlZddlmZddlmZddlm Z dZ eGdd Z Gd d Z d d Z y)z*Find functions and classes in Python code.) annotationsN) dataclass)cast) CodeRegion)bodyorelse finalbodyhandlerscasesc0eZdZUdZded<ded<ded<y)Contextz0The nested named context of a function or class.strnamekindzset[int]linesN)__name__ __module__ __qualname____doc____annotations__U/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/coverage/regions.pyr r s: I I Orr cHeZdZdZd dZd dZd dZd dZd dZddZ ddZ y ) RegionFinderzAn ast visitor that will find and track regions of code. Functions and classes are tracked by name. Results are in the .regions attribute. c g|_g|_yN)regionscontextselfs r__init__zRegionFinder.__init__'s)+ &( rcL|jtj|y)zCParse `source` and walk the ast to populate the .regions attribute.N) handle_nodeastparse)r!sources r parse_sourcezRegionFinder.parse_source+s 6*+rcFdjd|jDS)z6Get the current fully qualified name we're processing..c34K|]}|jywr)r).0cs r z,RegionFinder.fq_node_name..1s515s)joinrr s r fq_node_namezRegionFinder.fq_node_name/sxx5 555rct|tjtjfr|j |yt|tj r|j |y|j|y)zRecursively handle any node.N) isinstancer% FunctionDefAsyncFunctionDefhandle_FunctionDefClassDefhandle_ClassDefhandle_node_body)r!nodes rr$zRegionFinder.handle_node3sS dS__c.B.BC D  # #D ) cll +   &  ! !$ 'rc`tD]%}t||dD]}|j|'y)zDRecursively handle the nodes in this node's statement lists, if any.rN)STATEMENT_FIELDSgetattrr$)r!r9field stmts_nodes rr8zRegionFinder.handle_node_body<s7% -E%dE26 -   , - -rc tt|jdjt t |jdj dz}|jr>|jdjdk(r"|jdxj|zc_ t|jD](}|jdk(s|xj|zc_ n|jjt|jd||jjtd|j!|j||j#||jj%y)z Called for `def` or `async def`.rclassfunctionrrstartrN)setrangerlinenorint end_linenorrrreversedappendr rrrr0r8popr!r9rancestors rr5zRegionFinder.handle_FunctionDefBsE$))A,--tC29Q9Q/RUV/VWX <?  &&(kk    d#  . (H}}'%' (rN)returnNone)r'rrQrR)rQr)r9zast.ASTrQrR)r9z&ast.FunctionDef | ast.AsyncFunctionDefrQrR)r9z ast.ClassDefrQrR) rrrrr"r(r0r$r8r5r7rrrrrs*),6(- .(rrcPt}|j||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)r'rfs r code_regionsrUps!* BOOF ::r)r'rrQzlist[CodeRegion])r __future__rr% dataclassesrtypingrcoverage.pluginrr;r rrUrrrrZsO1# !& H  N(N(br