Merge bootstack-ops:jjo-add-ipmiconsole_py into bootstack-ops:master

[JuanJo Ciarlante]

[jjo] add ipmiconsole.py, fix juju_bundle_export.py symlink

[jjo] add ipmiconsole.py, fix juju_bundle_export.py symlink

[Jill Rouleau (jillrouleau)]

review: Approve (lgtm)

diff --git a/bootstack-ops/ipmiconsole.py b/bootstack-ops/ipmiconsole.py
new file mode 100644
index 0000000..0c895f6
--- /dev/null
+++ b/bootstack-ops/ipmiconsole.py
@@ -0,0 +1,98 @@
+#!/usr/bin/env python3
+# ipmiconsole.py: run ipmitool sol activate from maas data via maas CLI
+#
+# Author: JuanJo Ciarlante <[e-mail]>
+# License: GPLv3+
+#
+# E.g. usage:
+#    ./ipmiconsole.py foo-c-001
+#    ./ipmiconsole.py --nodes_list
+#
+# To all nodes with passed prefix for "login:" via ipmi SOL:
+#    ./ipmiconsole.py --login_test foo-c-
+#
+import argparse
+import os
+import subprocess
+import sys
+import json
+import pexpect
+import logging
+
+
+def parse_args():
+    parser = argparse.ArgumentParser(
+        description='Connect to node via IPMI console')
+    parser.add_argument('hostname', nargs='?',
+                        help='maas hostname or prefix')
+    parser.add_argument('-u', dest='maas_user', nargs='?', default='maas',
+                        help='maas user (if not "maas")')
+    parser.add_argument('--nodes_list', action='store_true', default=False,
+                        help='list nodes prefixed by hostname and exit')
+    parser.add_argument('--login_test', action='store_true', default=False,
+                        help='test "login:" prompt nodes prefixed by hostname')
+    return parser.parse_args(sys.argv[1:])
+
+
+def get_power_params(args, system_id):
+    return json.loads(subprocess.check_output(
+                      ['maas', args.maas_user, 'node',
+                       'power-parameters', system_id]).decode())
+
+
+def get_ipmitool_sol(args, system_id):
+    cmd = ('ipmitool -H {power_address} -U {power_user} -P {power_pass} '
+           '-I lanplus sol activate').format(**get_power_params(args,
+                                                                system_id))
+    return cmd
+
+
+def get_node_ids(args, exact=False):
+    maas_cmd = ['maas', args.maas_user, 'nodes', 'read']
+    if exact and args.hostname:
+        maas_cmd.append('hostname={}'.format(args.hostname))
+    logger.debug('maas_cmd: {}'.format(maas_cmd))
+    maas_nodes = json.loads(subprocess.check_output(maas_cmd).decode())
+    prefix = args.hostname if args.hostname else ""
+    node_ids = [(node['hostname'], node['system_id']) for node in maas_nodes
+                if node['hostname'].startswith(prefix)]
+    return [node_id for node_id in sorted(node_ids, key=lambda x: x[0])]
+
+
+def main():
+    args = parse_args()
+    if args.nodes_list:
+        for (node, id) in get_node_ids(args):
+            print("{:32} {}".format(node, id))
+        sys.exit(0)
+    if args.login_test:
+        for (node, id) in get_node_ids(args):
+            cmd = get_ipmitool_sol(args, id)
+            expect_conn = pexpect.spawn(cmd)
+            try:
+                expect_conn.expect('SOL Session operational', timeout=5)
+                expect_conn.sendline("")
+                expect_conn.expect('login:', timeout=5)
+                print('PASS: {}'.format(node))
+            except (pexpect.exceptions.TIMEOUT, pexpect.exceptions.EOF) as e:
+                print('FAIL: {} - Error: {}'.format(node,
+                                                    str(e).split('\n')[0]))
+            finally:
+                expect_conn.close(force=True)
+        sys.exit(0)
+
+    if not args.hostname:
+       raise ValueError("missing hostname")
+
+    node, system_id = get_node_ids(args, exact=True)[0]
+    cmd = get_ipmitool_sol(args, system_id)
+    logger.debug("node: {}, system_id: {}, cmd: {}".format(
+        node, system_id, cmd))
+    os.system(cmd)
+
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger()
+
+
+if __name__ == '__main__':
+    main()
diff --git a/bootstack-ops/juju_bundle_export.py b/bootstack-ops/juju_bundle_export.py
index 25ff6d3..c90db6c 120000
--- a/bootstack-ops/juju_bundle_export.py
+++ b/bootstack-ops/juju_bundle_export.py
@@ -1 +1 @@
-../ops-bundle/common/juju_bundle_export.py
\ No newline at end of file
+../ops-bundle/scripts/juju_bundle_export.py
\ No newline at end of file