source: doc/theses/andrew_beach_MMath/code/cond_catch.py@ d83b266

#!/usr/bin/env python3

# Conditional Match (or Re-Raise)

from time import thread_time_ns


class EmptyException(Exception):
    pass


should_catch = False


def throw_exception():
    raise EmptyException()


def cond_catch():
    try:
        throw_exception()
    except EmptyException as exc:
        if not should_catch:
            raise


def main(argv):
    times = 1
    if 1 < len(argv):
        times = int(argv[1])
    if 2 < len(argv):
        should_catch = 0 < int(argv[2])

    start_time = thread_time_ns()
    for count in range(times):
        try:
            cond_catch();
        except EmptyException:
            pass

    end_time = thread_time_ns()
    print('Run-Time (ns):', end_time - start_time)


if '__main__' == __name__:
    import sys
    main(sys.argv)