Hexor/OutFleet
1
0
Fork 0
mirror of https://github.com/house-of-vanity/OutFleet.git synced 2025-07-07 01:24:06 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix V1 api define

Browse Source
This commit is contained in:
Alexandr Bogomyakov
2024-03-18 20:02:33 +02:00
parent 614140840d
commit 0e7cabe336
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
k8s.py
View File

@ -33,12 +33,12 @@ def write_config(config):
data={"config.yaml": yaml.dump(config)} data={"config.yaml": yaml.dump(config)}
) )
try: try:
api_response = v1.create_namespaced_config_map( api_response = V1.create_namespaced_config_map(
namespace=NAMESPACE, namespace=NAMESPACE,
body=config_map, body=config_map,
) )
except ApiException as e: except ApiException as e:
api_response = v1.patch_namespaced_config_map( api_response = V1.patch_namespaced_config_map(
name="config-outfleet", name="config-outfleet",
namespace=NAMESPACE, namespace=NAMESPACE,
body=config_map, body=config_map,
@ -48,18 +48,22 @@ def write_config(config):
NAMESPACE = False NAMESPACE = False
SERVERS = list() SERVERS = list()
CONFIG = None CONFIG = None
V1 = None
try: try:
config.load_incluster_config() config.load_incluster_config()
v1 = client.CoreV1Api() V1 = client.CoreV1Api()
with open("/var/run/secrets/kubernetes.io/serviceaccount/namespace") as f: with open("/var/run/secrets/kubernetes.io/serviceaccount/namespace") as f:
NAMESPACE = f.read().strip() NAMESPACE = f.read().strip()
log.info(f"Found Kubernetes environment. Deployed to namespace '{NAMESPACE}'") log.info(f"Found Kubernetes environment. Deployed to namespace '{NAMESPACE}'")
CONFIG = yaml.safe_load(V1.read_namespaced_config_map(name="config-outfleet", namespace=NAMESPACE).data['config.yaml'])
log.info(f"ConfigMap config.yaml loaded from Kubernetes API. Servers: {len(CONFIG['servers'])}, Clients: {len(CONFIG['clients'])}")
except: except:
log.info("Kubernetes environment not detected") log.info("Kubernetes environment not detected")
try: try:
CONFIG = yaml.safe_load(v1.read_namespaced_config_map(name="config-outfleet", namespace=NAMESPACE).data['config.yaml']) CONFIG = yaml.safe_load(V1.read_namespaced_config_map(name="config-outfleet", namespace=NAMESPACE).data['config.yaml'])
log.info(f"ConfigMap config.yaml loaded from Kubernetes API. Servers: {len(CONFIG['servers'])}, Clients: {len(CONFIG['clients'])}") log.info(f"ConfigMap config.yaml loaded from Kubernetes API. Servers: {len(CONFIG['servers'])}, Clients: {len(CONFIG['clients'])}")
except ApiException as e: except ApiException as e:
log.warning(f"ConfigMap not found. Fisrt run?") log.warning(f"ConfigMap not found. Fisrt run?")
@ -67,5 +71,5 @@ except ApiException as e:
if not CONFIG: if not CONFIG:
log.info(f"Creating new ConfigMap [config-outfleet]") log.info(f"Creating new ConfigMap [config-outfleet]")
write_config({"clients": [], "servers": [], "ui_hostname": "accessible-address.com"}) 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']) CONFIG = yaml.safe_load(V1.read_namespaced_config_map(name="config-outfleet", namespace=NAMESPACE).data['config.yaml'])