vsphere-automation-sdk-python/samples/vsphere/logforwarding/log_forwarding.py

#!/usr/bin/env python
"""
* *******************************************************
* Copyright (c) VMware, Inc. 2017, 2018, 2018. All Rights Reserved.
* SPDX-License-Identifier: MIT
* *******************************************************
*
* DISCLAIMER. THIS PROGRAM IS PROVIDED TO YOU "AS IS" WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, WHETHER ORAL OR WRITTEN,
* EXPRESS OR IMPLIED. THE AUTHOR SPECIFICALLY DISCLAIMS ANY IMPLIED
* WARRANTIES OR CONDITIONS OF MERCHANTABILITY, SATISFACTORY QUALITY,
* NON-INFRINGEMENT AND FITNESS FOR A PARTICULAR PURPOSE.
"""

__author__ = 'VMware, Inc.'
__vcenter_version__ = '6.7+'

from samples.vsphere.common import sample_cli
from samples.vsphere.common import sample_util
from samples.vsphere.common import vapiconnect

from com.vmware.appliance.logging_client import Forwarding


class LogForwarding(object):
    """
    Demonstrates log forwarding API operations

    Prerequisites:
        - vCenter
        - Log host listening to syslog packets over any of the supported
          protocols UDP/TCP/TLS
    """

    def __init__(self):
        self.loghost = None
        self.protocol = None
        self.port = None
        self.stub_config = None
        self.log_forwarding_client = None

    def setup(self):
        parser = sample_cli.build_arg_parser()

        parser.add_argument(
            '--loghost', required=True, action='store', help='The log host')
        parser.add_argument(
            '--port',
            required=True,
            action='store',
            help='The log host port number')
        parser.add_argument(
            '--protocol',
            required=True,
            action='store',
            help='The log host protocol (TCP/UDP/TLS)')

        args = sample_util.process_cli_args(parser.parse_args())
        self.loghost = args.loghost
        self.protocol = args.protocol
        self.port = int(args.port)

        # Connect to vAPI services
        self.stub_config = vapiconnect.connect(
            host=args.server,
            user=args.username,
            pwd=args.password,
            skip_verification=args.skipverification)

        self.log_forwarding_client = Forwarding(self.stub_config)

    def run(self):
        # Set log forwarding configuration
        self.set_log_forwarding()

        # Get log forwarding configuration
        self.get_log_forwarding()

        # Test log forwarding configuration
        self.test_log_forwarding()

        # Update log forwarding configuration
        self.update_log_forwarding()

    def set_log_forwarding(self):
        log_forwarding_config = [
            Forwarding.Config(
                hostname=self.loghost, port=self.port, protocol=self.protocol)
        ]
        self.log_forwarding_client.set(log_forwarding_config)

    def get_log_forwarding(self):
        configs = self.log_forwarding_client.get()

        for cfg in configs:
            print('Loghost: {}, Port: {}, Protocol: {}'.format(
                cfg.hostname, cfg.port, cfg.protocol))

    def test_log_forwarding(self):
        test_response = self.log_forwarding_client.test(True)

        print("\nLog forwarding test response:")
        for resp in test_response:
            print('Loghost: {}, State: {}, Message: {}'.format(
                resp.hostname, resp.state,
                resp.message.default_message if resp.message else None))

    def update_log_forwarding(self):
        # Read log forwarding configuration
        log_forwarding_config = self.log_forwarding_client.get()

        # Delete the newly added configuration
        log_forwarding_config = list(
            filter(lambda cfg: cfg.hostname != self.loghost,
                   log_forwarding_config))

        # Apply the modified log forwarding configuration
        self.log_forwarding_client.set(log_forwarding_config)


def main():
    log_forwarding = LogForwarding()
    log_forwarding.setup()
    log_forwarding.run()


if __name__ == '__main__':
    main()