summaryrefslogtreecommitdiffstats
path: root/vicn/bin/vicn.py
blob: 9a43cf6d23b4d8e553fe5650b03718efcaecc9d1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# Copyright (c) 2017 Cisco and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

import os, sys, time, asyncio, argparse, shutil, threading, asyncio, logging
import traceback

log = logging.getLogger(__name__)

PATH=os.path.join(os.path.dirname(__file__), os.path.pardir, os.path.pardir)
sys.path.insert(0, os.path.abspath(PATH))

from netmodel.model.query       import Query
from netmodel.util.daemon       import Daemon
from netmodel.util.log          import textcolor, initialize_logging
from vicn.clients.command_line  import VICNCmd
from vicn.core.api              import API
from vicn.core.resource_mgr     import ResourceManager
from vicn.resource.node         import Node

class ArgumentParser(argparse.ArgumentParser):
    def error(self, message):
        self._print_message(textcolor('red', '{prog}: error: {message}\n'.format(prog=self.prog, message=message)),
                            sys.stderr)

        self.print_usage(sys.stdout)
        sys.exit(-1)

class VICNDaemon(Daemon):
    def initialize(self):
        # FIXME UGLY
        n_times = 1
        background = False
        setup = False
        scenario = None
        node_list, net, ndn, mob, cluster = None, None, None, None, None

        parser = ArgumentParser(description=textcolor('green', "Batch usage of VICN."))
        parser.add_argument('-s', metavar='configuration_file_path',
                            help="JSON file containing the topology")
        parser.add_argument('-n', metavar='n_times', type=int,  help='Execute the test multiple times')
        parser.add_argument('-x', action='store_false', help='No automatic execution')

        arguments = parser.parse_args()
        args = vars(arguments)


        for option in args.keys():
            if args[option] is not None:
                if option == "s":
                    print(" * Loading the configuration file at {0}".format(args[option]))
                    scenario = args[option]
                elif option == "t" and args[option] is True:
                    background = True
                elif option == "x" and args[option] is True:
                    setup = True
                elif option == "n":
                    n_times = args[option]

        self._api = API()
        self._api.configure(scenario, setup)

        if node_list is not None:
            ResourceManager().set(node_list)

    def main(self):
        """
        Main asyncio loop.

        Ctrl+C properly terminates the loop by terminating all running
        instances.
        """
        loop = asyncio.get_event_loop()
        try:
            loop.run_forever()
        except KeyboardInterrupt:
            pass
        except Exception as e:
            import traceback
            traceback.print_exc()
        finally:
            loop.stop()
            self._api.terminate()

def main():
    initialize_logging()
    VICNDaemon('vicn', no_daemon = True).start()

if __name__ == "__main__":
    main()