ce

#!/usr/bin/env python3

"""
About: ComNetsEmu CLI utility
"""

import argparse
import sys

from mininet.log import LEVELS, debug, error, info, lg
from comnetsemu.clean import cleanup


def run():
    parser = argparse.ArgumentParser()
    parser.add_argument("-v", "--verbosity", choices=LEVELS.keys(), default="info")
    parser.add_argument("-c", "--clean", action="store_true", help="Run cleanups")
    args = parser.parse_args()

    lg.setLogLevel(args.verbosity)

    if args.clean:
        cleanup()


#  TODO:  <05-08-19, Zuo> Add helpers to setup Userspace network framework like DPDK #

if __name__ == "__main__":
    try:
        run()
    except KeyboardInterrupt:
        info("\n\nKeyboard Interrupt. Shutting down and cleaning up...\n\n")
        cleanup()
    except Exception:
        # Print exception
        type_, val_, trace_ = sys.exc_info()
        errorMsg = (
            "-" * 80
            + "\n"
            + "Caught exception. Cleaning up...\n\n"
            + "%s: %s\n" % (type_.__name__, val_)
            + "-" * 80
            + "\n"
        )
        error(errorMsg)
        # Print stack trace to debug log
        import traceback

        stackTrace = traceback.format_exc()
        debug(stackTrace + "\n")