osmclient/sol005/vim.py

   1 # Copyright 2018 Telefonica
   2 #
   3 # All Rights Reserved.
   4 #
   5 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
   6 #    not use this file except in compliance with the License. You may obtain
   7 #    a copy of the License at
   8 #
   9 #         http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 #    Unless required by applicable law or agreed to in writing, software
  12 #    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  13 #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  14 #    License for the specific language governing permissions and limitations
  15 #    under the License.
  16
  17 """
  18 OSM vim API handling
  19 """
  20
  21 from osmclient.common import utils
  22 from osmclient.common.exceptions import ClientException
  23 from osmclient.common.exceptions import NotFound
  24 import yaml
  25
  26
  27 class Vim(object):
  28     def __init__(self, http=None, client=None):
  29         self._http = http
  30         self._client = client
  31         self._apiName = '/admin'
  32         self._apiVersion = '/v1'
  33         self._apiResource = '/vims'
  34         self._apiBase = '{}{}{}'.format(self._apiName,
  35                                         self._apiVersion, self._apiResource)
  36     def create(self, name, vim_access):
  37         if 'vim-type' not in vim_access:
  38             #'openstack' not in vim_access['vim-type']):
  39             raise Exception("vim type not provided")
  40
  41         vim_account = {}
  42         vim_config = {'hello': 'hello'}
  43         vim_account['name'] = name
  44         vim_account = self.update_vim_account_dict(vim_account, vim_access)
  45
  46         vim_config = {}
  47         if 'config' in vim_access and vim_access['config'] is not None:
  48            vim_config = yaml.load(vim_access['config'])
  49
  50         vim_account['config'] = vim_config
  51
  52         resp = self._http.post_cmd(endpoint=self._apiBase,
  53                                        postfields_dict=vim_account)
  54         if not resp or 'id' not in resp:
  55             raise ClientException('failed to create vim {}: {}'.format(
  56                                   name, resp))
  57         else:
  58             print resp['id']
  59
  60     def update(self, vim_name, vim_account):
  61         vim = self.get(vim_name)
  62         #resp = self._http.put_cmd(endpoint='{}/{}'.format(self._apiBase,vim['_id']),
  63         resp = self._http.patch_cmd(endpoint='{}/{}'.format(self._apiBase,vim['_id']),
  64                                        postfields_dict=vim_account)
  65         #print 'RESP: {}'.format(resp)
  66         if not resp or 'id' not in resp:
  67             raise ClientException('failed to update vim: '.format(resp))
  68         else:
  69             print resp['id']
  70
  71     def update_vim_account_dict(self, vim_account, vim_access):
  72         vim_account['vim_type'] = vim_access['vim-type']
  73         vim_account['description'] = vim_access['description']
  74         vim_account['vim_url'] = vim_access['vim-url']
  75         vim_account['vim_user'] = vim_access['vim-username']
  76         vim_account['vim_password'] = vim_access['vim-password']
  77         vim_account['vim_tenant_name'] = vim_access['vim-tenant-name']
  78         return vim_account
  79
  80     def get_id(self, name):
  81         """Returns a VIM id from a VIM name
  82         """
  83         for vim in self.list():
  84             if name == vim['name']:
  85                 return vim['uuid']
  86         raise NotFound("vim {} not found".format(name))
  87
  88     def delete(self, vim_name):
  89         vim_id = vim_name
  90         if not utils.validate_uuid4(vim_name):
  91             vim_id = self.get_id(vim_name)
  92         http_code, resp = self._http.delete_cmd('{}/{}'.format(self._apiBase,vim_id))
  93         #print 'RESP: {}'.format(resp)
  94         if http_code == 202:
  95             print 'Deletion in progress'
  96         elif http_code == 204:
  97             print 'Deleted'
  98         else:
  99             raise ClientException("failed to delete vim {} - {}".format(vim_name, resp))
 100
 101     def list(self, filter=None):
 102         """Returns a list of VIM accounts
 103         """
 104         filter_string = ''
 105         if filter:
 106             filter_string = '?{}'.format(filter)
 107         resp = self._http.get_cmd('{}{}'.format(self._apiBase,filter_string))
 108         if not resp:
 109             return list()
 110         vim_accounts = []
 111         for datacenter in resp:
 112             vim_accounts.append({"name": datacenter['name'], "uuid": datacenter['_id']
 113                         if '_id' in datacenter else None})
 114         return vim_accounts
 115
 116     def get(self, name):
 117         """Returns a VIM account based on name or id
 118         """
 119         vim_id = name
 120         if not utils.validate_uuid4(name):
 121             vim_id = self.get_id(name)
 122         resp = self._http.get_cmd('{}/{}'.format(self._apiBase,vim_id))
 123         if not resp or '_id' not in resp:
 124             raise ClientException('failed to get vim info: '.format(
 125                                   resp))
 126         else:
 127             return resp
 128         raise NotFound("vim {} not found".format(name))
 129