bin/easyovs

#!/usr/bin/python

__author__ = 'baohua'

import os
from oslo_config import cfg
import sys
import time

from easyovs import VERSION, config
from easyovs.bridge_ctrl import br_addflow, br_delflow, br_dump, br_list, \
    br_show
from easyovs.cli import CLI
from easyovs.common import CMDS_ONE, CMDS_TWO, CMDS_OTHER
from easyovs.iptables import show_iptables_rules
from easyovs.log import info, debug, error, output, LEVELS, lg
from easyovs.neutron import show_port_info
from easyovs.util import cleanup, fmt_flow_str


# Fix setuptools' evil madness, and open up (more?) security holes
if 'PYTHONPATH' in os.environ:
    sys.path = os.environ['PYTHONPATH'].split(':') + sys.path

class Platform(object):
    """
    Build, setup, and run the platform.
    """
    def __init__(self):
        self.options = None
        self.args = None  # May be used someday for more CLI scripts

        self.parse_args()
        self.setup()
        self.begin()

    def parse_args(self):
        config.init(sys.argv[1:])

    def setup(self):
        """
        Setup and validate environment.
        """

        # set logging verbosity
        if LEVELS[cfg.CONF.verbosity] > LEVELS['output']:
            print ('*** WARNING: selected verbosity level (%s) will hide CLI '
                   'output!\n'
                   'Please restart easyOVS with -v [debug, info, output].'
                   % cfg.CONF.verbosity)
        lg.set_log_level(cfg.CONF.verbosity)
        info("Set log level to %s\n" % cfg.CONF.verbosity)

    def begin(self):
        if cfg.CONF.clean:
            info("Cleaning environment\n")
            cleanup()
            exit()

        output('EasyOVS %s, type help for information\n' % VERSION)
        start = time.time()
        cmd = cfg.CONF.cmd
        info("cmd = %s\n" % cmd)
        if len(cmd.split()) == 1 and cmd in CMDS_ONE:
            if cmd == 'cli':
                CLI()
            elif cmd == 'list':
                self.list()
        elif len(cmd.split()) == 2:
            if cmd.split()[1] in CMDS_TWO:  # e.g., br0 show
                arg, func = cmd.split()[0], cmd.split()[1]
                getattr(self, func)(arg)
            elif cmd.split()[0] in CMDS_TWO:  # e.g., show br0
                func, arg = cmd.split()[0], cmd.split()[1]
                getattr(self, func)(arg)
            else:
                output('Wrong command format is given\n')
        elif len(cmd.split()) >= 3:
            if cmd.split()[1] in CMDS_OTHER:  # e.g., br0 delflow 9
                br, func, arg = cmd.split()[0], cmd.split()[1], ' '.join(
                    cmd.split()[2:])
                getattr(self, func)(br, arg.replace(',', ' '))
            elif cmd.split()[0] in CMDS_OTHER:  # e.g., delflow br0 9
                br, func, arg = cmd.split()[1], cmd.split()[0], ' '.join(
                    cmd.split()[2:])
                getattr(self, func)(br, arg.replace(',', ' '))
            elif cmd.split()[0] == 'ipt':  # e.g., ipt 10.0.0.1, 10.0.0.2
                func, arg = cmd.split()[0], ' '.join(cmd.split()[1:])
                getattr(self, func)(arg)
            else:
                output('Wrong command format is given\n')
        else:
            output('Wrong command format is given\n')
        elapsed = float(time.time() - start)
        info('\n### Completed in %0.3f seconds ###\n' % elapsed)

    @staticmethod
    def dump(br):
        br_dump(br)

    @staticmethod
    def ipt(ips):
        show_iptables_rules(ips)

    @staticmethod
    def query(keyword):
        show_port_info(keyword)

    @staticmethod
    def list():
        br_list()

    @staticmethod
    def show(br):
        br_show(br)

    @staticmethod
    def addflow(br, flow_str):
        br_addflow(br, fmt_flow_str(flow_str))

    @staticmethod
    def delflow(br, flow_id):
        br_delflow(br, flow_id)


if __name__ == "__main__":
    try:
        Platform()
    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")
        cleanup()