Add support to upload official image from OpenStack

1. Add support to upload official image from OpenStack;
2. Move dib and kb_server to upper level directory;
3. Fix the issue of dumping pecan server logs using Rest API;

Change-Id: I636fe4a1a2213d72c37fc09f37e1f70b2ca8c394

kb_server/kb_server/controllers/api_cfg.py

@@ -17,7 +17,7 @@ import json
 import os
 import sys
 import traceback
-kb_main_path = os.path.split(os.path.abspath(__file__))[0] + "/../../.."
+kb_main_path = os.path.split(os.path.abspath(__file__))[0] + "/../../../kloudbuster"
 sys.path.append(kb_main_path)
 
 from credentials import Credentials
@@ -36,6 +36,9 @@ class ConfigController(object):
     def default_config(self):
         kb_config = KBConfig()
         pdict = eval(str(kb_config.config_scale))
+        # Normally we don't allow the clients to change below configs
+        pdict['client'].pop('tp_tool')
+        pdict['client'].pop('http_tool')
         return json.dumps(pdict)
 
     @expose(generic=True)

kb_server/kb_server/controllers/api_kb.py

@@ -16,7 +16,7 @@ import json
 import os
 import sys
 import threading
-kb_main_path = os.path.split(os.path.abspath(__file__))[0] + "/../../.."
+kb_main_path = os.path.split(os.path.abspath(__file__))[0] + "/../../../kloudbuster"
 sys.path.append(kb_main_path)
 
 from kb_session import KBSessionManager

kloudbuster/kb_gen_chart.py

@@ -163,7 +163,7 @@ def gen_chart(file_list, chart_dest, browser, line_rate):
             data_list.append(results)
     if not line_rate:
         line_rate = guess_line_rate(data_list)
-    print line_rate
+        print 'Guessed line rate: %s Gbps.' % line_rate
     chart = KbReport(data_list, line_rate)
     print('Generating report to ' + chart_dest + '...')
     chart.plot(chart_dest)

kloudbuster/kb_vm_agent.py

@@ -1 +1 @@
-dib/elements/kloudbuster/static/kb_test/kb_vm_agent.py
+../kb_dib/elements/kloudbuster/static/kb_test/kb_vm_agent.py

kloudbuster/kloudbuster.py

@@ -15,16 +15,16 @@
 
 import json
 from multiprocessing.pool import ThreadPool
-import os
 import sys
 import threading
+import time
 import traceback
 
 from __init__ import __version__
 import base_compute
 import base_network
 import glanceclient.exc as glance_exception
-from glanceclient.v2 import client as glanceclient
+from glanceclient.v1 import client as glanceclient
 from kb_config import KBConfig
 from kb_res_logger import KBResLogger
 from kb_runner import KBRunner
@@ -236,7 +236,8 @@ class KloudBuster(object):
         self.kb_runner = None
         self.fp_logfile = None
 
-    def check_and_upload_images(self):
+    def check_and_upload_images(self, retry_count=150):
+        retry = 0
         keystone_list = [create_keystone_client(self.server_cred)[0],
                          create_keystone_client(self.client_cred)[0]]
         keystone_dict = dict(zip(['Server kloud', 'Client kloud'], keystone_list))
@@ -250,31 +251,37 @@ class KloudBuster(object):
             glance_client = glanceclient.Client(glance_endpoint, token=keystone.auth_token)
             try:
                 # Search for the image
-                glance_client.images.list(filters={'name': img_name_dict[kloud]}).next()
+                img = glance_client.images.list(filters={'name': img_name_dict[kloud]}).next()
-                return True
+                continue
             except StopIteration:
                 pass
 
             # Trying to upload images
-            kb_image_name = 'dib/' + kb_vm_agent.get_image_name() + '.qcow2'
+            kb_image_name = kb_vm_agent.get_image_name() + '.qcow2'
-            if not os.path.exists(kb_image_name):
+            image_url = 'http://storage.apps.openstack.org/images/%s' % kb_image_name
-                LOG.error("VM Image not in Glance and could not find " + kb_image_name +
+            LOG.info("Image is not found in %s, uploading from OpenStack App Store..." % kloud)
-                          " to upload, please refer to dib/README.rst for how to build"
+            try:
-                          " image for KloudBuster.")
+                img = glance_client.images.create(name=img_name_dict[kloud],
+                                                  disk_format="qcow2",
+                                                  container_format="bare",
+                                                  is_public=True,
+                                                  copy_from=image_url)
+                while img.status in ['queued', 'saving'] and retry < retry_count:
+                    img = glance_client.images.find(name=img.name)
+                    retry = retry + 1
+                    time.sleep(2)
+                if img.status != 'active':
+                    raise Exception
+            except glance_exception.HTTPForbidden:
+                LOG.error("Cannot upload image without admin access. Please make sure the "
+                          "image is uploaded and is either public or owned by you.")
                 return False
-            LOG.info("Image is not found in %s, uploading %s..." % (kloud, kb_image_name))
+            except Exception:
-            with open(kb_image_name) as fimage:
+                LOG.error("Failed while uploading the image, please make sure the cloud "
-                try:
+                          "under test has the access to URL: %s." % image_url)
-                    image = glance_client.images.create(name=img_name_dict[kloud],
+                return False
-                                                        disk_format="qcow2",
+
-                                                        container_format="bare",
+        return True
-                                                        visibility='public')
-                    glance_client.images.upload(image['id'], fimage)
-                except glance_exception.HTTPForbidden:
-                    LOG.error("Cannot upload image without admin access. Please make sure the "
-                              "image is uploaded and is either public or owned by you.")
-                    return False
-            return True
 
     def print_provision_info(self):
         """

kloudbuster/log.py

@@ -46,11 +46,13 @@ def setup(product_name, logfile=None):
     dbg_color = handlers.ColorHandler.LEVEL_COLORS[logging.DEBUG]
     handlers.ColorHandler.LEVEL_COLORS[logging.KBDEBUG] = dbg_color
 
+    oslogging.setup(CONF, product_name)
     if logfile:
         if os.path.exists(logfile):
             os.remove(logfile)
         CONF.log_file = logfile
-    oslogging.setup(CONF, product_name)
+        hdlr = logging.FileHandler(logfile)
+        oslogging.getLogger(product_name).logger.addHandler(hdlr)
 
     if CONF.kb_debug:
         oslogging.getLogger(
@@ -67,9 +69,7 @@ def getLogger(name="unknown", version="unknown"):
 def delete_logfile(product_name):
     if CONF.log_file and os.path.exists(CONF.log_file):
         os.remove(CONF.log_file)
-    # Reset the logging to default (stdout)
     CONF.log_file = None
-    oslogging.setup(CONF, product_name)
 
 
 class KloudBusterContextAdapter(oslogging.KeywordArgumentAdapter):