upgrade-plugin renamed to upgrade-plugins

octane/commands/upgrade_plugins.py

@@ -0,0 +1,162 @@
+# 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.
+
+from __future__ import print_function
+
+import logging
+import re
+import uuid
+import yaml
+
+from octane.commands.upgrade_db import get_controllers
+from octane.util import ssh
+
+from cliff import command as cmd
+from fuelclient.objects import environment
+import zabbix_client
+
+LOG = logging.getLogger(__name__)
+
+
+def get_template_hosts_by_name(client, plugin_name):
+    return client.template.get(filter={'name': plugin_name},
+                               selectHosts=['name'])[0]['hosts']
+
+
+def get_host_snmp_ip(client, host_id):
+    # second type is SNMP type
+    return client.hostinterface.get(hosids=host_id,
+                                    output=['ip'],
+                                    filter={'type': 2})[0]['ip']
+
+
+def get_zabbix_url(env):
+    data = env.get_network_data()
+    ip = data['public_vip']
+    return 'http://{0}/zabbix'.format(ip)
+
+
+def get_astute_yaml(env):
+    node = next(get_controllers(env))
+    data, _ = ssh.call(['cat', '/etc/astute.yaml'], stdout=ssh.PIPE, node=node)
+    return yaml.load(data)
+
+
+def get_zabbix_credentials(env):
+    astute = get_astute_yaml(env)
+    return astute['zabbix']['username'], astute['zabbix']['password']
+
+
+def zabbix_monitoring_settings(env):
+    astute = get_astute_yaml(env)
+    return {'username': {'value': astute['zabbix']['username']},
+            'password': {'value': astute['zabbix']['password']},
+            'db_password': {'value': astute['zabbix']['db_password']},
+            'metadata': {'enabled': astute['zabbix']['enabled']}}
+
+
+def emc_vnx_settings(env):
+    astute = get_astute_yaml(env)
+    return {'emc_sp_a_ip': {'value': astute['storage']['emc_sp_a_ip']},
+            'emc_sp_b_ip': {'value': astute['storage']['emc_sp_b_ip']},
+            'emc_password': {'value': astute['storage']['emc_password']},
+            'emc_username': {'value': astute['storage']['emc_username']},
+            'emc_pool_name': {'value': astute['storage']['emc_pool_name']},
+            'metadata': {'enabled': astute['storage']['volumes_emc']}}
+
+
+def zabbix_snmptrapd_settings(env):
+    # go to controller and parse /etc/snmp/snmptrapd.conf
+    # ACCESS CONTROL sections
+    node = next(get_controllers(env))
+    data, _ = ssh.call(['cat', '/etc/snmp/snmptrapd.conf'], stdout=ssh.PIPE, node=node)
+    template = re.compile(r"authCommunity\s[a-z-,]+\s([a-z-]+)")
+    match = template.search(data)
+    return {'community': {'value': match.group(1)},
+            'metadata': {'enabled': True}}
+
+
+def zabbix_monitoring_emc_settings(env):
+    url = get_zabbix_url(env)
+    user, password = get_zabbix_credentials(env)
+    client = zabbix_client.ZabbixServerProxy(url)
+    client.user.login(user=user, password=password)
+    hosts = get_template_hosts_by_name(client, 'Template EMC VNX')
+    for host in hosts:
+        host['ip'] = get_host_snmp_ip(client, host['hostid'])
+    settings = ','.join(':'.join((host['name'], host['ip'])) for host in hosts)
+    return {'hosts': {'value': settings},
+            'metadata': {'enabled': True}}
+
+
+def zabbix_monitoring_extreme_networks_settings(env):
+    url = get_zabbix_url(env)
+    user, password = get_zabbix_credentials(env)
+    client = zabbix_client.ZabbixServerProxy(url)
+    client.user.login(user=user, password=password)
+    hosts = get_template_hosts_by_name(client, 'Template Extreme Networks')
+    for host in hosts:
+        host['ip'] = get_host_snmp_ip(client, host['hostid'])
+    settings = ','.join(':'.join((host['name'], host['ip'])) for host in hosts)
+    return {'hosts': {'value': settings},
+            'metadata': {'enabled': True}}
+
+
+def transfer_plugins_settings(orig_env_id, seed_env_id, plugins):
+    orig_env = environment.Environment(orig_env_id)
+    seed_env = environment.Environment(seed_env_id)
+    attrs = {}
+    for plugin_name in plugins:
+        plugin = plugin_name.replace('-', '_')
+        attrs[plugin] = PLUGINS[plugin_name](orig_env)
+        print(plugin)
+    seed_env.update_attributes({'editable': attrs})
+
+
+PLUGINS = {
+    'zabbix-monitoring': zabbix_monitoring_settings,
+    'emc-vnx': emc_vnx_settings,
+    'zabbix-snmptrapd': zabbix_snmptrapd_settings,
+    'zabbix-monitoring-emc': zabbix_monitoring_emc_settings,
+    'zabbix-monitoring-extreme-networks': \
+         zabbix_monitoring_extreme_networks_settings,
+}
+
+
+class UpgradePluginsCommand(cmd.Command):
+    """Transfer settings for specified plugin from ORIG_ENV to SEED_ENV"""
+
+    def get_parser(self, prog_name):
+        parser = super(UpgradePluginCommand, self).get_parser(prog_name)
+        parser.add_argument(
+            'orig_env',
+            type=int,
+            metavar='ORIG_ID',
+            help="ID of original environment")
+        parser.add_argument(
+            'seed_env',
+            type=int,
+            metavar='SEED_ID',
+            help="ID of seed environment")
+        parser.add_argument(
+            'plugins',
+            metavar='PLUGIN',
+            choices=PLUGINS,
+            nargs="+",
+            help="Choose from {0}".format(PLUGINS.keys()))
+
+        return parser
+
+    def take_action(self, parsed_args):
+        transfer_plugins_settings(parsed_args.orig_env,
+                                  parsed_args.seed_env,
+                                  parsed_args.plugins)

requirements.txt

@@ -17,4 +17,4 @@ python-fuelclient>=6.1
 cliff>=1.7.0,<=1.9.0  # should already be pulled by python-fuelclient
 paramiko==1.13.0
 
-zabbix-client
+zabbix-client==0.1.0

setup.cfg

@@ -35,4 +35,4 @@ octane =
     upgrade-db = octane.commands.upgrade_db:UpgradeDBCommand
     install-node = octane.commands.install_node:InstallNodeCommand
     upgrade-control = octane.commands.upgrade_controlplane:UpgradeControlPlaneCommand
-    upgrade-plugin = octane.commands.upgrade_plugin:UpgradePluginCommand
+    upgrade-plugins = octane.commands.upgrade_plugins:UpgradePluginsCommand