diff options
Diffstat (limited to 'cvp/opnfv_testapi')
| -rw-r--r-- | cvp/opnfv_testapi/resources/sut_handlers.py | 112 | ||||
| -rw-r--r-- | cvp/opnfv_testapi/resources/sut_models.py | 31 | ||||
| -rw-r--r-- | cvp/opnfv_testapi/router/url_mappings.py | 2 |
3 files changed, 145 insertions, 0 deletions
diff --git a/cvp/opnfv_testapi/resources/sut_handlers.py b/cvp/opnfv_testapi/resources/sut_handlers.py new file mode 100644 index 00000000..16c50b83 --- /dev/null +++ b/cvp/opnfv_testapi/resources/sut_handlers.py @@ -0,0 +1,112 @@ +############################################################################## +# Copyright (c) 2017 +# 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 +############################################################################## +import logging +import json +import os + +from opnfv_testapi.resources import handlers +from opnfv_testapi.resources import sut_models +from opnfv_testapi.tornado_swagger import swagger + +LOG = logging.getLogger(__name__) +LOG.setLevel(logging.DEBUG) + +RESULT_PATH = '/home/testapi/logs/{}/results' + + +class GenericSutHandler(handlers.GenericApiHandler): + def __init__(self, application, request, **kwargs): + super(GenericSutHandler, self).__init__(application, + request, + **kwargs) + self.table = "suts" + self.table_cls = sut_models.Sut + + +class HardwareHandler(GenericSutHandler): + @swagger.operation(nickname="getHardwareById") + def get(self, id): + endpoint_info = self._read_endpoint_info(id) + LOG.debug('Endpoint info: %s', endpoint_info) + + all_info = self._read_sut_info(id) + LOG.debug('All SUT info: %s', all_info) + + hardware_info = {k: self._get_single_host_info(v) + for k, v in all_info.items()} + LOG.debug('SUT info: %s', hardware_info) + + data = { + 'endpoint_info': endpoint_info, + 'hardware_info': hardware_info + } + + self.write(data) + + def _read_endpoint_info(self, id): + path = os.path.join(RESULT_PATH.format(id), 'endpoint_info.json') + try: + with open(path) as f: + endpoint_info = json.load(f) + except Exception: + endpoint_info = [] + + return endpoint_info + + def _read_sut_info(self, id): + path = os.path.join(RESULT_PATH.format(id), 'all_hosts_info.json') + try: + with open(path) as f: + all_info = json.load(f) + except Exception: + all_info = {} + return all_info + + def _get_single_host_info(self, single_info): + info = [] + facts = single_info.get('ansible_facts', {}) + + info.append(['hostname', facts.get('ansible_hostname')]) + + info.append(['product_name', facts.get('ansible_product_name')]) + info.append(['product_version', facts.get('ansible_product_version')]) + + processors = facts.get('ansible_processor', []) + try: + processor_type = '{} {}'.format(processors[0], processors[1]) + except IndexError: + LOG.exception('No Processor in SUT data') + processor_type = None + info.append(['processor_type', processor_type]) + info.append(['architecture', facts.get('ansible_architecture')]) + info.append(['processor_cores', facts.get('ansible_processor_cores')]) + info.append(['processor_vcpus', facts.get('ansible_processor_vcpus')]) + + memory = facts.get('ansible_memtotal_mb') + memory = round(memory * 1.0 / 1024, 2) if memory else None + info.append(['memory', '{} GB'.format(memory)]) + + devices = facts.get('ansible_devices', {}) + info.extend([self._get_device_info(k, v) for k, v in devices.items()]) + + lsb_description = facts.get('ansible_lsb', {}).get('description') + info.append(['OS', lsb_description]) + + interfaces = facts.get('ansible_interfaces') + info.append(['interfaces', interfaces]) + info.extend([self._get_interface_info(facts, i) for i in interfaces]) + info = [i for i in info if i] + + return info + + def _get_interface_info(self, facts, name): + mac = facts.get('ansible_{}'.format(name), {}).get('macaddress') + return [name, mac] if mac else [] + + def _get_device_info(self, name, info): + return ['disk_{}'.format(name), info.get('size')] diff --git a/cvp/opnfv_testapi/resources/sut_models.py b/cvp/opnfv_testapi/resources/sut_models.py new file mode 100644 index 00000000..b4a869b6 --- /dev/null +++ b/cvp/opnfv_testapi/resources/sut_models.py @@ -0,0 +1,31 @@ +############################################################################## +# Copyright (c) 2017 +# 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 +############################################################################## +from opnfv_testapi.resources import models +from opnfv_testapi.tornado_swagger import swagger + + +@swagger.model() +class Sut(models.ModelBase): + """ + """ + def __init__(self): + pass + + +@swagger.model() +class Suts(models.ModelBase): + """ + @property suts: + @ptype tests: C{list} of L{Sut} + """ + def __init__(self): + self.suts = list() + + @staticmethod + def attr_parser(): + return {'suts': Sut} diff --git a/cvp/opnfv_testapi/router/url_mappings.py b/cvp/opnfv_testapi/router/url_mappings.py index 73c771fe..83190ee4 100644 --- a/cvp/opnfv_testapi/router/url_mappings.py +++ b/cvp/opnfv_testapi/router/url_mappings.py @@ -11,6 +11,7 @@ from opnfv_testapi.resources import handlers from opnfv_testapi.resources import result_handlers from opnfv_testapi.resources import test_handlers from opnfv_testapi.resources import application_handlers +from opnfv_testapi.resources import sut_handlers from opnfv_testapi.ui.auth import sign from opnfv_testapi.ui.auth import user @@ -27,6 +28,7 @@ mappings = [ (r"/api/v1/cvp/applications/([^/]+)", application_handlers.ApplicationsGURHandler), + (r"/api/v1/suts/hardware/([^/]+)", sut_handlers.HardwareHandler), (r'/api/v1/auth/signin', sign.SigninHandler), |