mirror of https://github.com/chriskiehl/Gooey.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
117 lines
3.5 KiB
117 lines
3.5 KiB
'''
|
|
Created on Dec 11, 2013
|
|
|
|
@author: Chris
|
|
|
|
Collection of functions for extracting argparse related statements from the
|
|
client code.
|
|
'''
|
|
|
|
import os
|
|
import ast
|
|
import _ast
|
|
from itertools import chain
|
|
|
|
import codegen
|
|
from monkey_parser import MonkeyParser
|
|
from gooey.gui.action_sorter import ActionSorter
|
|
from parser_exceptions import ParserError
|
|
|
|
|
|
def parse_source_file(file_name):
|
|
"""
|
|
Parses the AST of Python file for lines containing
|
|
references to the argparse module.
|
|
|
|
returns the collection of ast objects found.
|
|
|
|
Example client code:
|
|
|
|
1. parser = ArgumentParser(desc="My help Message")
|
|
2. parser.add_argument('filename', help="Name of the file to load")
|
|
3. parser.add_argument('-f', '--format', help='Format of output \nOptions: ['md', 'html']
|
|
4. args = parser.parse_args()
|
|
|
|
Variables:
|
|
* nodes Primary syntax tree object
|
|
* argparse_assignments The assignment of the ArgumentParser (line 1 in example code)
|
|
* add_arg_assignments Calls to add_argument() (lines 2-3 in example code)
|
|
* parser_var_name The instance variable of the ArgumentParser (line 1 in example code)
|
|
* ast_source The curated collection of all parser related nodes in the client code
|
|
"""
|
|
|
|
nodes = ast.parse(_openfile(file_name))
|
|
|
|
module_imports = get_nodes_by_instance_type(nodes, _ast.Import)
|
|
specific_imports = get_nodes_by_instance_type(nodes, _ast.ImportFrom)
|
|
|
|
assignment_objs = get_nodes_by_instance_type(nodes, _ast.Assign)
|
|
call_objects = get_nodes_by_instance_type(nodes, _ast.Call)
|
|
|
|
argparse_assignments = get_nodes_by_containing_attr(assignment_objs, 'ArgumentParser')
|
|
add_arg_assignments = get_nodes_by_containing_attr(call_objects, 'add_argument')
|
|
# parse_args_assignment = get_nodes_by_containing_attr(call_objects, 'parse_args')
|
|
|
|
ast_argparse_source = chain(
|
|
module_imports,
|
|
specific_imports,
|
|
argparse_assignments,
|
|
add_arg_assignments
|
|
# parse_args_assignment
|
|
)
|
|
# for i in ast_argparse_source:
|
|
# print i
|
|
return ast_argparse_source
|
|
|
|
def _openfile(file_name):
|
|
with open(file_name, 'rb') as f:
|
|
return f.read()
|
|
|
|
def get_nodes_by_instance_type(nodes, object_type):
|
|
return [node for node in walk_tree(nodes) if isinstance(node, object_type)]
|
|
|
|
def get_nodes_by_containing_attr(nodes, attr):
|
|
return [node for node in nodes if attr in walk_tree(node)]
|
|
|
|
def walk_tree(node):
|
|
yield node
|
|
d = node.__dict__
|
|
for key, value in d.iteritems():
|
|
if isinstance(value, list):
|
|
for val in value:
|
|
for _ in walk_tree(val): yield _
|
|
elif 'ast' in str(type(value)):
|
|
for _ in walk_tree(value): yield _
|
|
else:
|
|
yield value
|
|
|
|
|
|
def convert_to_python(ast_source):
|
|
"""
|
|
Converts the ast objects back into human readable Python code
|
|
"""
|
|
return map(codegen.to_source, ast_source)
|
|
|
|
def extract_parser(modulepath):
|
|
ast_source = parse_source_file(modulepath)
|
|
if ast_source:
|
|
python_code = convert_to_python(ast_source)
|
|
return MonkeyParser(python_code)
|
|
return None
|
|
|
|
if __name__ == '__main__':
|
|
filepath = os.path.join(os.path.dirname(__file__),
|
|
'mockapplications',
|
|
'example_argparse_souce_in_main.py')
|
|
|
|
nodes = ast.parse(_openfile(filepath))
|
|
#
|
|
ast_source = parse_source_file(filepath)
|
|
python_code = convert_to_python(list(ast_source))
|
|
for i in python_code: print i
|
|
# parser = MonkeyParser(python_code)
|
|
# factory = ActionSorter(parser._actions)
|
|
# print factory._positionals
|
|
|
|
|
|
|