1
0
mirror of https://github.com/vmware/vsphere-automation-sdk-python.git synced 2024-11-29 12:00:00 -05:00
vsphere-automation-sdk-python/samples/vsphere/logforwarding/log_forwarding.py
Tianhao He f1032b4e43 vSphere 6.7 Release
Signed-off-by: Tianhao He <het@vmware.com>
2018-04-17 11:01:27 -07:00

133 lines
4.4 KiB
Python

#!/usr/bin/env python
"""
* *******************************************************
* Copyright (c) VMware, Inc. 2017, 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 tabulate import tabulate
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()
print("\nLog forwarding configurations:")
table = [[cfg.hostname, cfg.port, cfg.protocol] for cfg in configs]
headers = ["Loghost", "Port", "Protocol"]
print(tabulate(table, headers))
def test_log_forwarding(self):
test_response = self.log_forwarding_client.test(True)
print("\nLog forwarding test response:")
table = [[resp.hostname,
resp.state,
resp.message.default_message if resp.message else None]
for resp in test_response]
headers = ["Loghost", "State", "Message"]
print(tabulate(table, headers))
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()