aboutsummaryrefslogtreecommitdiffstats
path: root/sdv/docker/sdvstate/tools/kube_utils.py
blob: 3468ac67a735feea0a8ea112998f89016e3ab0c5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
# Copyright 2020 University Of Delhi.
#
# 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.



"""
Kubernetes cluster api helper functions
"""

from kubernetes import client, config
from kubernetes.stream import stream

from tools.conf import settings    # pylint: disable=import-error


def load_kube_api():
    """
    Loads kubernetes api
    """
    config.load_kube_config(settings.getValue('kube_config'))
    api = client.CoreV1Api()
    settings.setValue('kube_api', api)


def kube_api():
    """
    Returns kube_api object
    """
    return settings.getValue('kube_api')


def get_pod_with_labels(labels):
    """
    Returns json details any one pod with matching labels

    :param labels: labels to find matching pod
    :return: pod details
    """
    api = kube_api()
    pod = api.list_pod_for_all_namespaces(label_selector=labels).items[0]
    return pod


def kube_exec(pod, cmd):
    """
    Executes `cmd` inside `pod` and returns response

    :param pod: pod object
    :param cmd: command to execute inside pod
    :return: response from pod
    """
    api = kube_api()
    response = stream(api.connect_get_namespaced_pod_exec,
                      pod.metadata.name, pod.metadata.namespace, command=cmd,
                      stderr=True, stdin=False, stdout=True, tty=False)
    return response


def kube_curl(*args):
    """
    executes curl cmd in kubernetes network

    :param args: comma separated list of args to pass to curl
    :return: http response
    """
    args = list(args)
    args.insert(0, "curl")

    labels = "application=prometheus-openstack-exporter,component=exporter"
    pod = get_pod_with_labels(labels)

    response = kube_exec(pod, args)

    return response