#!/usr/bin/python3
import json
import datetime
import sys
import os
import math
import urllib3
import requests
import re

urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)

zabbix_sender = "/usr/local/zabbix/bin/zabbix_sender"
zabbix_server = "192.168.1.233"
clientname = "k8s_master"

# k8s的api地址
url = "https://192.168.1.140:6443"

# k8s的token
token = "e11111111-k8s-token-111111w"


def get_result(api_name):
    headers = {"Authorization": "Bearer " + token}
    json_data = requests.get(url + api_name, headers=headers, verify=False)
    return json_data.json()


# 时间转换,把所有服务的创建时间变成运行时间
def Trantime(time_data):
    time = re.match(r"(\d+)-(\d+)-(\d+).*?(\d+):(\d+):(\d+)", time_data)
    total_send = round((datetime.datetime.now() - datetime.datetime(int(time.group(1)), int(time.group(2)),
                                                                    int(time.group(3)), int(time.group(4)),
                                                                    int(time.group(5)),
                                                                    int(time.group(6)))).total_seconds())
    return str(math.floor(total_send / 86400)).split(".")[0] + "d" + \
           str(math.floor((total_send % 86400) / 3600)).split(".")[0] + "h"


# 把所有的单位转换成b
def Tranunit(unit_data):
    unit = re.match(r"(\d+)(.*)", unit_data)
    value = int(unit.group(1))
    if unit.group(2) == "K":
        return value * 1000
    elif unit.group(2) == "Ki":
        return value * 1024
    elif unit.group(2) == "M":
        return value * 1000 * 1000
    elif unit.group(2) == "Mi":
        return value * 1024 * 1024
    elif unit.group(2) == "G":
        return value * 1000 * 1000 * 1000
    elif unit.group(2) == "Gi":
        return value * 1024 * 1024 * 1024
    elif unit.group(2) == "n":
        return math.ceil(value / 1000 / 1000)
    elif unit.group(2) == "m":
        return value
    else:
        return value


# 对数据进行zabbix_sender发送
def send_data(data, dis_key, key):
    result = json.dumps({"data": data}, sort_keys=True, ensure_ascii=False)
    cmd = "{0} -z {1} -s {2} -k {3} -o '{4}'>/dev/null".format(zabbix_sender, zabbix_server, clientname, dis_key,
                                                               result)
    os.system(cmd)
    for i in data:
        for value in i.keys():
            if value != "{#NAME}":
                cmd = "{0} -z {1} -s {2} -k {3}[{4}.{5}] -o '{6}'>/dev/null".format(zabbix_sender, zabbix_server,
                                                                                    clientname, key, i["{#NAME}"],
                                                                                    i[value][0], i[value][1])
                os.system(cmd)
                print(cmd)


# 获取所有k8s节点的信息
def get_node():
    get_nodes = []
    node_result = get_result("/api/v1/nodes")
    node_use_result = get_result("/apis/metrics.k8s.io/v1beta1/nodes")
    for j in node_use_result.get("items"):
        for i in node_result.get("items"):
            if i.get("metadata").get("name") == j.get("metadata").get("name"):
                data = {"{#NAME}": i.get("metadata").get("name"),
                        "{#STATUS}": ["status", i.get("status").get("conditions")[-1].get("type") if
                        i.get("status").get("conditions")[-1].get("status") == "True" else "NotReady"],
                        "{#IP}": ["ip", i.get("status").get("addresses")[0].get("address")],
                        "{#KUBELET_VERSION}": ["version", i.get("status").get("nodeInfo").get("kubeletVersion")],
                        "{#OS_IMAGE}": ["os_image", i.get("status").get("nodeInfo").get("osImage")],
                        "{#CPU}": ["cpu", i.get("status").get("capacity").get("cpu")],
                        "{#MEMORY}": ["memory", Tranunit(i.get("status").get("capacity").get("memory"))],
                        "{#LIMIT_STORAGE}": ["storage",
                                             Tranunit(i.get("status").get("capacity").get("ephemeral-storage"))],
                        "{#RUNTIME}": ["runtime", Trantime(i.get("metadata").get("creationTimestamp"))],
                        "{#USECPU}": ["usecpu", Tranunit(j.get("usage").get("cpu"))],
                        "{#USEMEMORY}": ["usememory", Tranunit(j.get("usage").get("memory"))]
                        }

        get_nodes.append(data)
    send_data(get_nodes, "getnode", "node")


# 获取k8s组件的健康信息
def get_health():
    get_healths = []
    health_result = get_result("/api/v1/componentstatuses")
    for i in health_result.get("items"):
        data = {}
        data = {"{#NAME}": i.get("metadata").get("name"),
                "{#STATUS}": ["status", i.get("conditions")[0].get("type")],
                "{#MESSAGE}": ["message", i.get("conditions")[0].get("message")]
                }
        get_healths.append(data)
    send_data(get_healths, "gethealth", "health")


# 获取k8s的所有pod
def get_pod():
    get_pods = []
    pod_result = get_result("/api/v1/pods")
    for i in pod_result.get("items"):
        data = {"{#NAME}": i.get("metadata").get("name"),
                "{#RUNTIME}": ["runtime", Trantime(i.get("metadata").get("creationTimestamp"))],
                "{#STATUS}": ["status", i.get("status").get("phase")],
                "{#RESTARTCOUNT}": ["restartcount", i.get("status").get("containerStatuses")[0].get("restartCount")]
                }

        get_pods.append(data)
    send_data(get_pods, "getpod", "pod")


if __name__ == "__main__":
    cmd = "{0}()".format(sys.argv[1])
    eval(cmd)
    print("采集完成")