autograder.cli.util.extract-code

 1# pylint: disable=invalid-name
 2
 3"""
 4Pull the code from a file and output the sanitized version as if it was being graded.
 5"""
 6
 7import argparse
 8import sys
 9
10import edq.util.code
11import edq.util.dirent
12
13import autograder.cli.parser
14
15def run_cli(args: argparse.Namespace) -> int:
16    """ Run the CLI. """
17
18    text = edq.util.code.extract_code(args.path)
19    ast = edq.util.code.parse_module_code(text)
20    source = edq.util.code.ast_to_source(ast)
21
22    if (args.out_path is None):
23        print(source)
24        return 0
25
26    edq.util.dirent.write_file(args.out_path, source)
27
28    return 0
29
30def main() -> int:
31    """ Get a parser, parse the args, and call run. """
32    return run_cli(_get_parser().parse_args())
33
34def _get_parser() -> argparse.ArgumentParser:
35    parser = autograder.cli.parser.get_parser(__doc__.strip())
36
37    parser.add_argument('path', metavar = 'PATH',
38        action = 'store', type = str,
39        help = 'Path to the file to extract code from.')
40
41    parser.add_argument('-o', '--out-path', dest = 'out_path',
42        action = 'store', type = str, default = None,
43        help = 'If specified, the code will be written here instead of output on stdout.')
44
45    return parser
46
47if (__name__ == '__main__'):
48    sys.exit(main())