69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
# Copyright 2015 Huawei Technologies Co.,LTD.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from oslo_config import cfg
|
|
|
|
from magnum.common.pythonk8sclient.client import ApivbetaApi
|
|
from magnum.common.pythonk8sclient.client import swagger
|
|
from magnum.conductor import utils
|
|
from magnum.i18n import _
|
|
|
|
|
|
kubernetes_opts = [
|
|
cfg.StrOpt('k8s_protocol',
|
|
default='http',
|
|
help=_('Default protocol of k8s master endpoint '
|
|
'(http or https).')),
|
|
cfg.IntOpt('k8s_port',
|
|
default=8080,
|
|
help=_('Default port of the k8s master endpoint.')),
|
|
]
|
|
|
|
cfg.CONF.register_opts(kubernetes_opts, group='kubernetes')
|
|
|
|
|
|
class K8sAPI(ApivbetaApi.ApivbetaApi):
|
|
|
|
def __init__(self, context, obj):
|
|
# retrieve the URL of the k8s API endpoint
|
|
k8s_api_endpoint = self._retrieve_k8s_api_endpoint(context, obj)
|
|
|
|
# build a connection with Kubernetes master
|
|
client = swagger.ApiClient(k8s_api_endpoint)
|
|
|
|
super(K8sAPI, self).__init__(client)
|
|
|
|
@staticmethod
|
|
def _retrieve_k8s_api_endpoint(context, obj):
|
|
if hasattr(obj, 'bay_uuid'):
|
|
obj = utils.retrieve_bay(context, obj)
|
|
|
|
params = {
|
|
'k8s_protocol': cfg.CONF.kubernetes.k8s_protocol,
|
|
'api_address': obj.api_address
|
|
}
|
|
return "%(k8s_protocol)s://%(api_address)s" % params
|
|
|
|
|
|
def create_k8s_api(context, obj):
|
|
"""Create a kubernetes API client
|
|
|
|
Creates connection with Kubernetes master and creates ApivbetaApi instance
|
|
to call Kubernetes APIs.
|
|
|
|
:param context: The security context
|
|
:param obj: A bay or a k8s object (Pod, Service, ReplicationController)
|
|
"""
|
|
return K8sAPI(context, obj)
|