update(app): try k8s fetch
This commit is contained in:
@@ -1,32 +1,33 @@
|
||||
from fastapi import APIRouter
|
||||
from pydantic import BaseModel
|
||||
from typing import List
|
||||
from database import insert_kubernetes_node, fetch_all
|
||||
from kubernetes import client, config
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
class KubernetesNode(BaseModel):
|
||||
cluster_name: str
|
||||
node_name: str
|
||||
cpu: int
|
||||
memory: str
|
||||
storage: str
|
||||
type: str
|
||||
namespaces: List[str]
|
||||
# Initialize Kubernetes client
|
||||
def init_k8s_client():
|
||||
try:
|
||||
config.load_incluster_config()
|
||||
except:
|
||||
config.load_kube_config() # local testing
|
||||
|
||||
@router.get("/k8s/data")
|
||||
def get_k8s_data():
|
||||
return {"kubernetes_nodes": fetch_all("kubernetes_nodes")}
|
||||
init_k8s_client()
|
||||
v1 = client.CoreV1Api()
|
||||
|
||||
@router.post("/k8s/data")
|
||||
def add_k8s_data(node: KubernetesNode):
|
||||
insert_kubernetes_node(
|
||||
cluster_name=node.cluster_name,
|
||||
node_name=node.node_name,
|
||||
cpu=node.cpu,
|
||||
memory=node.memory,
|
||||
storage=node.storage,
|
||||
node_type=node.type,
|
||||
namespaces=node.namespaces
|
||||
)
|
||||
return {"message": f"Kubernetes node '{node.node_name}' in cluster '{node.cluster_name}' added successfully."}
|
||||
# Fetch nodes
|
||||
nodes = v1.list_node()
|
||||
node_data = []
|
||||
for node in nodes.items:
|
||||
node_data.append({
|
||||
"node_name": node.metadata.name,
|
||||
"cpu": node.status.capacity.get("cpu"),
|
||||
"memory": node.status.capacity.get("memory"),
|
||||
"pods_allocatable": node.status.allocatable.get("pods"),
|
||||
})
|
||||
|
||||
# Fetch namespaces
|
||||
namespaces = v1.list_namespace()
|
||||
namespace_data = [ns.metadata.name for ns in namespaces.items]
|
||||
|
||||
return {"nodes": node_data, "namespaces": namespace_data}
|
||||
|
||||
Reference in New Issue
Block a user