summaryrefslogtreecommitdiffstats
path: root/lib
diff options
context:
space:
mode:
authorTim Rozet <trozet@redhat.com>2016-08-31 01:40:12 +0000
committerGerrit Code Review <gerrit@172.30.200.206>2016-08-31 01:40:13 +0000
commitf01595f801e1d78f84b43c111f2955f67573e263 (patch)
tree37d553b3b2ced09a87923fd9e1328d7608aec048 /lib
parent04a7bf2596ea69def02e8aaedd32f091b35ba7ce (diff)
parent4b84887ed2322db6b04924bdee7b44d3dcf7c32d (diff)
Merge "Adds ability to power off nodes in clean"
Diffstat (limited to 'lib')
-rw-r--r--lib/python/apex/__init__.py1
-rw-r--r--lib/python/apex/clean.py39
-rw-r--r--lib/python/apex/common/utils.py8
-rwxr-xr-xlib/python/apex_python_utils.py11
4 files changed, 59 insertions, 0 deletions
diff --git a/lib/python/apex/__init__.py b/lib/python/apex/__init__.py
index e1b8b547..9993dc97 100644
--- a/lib/python/apex/__init__.py
+++ b/lib/python/apex/__init__.py
@@ -11,3 +11,4 @@
from .network_settings import NetworkSettings
from .deploy_settings import DeploySettings
from .network_environment import NetworkEnvironment
+from .clean import clean_nodes
diff --git a/lib/python/apex/clean.py b/lib/python/apex/clean.py
new file mode 100644
index 00000000..184b5ec9
--- /dev/null
+++ b/lib/python/apex/clean.py
@@ -0,0 +1,39 @@
+##############################################################################
+# Copyright (c) 2016 Tim Rozet (trozet@redhat.com) and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+# Clean will eventually be migrated to this file
+
+import logging
+import pyipmi
+import pyipmi.interfaces
+import sys
+
+from .common import utils
+
+
+def clean_nodes(inventory):
+ inv_dict = utils.parse_yaml(inventory)
+ if inv_dict is None or 'nodes' not in inv_dict:
+ logging.error("Inventory file is empty or missing nodes definition")
+ sys.exit(1)
+ for node, node_info in inv_dict['nodes'].items():
+ logging.info("Cleaning node: {}".format(node))
+ try:
+ interface = pyipmi.interfaces.create_interface(
+ 'ipmitool', interface_type='lanplus')
+ connection = pyipmi.create_connection(interface)
+ connection.session.set_session_type_rmcp(node_info['ipmi_ip'])
+ connection.target = pyipmi.Target(0x20)
+ connection.session.set_auth_type_user(node_info['ipmi_user'],
+ node_info['ipmi_pass'])
+ connection.session.establish()
+ connection.chassis_control_power_down()
+ except Exception as e:
+ logging.error("Failure while shutting down node {}".format(e))
+ sys.exit(1)
diff --git a/lib/python/apex/common/utils.py b/lib/python/apex/common/utils.py
index b7678a20..fe34096d 100644
--- a/lib/python/apex/common/utils.py
+++ b/lib/python/apex/common/utils.py
@@ -7,9 +7,17 @@
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
+import yaml
+
def str2bool(var):
if isinstance(var, bool):
return var
else:
return var.lower() in ("true", "yes")
+
+
+def parse_yaml(yaml_file):
+ with open(yaml_file) as f:
+ parsed_dict = yaml.load(f)
+ return parsed_dict
diff --git a/lib/python/apex_python_utils.py b/lib/python/apex_python_utils.py
index b6aaafaa..1f5e407c 100755
--- a/lib/python/apex_python_utils.py
+++ b/lib/python/apex_python_utils.py
@@ -7,6 +7,7 @@
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
+import apex
import argparse
import sys
import logging
@@ -61,6 +62,10 @@ def parse_deploy_settings(args):
settings.dump_bash()
+def run_clean(args):
+ apex.clean_nodes(args.file)
+
+
def find_ip(args):
"""
Get and print the IP from a specific interface
@@ -192,6 +197,12 @@ def get_parser():
help='path to deploy settings file')
deploy_settings.set_defaults(func=parse_deploy_settings)
+ clean = subparsers.add_parser('clean',
+ help='Parse deploy settings file')
+ clean.add_argument('-f', '--file',
+ help='path to inventory file')
+ clean.set_defaults(func=run_clean)
+
return parser