breuleux / voir

Modular general instrumentation for Python
0 stars 2 forks source link

Voir

Documentation

Voir lets you wrap your scripts to display certain values or metrics and/or analyze their behavior. To use Voir:

Here are a few things you can do:

Functioning

An instrument is a function defined in voirfile.py that begins with instrument_, or a value in the __instruments__ dictionary of the same file. It takes one argument, the overseer, and its purpose is to do things at various points during execution. In order to do so, it is implemented as a generator function: all it has is to yield one of the overseer's phases, and the overseer will return to the instrument after that phase is finished. The overseer defines the order, so you only need to yield the phases you want to wait for.

def instrument_show_phases(ov):
    yield ov.phases.init
    # Voir has initialized itself. You can add command-line arguments here.

    yield ov.phases.parse_args
    # The command-line arguments have been parsed.

    yield ov.phases.load_script
    # The script has been loaded: its imports have been done, its functions defined,
    # but the top level statements have not been executed. You can perform some
    # manipulations prior to the script running.

    yield ov.phases.run_script
    # The script has finished.

    yield ov.phases.finalize

Voir also logs events in the 3rd file descriptor if it is open, or to the $DATA_FD descriptor. Consequently, if you run voir script.py 3>&1 you should be able to see the list of phases.

Example

This instrument adds a --time command-line argument to Voir. When given, it will calculate the time the script took to load and import its dependencies, and then the time it took to run, and it will print out these times.

def instrument_time(ov):
    yield ov.phases.init

    ov.argparser.add_argument("--time", action="store_true")

    yield ov.phases.parse_args

    if ov.options.time:
        t0 = time.time()

        yield ov.phases.load_script

        t1 = time.time()
        print(f"Load time: {(t1 - t0) * 1000}ms")

        yield ov.phases.run_script

        t2 = time.time()
        print(f"Run time: {(t2 - t1) * 1000}ms")

The --time argument goes BEFORE the script, so you would invoke it like this:

voir --time script.py SCRIPT_ARGUMENTS ...