def test_take_imports_drops_all_non_imports_statements(self):
lines = '''
import argparse
from argparse import ArgumentParser
parser = ArgumentParser(description='Example Argparse Program')
parser.parse_args()
'''.split('\n')[1:]
self.assertEqual(2, len(list(code_prep.take_imports(lines))))
self.assertEqual('import argparse', list(code_prep.take_imports(lines))[0])
def _build_argparser_from_client_source(self, source_code):
'''
runs the client code by evaling each line.
Example input Code:
parser = ArgumentParser(description=program_license, formatter_class=RawDescriptionHelpFormatter)
parser.add_argument("-r", "--recursive", dest="recurse", action="store_true", help="recurse into subfolders [default: %(default)s]")
parser.add_argument("-v", "--verbose", dest="verbose", action="count", help="set verbosity level [default: %(default)s]")
Method extracts the instance name (e.g. parser) from the first line,
and instantiates it in a local variable by evaling the rest of the lines.
Each subsequent line updates the local variable in turn.
'''
imports = filter(lambda x: 'gooey' not in x, code_prep.take_imports(source_code))
arg_code = code_prep.drop_imports(source_code)
updated_source_code = code_prep.update_parser_varname('clients_parser', arg_code)
for _import in imports:
exec(_import)
first_line = updated_source_code.pop(0)
clients_parser, assignment = code_prep.split_line(first_line)
clients_parser = eval(assignment)
for line in updated_source_code:
eval(line)
return clients_parser