OpenDroneMap-WebODM/app/scripts/extract_odm_strings.py

#!/usr/bin/python3

import argparse, os, urllib.request, ast, sys
from io import StringIO

# Misc variables needed to get config to run
__version__ = '?'
class context:    
    root_path = ''
    num_cores = 4
class io:
    def path_or_json_string_to_dict(s):
        return s
def path_or_json_string(s):
    return s
class log:
    def ODM_ERROR(s):
        pass

def extract_odm_strings(url, outfile):
    strings = []
    print("Fetching %s ..." % url)
    res = urllib.request.urlopen(url)
    config_file = res.read().decode('utf-8')
    # config_file = open("test.py").read()

    options = {}
    class ArgumentParserStub(argparse.ArgumentParser):
        def add_argument(self, *args, **kwargs):
            argparse.ArgumentParser.add_argument(self, *args, **kwargs)
            options[args[0]] = {}
            for name, value in kwargs.items():
                options[args[0]][str(name)] = str(value)
        
        def add_mutually_exclusive_group(self):
            return ArgumentParserStub()

    # Voodoo! :)
    # ( parse AST, extract "def config()" function, set module to only
    # contain that function, execute module in current scope, 
    # run config function)
    root = ast.parse(config_file)
    new_body = []
    for stmt in root.body:
        # Assignments
        if hasattr(stmt, 'targets'):
            new_body.append(stmt)

        # Functions
        elif hasattr(stmt, 'name'):
            new_body.append(stmt)

    root.body = new_body
    exec(compile(root, filename="<ast>", mode="exec"), globals())



    config(["--project-path", "/bogus", "name"], parser=ArgumentParserStub())
    for opt in options:
        h = options[opt].get('help')
        if h:
            h = h.replace("\n", "")
            strings.append(h)

    strings = list(set(strings))
    print("Found %s ODM strings" % len(strings))
    if len(strings) > 0:
        with open(outfile, "w") as f:
            f.write("// Auto-generated with extract_odm_strings.py, do not edit!\n\n")

            for s in strings:
                f.write("_(\"%s\");\n" % s.replace("\"", "\\\""))

        print("Wrote %s" % outfile)
    else:
        print("No strings found")


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Extract ODM strings.')
    parser.add_argument('input', type=str,
                        help='URL to ODM\'s config.py')
    parser.add_argument('output', type=str,
                        help='Where to write resulting translation file')
    args = parser.parse_args()

    url = args.input
    outfile = args.output