kea-shell.in 3.99 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
#!@PYTHON@

# Copyright (C) 2017 Internet Systems Consortium, Inc. ("ISC")
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

"""
Text client for Control Agent process
"""

# First, let's import the right kea_connector.
# We have two versions: one for python 2.x and another for python 3.x.
# Sadly, there's no unified way to handle http connections. The recommended
# way is to use Requests (http://docs.python-requests.org/en/master/), but
# that's a stand alone package that requires separate installation. One of
# the design requirements was to not require any additional packages, so
# the code uses standard libraries available in python. Hence two versions.
Francis Dupont's avatar
Francis Dupont committed
20
import os
21 22 23 24
import sys
import signal
import argparse

Francis Dupont's avatar
Francis Dupont committed
25 26
sys.path.append('@PKGPYTHONDIR@')

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
from kea_conn import CARequest # CAResponse

if sys.version_info[0] == 2:
    # This is Python 2.x
    import kea_connector2 as kea_connector
elif sys.version_info[0] == 3:
    # This is Python 3.x
    import kea_connector3 as kea_connector
else:
    # This is... have no idea what it is.
    raise SystemExit("Unknown python version:" + str(sys.version_info[0]))

def timeout_handler(signum, frame):
    """Connection timeoout handler"""
    del signum, frame
    print("Connection timeout")
    sys.exit(1)

VERSION = "@PACKAGE_VERSION@"

def shell_body():
    """
    Second step: Need to parse command line parameters. We will use
    argparse for that purpose. It does great job with having default
    values, taking care of the help and sanity checking input
    parameters.
    """
    parser = argparse.ArgumentParser(description='kea-shell is a simple text '
                                     'client that uses REST interface to '
                                     'connect to Kea Control Agent.')
    parser.add_argument('--host', type=str, default='127.0.0.1',
                        help='hostname of the CA to connect to '
                        '(defaul:; 127.0.0.1)')
    parser.add_argument('--port', type=int, default=8000,
                        help='TCP port of the CA to connect to '
                        '(default: 8000)')
    parser.add_argument('--timeout', type=int, default='10',
                        help='Timeout (in seconds) when attempting to '
                        'connect to CA (default: 10)')
    parser.add_argument('command', type=str, nargs="?",
                        default='list-commands',
                        help='command to be executed. If not specified, '
                        '"list-commands" is used')
    parser.add_argument('-v', action="store_true", help="Prints version")
    cmd_args = parser.parse_args()

    if cmd_args.v:
        print(VERSION)
        exit(0)

Tomek Mrugalski's avatar
Tomek Mrugalski committed
77
    # Ok, now it's time to put the parameters parsed into the structure to be
78 79 80 81 82 83 84 85 86 87 88 89 90
    # used by the connection.
    params = CARequest()
    params.command = cmd_args.command
    params.http_host = cmd_args.host
    params.http_port = cmd_args.port
    params.timeout = cmd_args.timeout
    params.version = VERSION

    # Load command processor
    # @todo - command specific processing will be added as part of
    # future work (either #5138 or #5139, whichever is implemented
    # first)

91
    # Read arguments from stdin (they're optional for some commands)
92
    for line in sys.stdin:
93 94 95 96 97
        params.args += line

    # Now we have the arguments so we can build the request
    params.generate_body()
    params.generate_headers()
98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116

    # Set the timeout timer. If the connection takes too long,
    # it will send a signal to us.
    signal.signal(signal.SIGALRM, timeout_handler)
    signal.alarm(params.timeout)

    # Ok, everything is ready. Let's send the command and get a response.
    try:
        resp = kea_connector.send_to_control_agent(params)
    except Exception as exc:
        print("Failed to run: " + str(exc))
        sys.exit(1)

    resp.print_response()

    sys.exit(0)

if __name__ == "__main__":
    shell_body()