Skip to content

Commit cec399d

Browse files
Feat/addding kubernetes api client setup (#4)
* feat: removing slowapi and adding setup_kubernetes
1 parent b3a67d4 commit cec399d

File tree

2 files changed

+27
-3
lines changed

2 files changed

+27
-3
lines changed

glueops/setup_kubernetes.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
import os
2+
import kubernetes
3+
4+
def load_kubernetes_config(logger):
5+
try:
6+
if os.getenv('KUBERNETES_SERVICE_HOST'):
7+
logger.info("KUBERNETES_SERVICE_HOST has a value so we must be running inside of kubernetes")
8+
logger.info('Attempting to load in-cluster Kubernetes configuration')
9+
kubernetes.config.load_incluster_config()
10+
else:
11+
logger.info('Using local Kubeconfig because we are not running in kubernetes')
12+
kubernetes.config.load_kube_config()
13+
logger.info('Successfully loaded Kubernetes configuration')
14+
except kubernetes.config.ConfigException as e:
15+
logger.error(f'Kubernetes configuration error: {e}')
16+
os.sys.exit('Failed to load Kubernetes configuration, exiting application')
17+
18+
try:
19+
v1 = kubernetes.client.CoreV1Api()
20+
custom_api = kubernetes.client.CustomObjectsApi()
21+
return v1, custom_api
22+
except Exception as e:
23+
logger.error(f'Error setting up Kubernetes API clients: {e}')
24+
os.sys.exit('Failed to setup Kubernetes API clients, exiting application')

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,13 @@
33

44
setup(
55
name='glueops-helpers',
6-
version='0.2.0',
6+
version='0.3.0',
77
packages=find_packages(),
88
install_requires=[
99
'requests',
1010
'boto3',
11-
'slowapi',
12-
'cryptography'
11+
'cryptography',
12+
'kubernetes'
1313
],
1414
entry_points={
1515
# If needed, you can add entry points for command line utilities here

0 commit comments

Comments
 (0)