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.sol005 import k8scluster
  38 from osmclient.sol005 import repo
  39 from osmclient.sol005 import osmrepo
  40 from osmclient.common import package_tool
  41 import json
  42 import logging
  43
  44
  45 class Client(object):
  46     def __init__(
  47         self,
  48         host=None,
  49         so_port=9999,
  50         user="admin",
  51         password="admin",
  52         project="admin",
  53         **kwargs
  54     ):
  55
  56         self._user = user
  57         self._password = password
  58         self._project = project
  59         self._project_domain_name = kwargs.get("project_domain_name")
  60         self._user_domain_name = kwargs.get("user_domain_name")
  61         self._logger = logging.getLogger("osmclient")
  62         self._auth_endpoint = "/admin/v1/tokens"
  63         self._headers = {}
  64         self._token = None
  65         if len(host.split(":")) > 1:
  66             # backwards compatible, port provided as part of host
  67             self._host = host.split(":")[0]
  68             self._so_port = host.split(":")[1]
  69         else:
  70             self._host = host
  71             self._so_port = so_port
  72
  73         self._http_client = http.Http(
  74             "https://{}:{}/osm".format(self._host, self._so_port), **kwargs
  75         )
  76         self._headers["Accept"] = "application/json"
  77         self._headers["Content-Type"] = "application/yaml"
  78         http_header = [
  79             "{}: {}".format(key, val) for (key, val) in list(self._headers.items())
  80         ]
  81         self._http_client.set_http_header(http_header)
  82
  83         self.vnfd = vnfd.Vnfd(self._http_client, client=self)
  84         self.nsd = nsd.Nsd(self._http_client, client=self)
  85         self.nst = nst.Nst(self._http_client, client=self)
  86         self.package = package.Package(self._http_client, client=self)
  87         self.ns = ns.Ns(self._http_client, client=self)
  88         self.nsi = nsi.Nsi(self._http_client, client=self)
  89         self.vim = vim.Vim(self._http_client, client=self)
  90         self.wim = wim.Wim(self._http_client, client=self)
  91         self.sdnc = sdncontroller.SdnController(self._http_client, client=self)
  92         self.vnf = vnf.Vnf(self._http_client, client=self)
  93         self.project = projectmodule.Project(self._http_client, client=self)
  94         self.user = usermodule.User(self._http_client, client=self)
  95         self.role = role.Role(self._http_client, client=self)
  96         self.pdu = pdud.Pdu(self._http_client, client=self)
  97         self.k8scluster = k8scluster.K8scluster(self._http_client, client=self)
  98         self.repo = repo.Repo(self._http_client, client=self)
  99         self.osmrepo = osmrepo.OSMRepo(self._http_client, client=self)
 100         self.package_tool = package_tool.PackageTool(client=self)
 101         """
 102         self.vca = vca.Vca(http_client, client=self, **kwargs)
 103         self.utils = utils.Utils(http_client, **kwargs)
 104         """
 105
 106     def get_token(self):
 107         self._logger.debug("")
 108         if self._token is None:
 109             postfields_dict = {
 110                 "username": self._user,
 111                 "password": self._password,
 112                 "project_id": self._project,
 113             }
 114             if self._project_domain_name:
 115                 postfields_dict["project_domain_name"] = self._project_domain_name
 116             if self._user_domain_name:
 117                 postfields_dict["user_domain_name"] = self._user_domain_name
 118             http_code, resp = self._http_client.post_cmd(
 119                 endpoint=self._auth_endpoint,
 120                 postfields_dict=postfields_dict,
 121                 skip_query_admin=True,
 122             )
 123             #            if http_code not in (200, 201, 202, 204):
 124             #                message ='Authentication error: not possible to get auth token\nresp:\n{}'.format(resp)
 125             #                raise ClientException(message)
 126
 127             token = json.loads(resp) if resp else None
 128             self._token = token["id"]
 129
 130             if self._token is not None:
 131                 self._headers["Authorization"] = "Bearer {}".format(self._token)
 132                 http_header = [
 133                     "{}: {}".format(key, val)
 134                     for (key, val) in list(self._headers.items())
 135                 ]
 136                 self._http_client.set_http_header(http_header)
 137
 138     def get_version(self):
 139         _, resp = self._http_client.get2_cmd(endpoint="/version", skip_query_admin=True)
 140         # print(http_code, resp)
 141         try:
 142             resp = json.loads(resp)
 143             version = resp.get("version")
 144             date = resp.get("date")
 145         except ValueError:
 146             version = resp.split()[2]
 147             date = resp.split()[4]
 148         return "{} {}".format(version, date)
 149
 150     def set_default_params(self, **kwargs):
 151         host = kwargs.pop("host", None)
 152         if host is not None:
 153             self._host = host
 154         port = kwargs.pop("port", None)
 155         if port is not None:
 156             self._so_port = port
 157         self._http_client.set_query_admin(**kwargs)