feat: Smaller code refactoring

src/api/api.py

@@ -15,11 +15,11 @@
 
 
 class CreateAnalysis(BaseModel):
-    analysis_id: str = 'flame-test'
-    project_id: str = 'project1'
-    registry_url: str = ''
-    registry_user: str = ''
-    registry_password: str = ''
+    analysis_id: str = 'analysis_id'
+    project_id: str = 'project_id'
+    registry_url: str = 'harbor.privateaim'
+    registry_user: str = 'robot_user'
+    registry_password: str = 'default_pw'
 
 
 @router.post("/", response_class=JSONResponse)

src/k8s/kubernetes.py

@@ -163,18 +163,22 @@ def _create_nginx_config_map(analysis_name: str,
 
     # get the service ip of the message broker and analysis service
     message_broker_service_name = get_element_by_substring(service_names, 'message-broker')
-    message_broker_service_ip = core_client.read_namespaced_service(name=message_broker_service_name, namespace=namespace).spec.cluster_ip
+    message_broker_service_ip = core_client.read_namespaced_service(name=message_broker_service_name,
+                                                                    namespace=namespace).spec.cluster_ip
 
     # wait until analysis pod receives a cluster ip
     analysis_ip = None
     while analysis_ip is None:
-        pod_list_object = core_client.list_namespaced_pod(label_selector=f"app={analysis_name}", watch=False, namespace=namespace)
+        pod_list_object = core_client.list_namespaced_pod(label_selector=f"app={analysis_name}",
+                                                          watch=False,
+                                                          namespace=namespace)
         analysis_ip = pod_list_object.items[0].status.pod_ip
         print(analysis_ip)
         time.sleep(1)
 
     # analysis_ip = core_client.read_namespaced_pod(name=analysis_name, namespace=namespace).spec.cluster_ip
-    # analysis_service_ip = core_client.read_namespaced_service(name=analysis_service_name, namespace=namespace).spec.cluster_ip
+    # analysis_service_ip = core_client.read_namespaced_service(name=analysis_service_name,
+    #                                                           namespace=namespace).spec.cluster_ip
     kong_proxy_name = get_element_by_substring(service_names, 'kong-proxy')
     result_service_name = get_element_by_substring(service_names, 'result-service')
     data = {

src/test/test_db.py

@@ -17,7 +17,7 @@ def tearDown(self):
 
     def test_create_analysis(self):
         analysis = self.database.create_analysis(analysis_id="123", image_registry_address="harbor.com",
-                                                 ports=[80, 443],pod_ids=["a1"], status="running")
+                                                 ports=[80, 443], pod_ids=["a1"], status="running")
         print(f"{analysis.analysis_id}, 123")
 
     def test_get_analysis(self):

src/utils/other.py

@@ -1,14 +1,7 @@
-import os
 from httpx import AsyncClient
 import asyncio
 
 
-# def create_image_address(analysis_id: str) -> str:
-#     return f"{_add_slash(os.getenv('HARBOR_URL'))}"\
-#            f"{_add_slash(os.getenv('NODE_NAME'))}"\
-#            f"{analysis_id}:latest"
-
-
 def _add_slash(string: str) -> str:
     return string + ('' if string.endswith('/') else '/')