osmclient/sol005/client.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 SOL005 client API
  19 """
  20
  21 #from osmclient.v1 import vca
  22 from osmclient.sol005 import vnfd
  23 from osmclient.sol005 import nsd
  24 from osmclient.sol005 import nst
  25 from osmclient.sol005 import nsi
  26 from osmclient.sol005 import ns
  27 from osmclient.sol005 import vnf
  28 from osmclient.sol005 import vim
  29 from osmclient.sol005 import wim
  30 from osmclient.sol005 import package
  31 from osmclient.sol005 import http
  32 from osmclient.sol005 import sdncontroller
  33 from osmclient.sol005 import project as projectmodule
  34 from osmclient.sol005 import user as usermodule
  35 from osmclient.sol005 import role
  36 from osmclient.sol005 import pdud
  37 from osmclient.common.exceptions import ClientException
  38 import json
  39
  40
  41 class Client(object):
  42
  43     def __init__(
  44         self,
  45         host=None,
  46         so_port=9999,
  47         user='admin',
  48         password='admin',
  49         project='admin',
  50         **kwargs):
  51
  52         self._user = user
  53         self._password = password
  54         self._project = project
  55         self._auth_endpoint = '/admin/v1/tokens'
  56         self._headers = {}
  57
  58         if len(host.split(':')) > 1:
  59             # backwards compatible, port provided as part of host
  60             self._host = host.split(':')[0]
  61             self._so_port = host.split(':')[1]
  62         else:
  63             self._host = host
  64             self._so_port = so_port
  65
  66         self._http_client = http.Http(
  67             'https://{}:{}/osm'.format(self._host,self._so_port))
  68         self._headers['Accept'] = 'application/json'
  69         self._headers['Content-Type'] = 'application/yaml'
  70         http_header = ['{}: {}'.format(key, val)
  71                        for (key, val) in list(self._headers.items())]
  72         self._http_client.set_http_header(http_header)
  73
  74         token = self.get_token()
  75         if not token:
  76             raise ClientException(
  77                     'Authentication error: not possible to get auth token')
  78         self._headers['Authorization'] = 'Bearer {}'.format(token)
  79         http_header.append('Authorization: Bearer {}'.format(token))
  80         self._http_client.set_http_header(http_header)
  81
  82         self.vnfd = vnfd.Vnfd(self._http_client, client=self)
  83         self.nsd = nsd.Nsd(self._http_client, client=self)
  84         self.nst = nst.Nst(self._http_client, client=self)
  85         self.package = package.Package(self._http_client, client=self)
  86         self.ns = ns.Ns(self._http_client, client=self)
  87         self.nsi = nsi.Nsi(self._http_client, client=self)
  88         self.vim = vim.Vim(self._http_client, client=self)
  89         self.wim = wim.Wim(self._http_client, client=self)
  90         self.sdnc = sdncontroller.SdnController(self._http_client, client=self)
  91         self.vnf = vnf.Vnf(self._http_client, client=self)
  92         self.project = projectmodule.Project(self._http_client, client=self)
  93         self.user = usermodule.User(self._http_client, client=self)
  94         self.role = role.Role(self._http_client, client=self)
  95         self.pdu = pdud.Pdu(self._http_client, client=self)
  96         '''
  97         self.vca = vca.Vca(http_client, client=self, **kwargs)
  98         self.utils = utils.Utils(http_client, **kwargs)
  99         '''
 100
 101     def get_token(self):
 102         postfields_dict = {'username': self._user,
 103                            'password': self._password,
 104                            'project_id': self._project}
 105         http_code, resp = self._http_client.post_cmd(endpoint=self._auth_endpoint,
 106                                                      postfields_dict=postfields_dict)
 107         if http_code not in (200, 201, 202, 204):
 108             raise ClientException(resp)
 109         token = json.loads(resp) if resp else None
 110         if token is not None:
 111             return token['_id']
 112         return None
 113