mirror of
https://github.com/house-of-vanity/OutFleet.git
synced 2025-07-06 17:14:07 +00:00
75 lines
2.5 KiB
Python
Executable File
75 lines
2.5 KiB
Python
Executable File
import base64
|
|
import json
|
|
import yaml
|
|
import logging
|
|
from kubernetes import client, config as kube_config
|
|
from kubernetes.client.rest import ApiException
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
|
|
datefmt="%d-%m-%Y %H:%M:%S",
|
|
)
|
|
|
|
log = logging.getLogger("OutFleet.k8s")
|
|
file_handler = logging.FileHandler("sync.log")
|
|
file_handler.setLevel(logging.DEBUG)
|
|
formatter = logging.Formatter(
|
|
"%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
|
)
|
|
file_handler.setFormatter(formatter)
|
|
log.addHandler(file_handler)
|
|
|
|
def write_config(config):
|
|
config_map = client.V1ConfigMap(
|
|
api_version="v1",
|
|
kind="ConfigMap",
|
|
metadata=client.V1ObjectMeta(
|
|
name=f"config-outfleet",
|
|
labels={
|
|
"app": "outfleet",
|
|
}
|
|
),
|
|
data={"config.yaml": yaml.dump(config)}
|
|
)
|
|
try:
|
|
api_response = V1.create_namespaced_config_map(
|
|
namespace=NAMESPACE,
|
|
body=config_map,
|
|
)
|
|
except ApiException as e:
|
|
api_response = V1.patch_namespaced_config_map(
|
|
name="config-outfleet",
|
|
namespace=NAMESPACE,
|
|
body=config_map,
|
|
)
|
|
log.info("Updated config in Kubernetes ConfigMap [config-outfleet]")
|
|
|
|
|
|
NAMESPACE = False
|
|
SERVERS = list()
|
|
CONFIG = None
|
|
V1 = None
|
|
|
|
try:
|
|
kube_config.load_incluster_config()
|
|
V1 = client.CoreV1Api()
|
|
try:
|
|
with open("/var/run/secrets/kubernetes.io/serviceaccount/namespace") as f:
|
|
NAMESPACE = f.read().strip()
|
|
log.info(f"Found Kubernetes environment. Deployed to namespace '{NAMESPACE}'")
|
|
try:
|
|
CONFIG = yaml.safe_load(V1.read_namespaced_config_map(name="config-outfleet", namespace=NAMESPACE).data['config.yaml'])
|
|
log.info(f"ConfigMap loaded from Kubernetes API. Servers: {len(CONFIG['servers'])}, Clients: {len(CONFIG['clients'])}")
|
|
except Exception as e:
|
|
try:
|
|
write_config({"clients": [], "servers": {}, "ui_hostname": "accessible-address.com"})
|
|
CONFIG = yaml.safe_load(V1.read_namespaced_config_map(name="config-outfleet", namespace=NAMESPACE).data['config.yaml'])
|
|
log.info("Created new ConfigMap [config-outfleet]")
|
|
except Exception as e:
|
|
log.info(f"Failed to create new ConfigMap [config-outfleet] {e}")
|
|
except:
|
|
log.info("Kubernetes environment not detected")
|
|
except:
|
|
log.info("Kubernetes environment not detected")
|