2021-12-13 02:38:10 +01:00
|
|
|
#!/usr/bin/env python3
|
2021-12-13 03:26:32 +01:00
|
|
|
import logging
|
|
|
|
|
import time
|
2021-12-13 02:38:10 +01:00
|
|
|
|
2021-12-13 03:26:32 +01:00
|
|
|
from kubernetes import config, client
|
|
|
|
|
from kubernetes.client.rest import ApiException
|
2021-12-13 02:38:10 +01:00
|
|
|
|
2021-12-13 03:26:32 +01:00
|
|
|
config.load_incluster_config()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def ensure_running_pod_on_site(site: str):
|
|
|
|
|
# Configs can be set in Configuration class directly or using helper utility
|
|
|
|
|
logger = logging.getLogger("k8s")
|
|
|
|
|
|
|
|
|
|
with client.ApiClient() as api_client:
|
|
|
|
|
# Create an instance of the API class
|
|
|
|
|
api_instance = client.CustomObjectsApi(api_client)
|
|
|
|
|
group = 'types.kubefed.io' # str | the custom resource's group
|
|
|
|
|
version = 'v1beta1' # str | the custom resource's version
|
|
|
|
|
namespace = 'birbnetes' # str | The custom resource's namespace
|
|
|
|
|
plural = 'federateddeployments' # str | the custom resource's plural name. For TPRs this would be lowercase plural kind.
|
|
|
|
|
name = f'svm-prefilter-service-{site}' # str | the custom object's name
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
api_response = api_instance.get_namespaced_custom_object(group, version, namespace, plural, name)
|
|
|
|
|
except ApiException as e:
|
|
|
|
|
logger.error("Exception when calling CustomObjectsApi->get_namespaced_custom_object: %s\n" % e)
|
|
|
|
|
raise
|
|
|
|
|
|
|
|
|
|
if api_response['spec']['template']['spec']['replicas'] != 0:
|
|
|
|
|
logging.debug("It seems like that this site have working replicas")
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
api_response['spec']['template']['spec']['replicas'] = 1
|
|
|
|
|
|
|
|
|
|
api_instance.patch_namespaced_custom_object(group, version, namespace, plural, name, api_response)
|
|
|
|
|
|
|
|
|
|
# TODO: Wait for pod to start up
|
|
|
|
|
time.sleep(7) # close enough
|
