Function main

tools/trace_orm_adapter.py:158–215

()

def main():
    global DEBUG_ADAPT_STEP

    parser = argparse.ArgumentParser()
    parser.add_argument(
        class="st">"-d", class="st">"--debug", type=int, help=class="st">"breakpoint at this adaptation step"
    )
    parser.add_argument(
        class="st">"-m",
        class="st">"--module",
        type=str,
        help=class="st">"import module name instead of running a script",
    )
    parser.add_argument(
        class="st">"args", metavar=class="st">"N", type=str, nargs=class="st">"*", help=class="st">"additional arguments"
    )

    argparse_args = []
    sys_argv = list(sys.argv)

    progname = sys_argv.pop(0)

    class="cm"># this is a little crazy, works at the moment for:
    class="cm"># module w args:
    class="cm">#    python tools/trace_orm_adapter.py -m  pytest test/orm/test_query.py -s
    class="cm"># script:
    class="cm">#   python tools/trace_orm_adapter.py test3.py
    has_module = False
    while sys_argv:
        arg = sys_argv.pop(0)
        if arg in (class="st">"-m", class="st">"--module", class="st">"-d", class="st">"--debug"):
            argparse_args.append(arg)
            argparse_args.append(sys_argv.pop(0))
            has_module = arg in (class="st">"-m", class="st">"--module")
        else:
            if not has_module:
                argparse_args.append(arg)
            else:
                sys_argv.insert(0, arg)
            break

    options = parser.parse_args(argparse_args)
    sys.argv = [class="st">"program.py"] + sys_argv

    if options.module == class="st">"pytest":
        sys.argv.extend([class="st">"--capture", class="st">"sys"])

    import runpy

    if options.debug:
        DEBUG_ADAPT_STEP = options.debug

    if options.module:
        runpy.run_module(options.module, run_name=class="st">"__main__")
    else:
        progname = options.args[0]

        runpy.run_path(progname)