U Ծ|eO @s2dZddlZGdddejZddZddZdS) zR This module provides helper functions to find the first line of a function body. Ncs8eZdZdZddZfddZejdddZZ 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 W/var/www/website-v5/atlas_env/lib/python3.8/site-packages/numba/misc/firstlinefinder.py__init__szFindDefFirstLine.__init__cs t|D]}t|q dS)N)astiter_child_nodessupervisit)rnodechild __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) namersetlinenodecorator_listaddrbody _is_docstringrr)rrZpossible_start_linesZ first_decorZ first_stmtr r r visit_FunctionDef!s       z"FindDefFirstLine.visit_FunctionDef) __name__ __module__ __qualname____doc__r rr FunctionDefr __classcell__r r rr r s rcCs0t|tjr,t|jtjr,t|jjtr,dSdS)NTF) isinstancer ExprvalueConstantstrrr r r r?s   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_filenamereadFileNotFoundErrorOSErrorr parserrr)pyfunccoZfinZ file_contenttreefinderr r r get_func_body_first_linenoGs    r6)r#r NodeVisitorrrr6r r r r s6