Merge "Capture metalsmith python logging"

tripleo_ansible/ansible_plugins/modules/metalsmith_instances.py

@@ -16,6 +16,8 @@
 __metaclass__ = type
 
 from concurrent import futures
+import io
+import logging
 
 from ansible.module_utils.basic import AnsibleModule
 from ansible.module_utils.openstack import openstack_cloud_from_module
@@ -182,9 +184,34 @@ options:
       - Maximum number of instances to provision at once. Set to 0 to have no
         concurrency limit
     type: int
+  log_level:
+    description:
+      - Set the logging level for the log which is available in the
+        returned 'logging' result.
+    default: info
+    choices:
+    - debug
+    - info
+    - warning
+    - error
 '''
 
 
+METALSMITH_LOG_MAP = {
+    'debug': logging.DEBUG,
+    'info': logging.INFO,
+    'warning': logging.WARNING,
+    'error': logging.ERROR
+}
+
+BASE_LOG_MAP = {
+    'debug': logging.INFO,
+    'info': logging.WARNING,
+    'warning': logging.WARNING,
+    'error': logging.ERROR
+}
+
+
 def _get_source(instance):
     image = instance.get('image')
     return sources.detect(image=image.get('href'),
@@ -310,6 +337,22 @@ def unprovision(provisioner, instances):
     return True
 
 
+def _configure_logging(log_level):
+    log_fmt = ('%(asctime)s %(levelname)s %(name)s: %(message)s')
+    urllib_level = logging.CRITICAL
+
+    base_level = BASE_LOG_MAP[log_level]
+    metalsmith_level = METALSMITH_LOG_MAP[log_level]
+
+    logging.basicConfig(level=base_level, format=log_fmt)
+    logging.getLogger('urllib3.connectionpool').setLevel(urllib_level)
+    logger = logging.getLogger('metalsmith')
+    logger.setLevel(metalsmith_level)
+    log_stream = io.StringIO()
+    logger.addHandler(logging.StreamHandler(log_stream))
+    return log_stream
+
+
 def main():
     argument_spec = openstack_full_argument_spec(
         **yaml.safe_load(DOCUMENTATION)['options']
@@ -321,6 +364,9 @@ def main():
         **module_kwargs
     )
 
+    log_stream = _configure_logging(module.params['log_level'])
+
+    try:
         sdk, cloud = openstack_cloud_from_module(module)
         provisioner = metalsmith.Provisioner(cloud_region=cloud.config)
         instances = module.params['instances']
@@ -343,6 +389,7 @@ def main():
                 changed=changed,
                 msg="{} instances provisioned".format(len(nodes)),
                 instances=instances,
+                logging=log_stream.getvalue()
             )
 
         if state == 'reserved':
@@ -351,14 +398,21 @@ def main():
                 changed=changed,
                 msg="{} instances reserved".format(len(nodes)),
                 ids=[node.id for node in nodes],
-            instances=instances
+                instances=instances,
+                logging=log_stream.getvalue()
             )
 
         if state == 'absent':
             changed = unprovision(provisioner, instances)
             module.exit_json(
                 changed=changed,
-            msg="{} nodes unprovisioned".format(len(instances))
+                msg="{} nodes unprovisioned".format(len(instances)),
+                logging=log_stream.getvalue()
+            )
+    except Exception as e:
+        module.fail_json(
+            msg=str(e),
+            logging=log_stream.getvalue()
         )
 
 

tripleo_ansible/playbooks/cli-overcloud-node-provision.yaml

@@ -100,8 +100,13 @@
         instances: "{{ baremetal_existing.not_found }}"
         state: reserved
         clean_up: true
+        log_level: info
       register: baremetal_reserved
 
+    - name: Metalsmith log for reserve instances
+      debug:
+        var: baremetal_reserved.logging
+
     # NOTE(cloudnull): This limits the concurrency so that we're not adding
     #                  more threads than needed.
     - name: Set concurrency fact
@@ -119,8 +124,13 @@
         clean_up: false
         timeout: "{{ node_timeout }}"
         concurrency: "{{ runtime_concurrency }}"
+        log_level: info
       register: baremetal_provisioned
 
+    - name: Metalsmith log for provision instances
+      debug:
+        var: baremetal_provisioned.logging
+
     - name: Populate environment
       tripleo_baremetal_populate_environment:
         environment: "{{ baremetal_instances.environment }}"