src/test-apps/happy/bin/weave-network-gateway.py

#!/usr/bin/env python3

#
#    Copyright (c) 2016-2017 Nest Labs, Inc.
#    All rights reserved.
#
#    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.
#

##
#    @file
#       A Happy command line utility that specifies Weave network gateway.
#
#       The command is executed by instantiating and running WeaveNetworkGateway class.
#

from __future__ import absolute_import
from __future__ import print_function
import getopt
import sys
import set_test_path

import WeaveNetworkGateway
from happy.Utils import *

if __name__ == "__main__":
    options = WeaveNetworkGateway.option()

    try:
        opts, args = getopt.getopt(sys.argv[1:], "hi:qadg:",
                                   ["help", "id=", "quiet", "add", "delete", "gateway="])

    except getopt.GetoptError as err:
        print(WeaveNetworkGateway.WeaveNetworkGateway.__doc__)
        print(hred(str(err)))
        sys.exit(hred("%s: Failed to parse arguments." % (__file__)))

    for o, a in opts:
        if o in ("-h", "--help"):
            print(WeaveNetworkGateway.WeaveNetworkGateway.__doc__)
            sys.exit(0)

        elif o in ("-q", "--quiet"):
            options["quiet"] = True

        elif o in ("-i", "--id"):
            options["network_id"] = a

        elif o in ("-v", "--gateway"):
            options["gateway"] = a

        elif o in ("-a", "--add"):
            options["add"] = True

        elif o in ("-d", "--delete"):
            options["delete"] = True

        else:
            assert False, "unhandled option"

    if len(args) == 1:
        options["network_id"] = args[0]

    if len(args) == 2:
        options["network_id"] = args[0]
        options["gateway"] = args[1]

    cmd = WeaveNetworkGateway.WeaveNetworkGateway(options)
    cmd.start()