U ,d @s>dZddlZddlmZGdddejZddZdd ZdS) zR This module provides helper functions to find the first line of a function body. N) PYVERSIONcs8eZdZdZddZfddZejdddZZ S) FindDefFirstLinez Attributes ---------- first_stmt_line : int or None This stores the first statement line number if the definition is found. Or, ``None`` if the definition is not found. cCs|j|_|j|_d|_dS)zf Parameters ---------- code : The function's code object. N)co_name_co_nameco_firstlineno_co_firstlinenofirst_stmt_line)selfcoder >/tmp/pip-unpacked-wheel-eu7e0c37/numba/misc/firstlinefinder.py__init__szFindDefFirstLine.__init__cs t|D]}t|q dS)N)astZiter_child_nodessupervisit)r nodechild __class__r r _visit_childrensz FindDefFirstLine._visit_childrenrcCsz|j|jkrlt|jg}|jr4|jd}||j|j|krl|jrl|jd}t|r`|jd}|j|_ dS| |dS)Nr) namersetlinenoZdecorator_listaddrbody _is_docstringrr)r rZpossible_start_linesZ first_decorZ first_stmtr r r visit_FunctionDef#s       z"FindDefFirstLine.visit_FunctionDef) __name__ __module__ __qualname____doc__r rrZ FunctionDefr __classcell__r r rr r s rcCsFt|tjrBtdkr"t|jtjSt|jtjrBt|jjtrBdSdS)N)TF) isinstancerZExprrvalueZStrConstantstrrr r r rAs  rc Csj|j}z"t|j}|}W5QRXWnttfk rBYdSXt|}t|}| ||j SdS)a  Look up the first line of function body using the file in ``pyfunc.__code__.co_filename``. Returns ------- lineno : int; or None The first line number of the function body; or ``None`` if the first line cannot be determined. N) __code__open co_filenamereadFileNotFoundErrorOSErrorrparserrr)ZpyfunccoZfinZ file_contenttreefinderr r r get_func_body_first_linenoKs    r4)r"rZnumba.core.utilsrZ NodeVisitorrrr4r r r r s  6