`i |dZddlZddlZddlZ ddlmZn #e$reZYnwxYwGddeZ e Z dS)z Part of the astor library for Python AST manipulation. License: 3-clause BSD Copyright (c) 2012-2015 Patrick Maupin Copyright (c) 2013-2015 Berker Peksag Functions that interact with the filesystem go here. N)opencdeZdZdZeddZedZedZddZdZ dS) CodeToAstawGiven a module, or a function that was compiled as part of a module, re-compile the module into an AST and extract the sub-AST for the function. Allow caching to reduce number of compiles. Also contains static helper utility functions to look for python files, to parse python files, and to extract the file/line information from a code object. Nc#Ktj|s!tj|Vtj|D]"\}}}|||vr d|DD]}||fV #dS)zReturn all the python files in a source tree Ignores any path that contains the ignore string This is not used by other class methods, but is designed to be used in code that uses this class. Nc3DK|]}|d|VdS).pyN)endswith).0xs c/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/astor/file_util.py z*CodeToAst.find_py_files..3s3AAqzz%/@/@A!AAAAAA)ospathisdirsplitwalk)srctreeignoresrcpath_fnamesfnames r find_py_fileszCodeToAst.find_py_files#sw}}W%% )'--(( ( ( ("$''"2"2 % % GQ!f&7&7AAVAAA % %un$$$$ %  % %rc t|5}|}dddn #1swxYwYnT#t$rG|dkrtjdtj}YnwxYw|dddd}|ds|dz }tj ||S)zParse a python file into an AST. This is a very thin wrapper around ast.parse TODO: Handle encodings other than the default for Python 2 (issue #26) Nstdinz Reading from stdin: z   )filename) fopenreadIOErrorsysstdoutwriterreplacer astparse)rffstrs r parse_filezCodeToAst.parse_file6s $u vvxx                $ $ $ J  8 9 9 99>>##DDD  $ ||FD))11$==}}T""  DLDy....s(>2 >6>6>ABBct|dd}d}||j}|j}|j}|dd}||fS)zReturns the file and line number of a code object. If the code object has a __file__ attribute (e.g. if it is a module), then the returned line number will be 0 __file__Nrz.pycr)getattr__code__ co_filenameco_firstlinenor&)codeobjrlinenum func_codes r get_file_infozCodeToAst.get_file_infoLsTT22 =(I)E.G fe,,g~rc|pi|_dSN)cache)selfr8s r __init__zCodeToAst.__init__]s[b rc&|j}||}||}||S|d}||x||df<}|jD])}t |t js||||jf<*||S)Nr) r8r5getr+body isinstancer' FunctionDeflineno)r9r2r8keyresultrmod_astobjs r __call__zCodeToAst.__call__`s   ))3  MA&*ooe&<&<<uajG< - -Cc3?33 ),E5#*% & &Szrr7) __name__ __module__ __qualname____doc__ staticmethodrr+r5r:rErr rrs%%%\%$//\/*\ !!!!     rr) rIr'r#rtokenizerr ImportErrorobjectr code_to_astrKrr rPs   &&&&&&& EEETTTTTTTTnikk s !!