Dev: integration fixes while generating helm-scripts dynamically.

automation_script.py

@@ -4,9 +4,11 @@ import sys
 
 from dotenv import load_dotenv
 
-from docker_automation.docker_automate_script import DockerVersionUpgrade
-
 load_dotenv()
+from helm_automation.helm_register import HelmRegistration
+
+from docker_automation.docker_automate_script import DockerVersionUpgrade
+from helm_automation.helm_automate_script import HelmVersionUpgrade
 from docker_automation.docker_register import DockerRegistration
 from scripts.config import EnvironmentVariables
 from scripts.logging import logger
@@ -25,6 +27,10 @@ def func_mapper(operation_type, parameters):
         "docker_registration": DockerRegistration(arguments=parameters, git_user_name=git_user_name,
                                                   git_access_token=git_access_token).module_registration,
         "docker_version_upgrade": DockerVersionUpgrade(arguments=parameters, git_user_name=git_user_name,
+                                                       git_access_token=git_access_token).module_registration,
+        "helm_version_upgrade": HelmVersionUpgrade(arguments=parameters, git_user_name=git_user_name,
+                                                   git_access_token=git_access_token).module_registration,
+        "helm_registration": HelmRegistration(arguments=parameters, git_user_name=git_user_name,
                                               git_access_token=git_access_token).module_registration
     }
     return func_mapper_dict.get(operation_type)

docker_automation/docker_automate_script.py

@@ -37,9 +37,9 @@ class DockerVersionUpgrade:
             _module_names = self.arguments["module_names"]
             _branch_name = self.arguments['branch_name'] or "master"
             _ilens_version = self.arguments["ilens_version"]
-            if not _ilens_version or not _release_version or not _client_name or not _git_repos and not _module_names or not global_configmap:
+            if not _ilens_version or not _release_version or not _client_name or not global_configmap:
                 print(
-                    "global_configmap, git_repos, module_names, client_name, ilens_version and release_version details "
+                    "global_configmap, client_name, ilens_version and release_version details "
                     "not found!!!!!")
                 sys.exit()
             variables_file = "variables.yml"

scripts/core/docker_handler.py

@@ -3,8 +3,9 @@ import os
 from scripts.config import EnvironmentVariables
 from scripts.core import ILensVersionHandler
 from scripts.core.git_handler import GitHandler
+from scripts.db.psql.databases import get_db_for_func
 from scripts.logging import logging
-from scripts.schemas import DockerComposeSchema
+from scripts.schemas import DockerComposeSchema, GetRequest
 from scripts.utils.common_utils import CommonUtils
 
 
@@ -39,13 +40,12 @@ class DockerHandler:
                     logging.debug("Failed to clone module!! Skipping Docker File Preparation")
                     continue
                 _module_data = self.common_util.convert_yaml_to_define_obj(variables_file_path)
-                # session_obj = get_db_for_func()
-                # module_info = self.db_handler.get_module_versions(
-                #     input_data=GetRequest(module_name=_module, client='iLens', ilens_version=ilens_version,
-                #                           release_version=release_version), db=session_obj)
-                # session_obj.close()
+                session_obj = get_db_for_func()
+                module_info = self.db_handler.get_module_versions(
+                    input_data=GetRequest(module_name=_module, client='iLens', ilens_version=ilens_version,
+                                          release_version=release_version), db=session_obj)
+                session_obj.close()
                 _module_data = self.common_util.convert_yaml_to_define_obj(variables_file_path)
-                module_info = {}
                 module_env_variables = _module_data.get('deployment', {}).get('environmentVar', [])
                 module_env_variables = {_v['name']: _v.get('value') for _v in module_env_variables}
                 image_url = module_info.get("image_url", '') if module_info else ''
@@ -105,11 +105,20 @@ class DockerHandler:
 
     def process_compose_data_by_template(self, modules, compose_info: dict, variables_file, arguments,
                                          tmp_path, output_path):
+        ilens_version = arguments.get("ilens_version")
+        branch_name = arguments.get("branch_name")
+        release_version = arguments.get("release_version")
         compose_dict = {"services": {}}
         try:
             for module in modules:
+                session_obj = get_db_for_func()
+                module_info = self.db_handler.get_module_versions(
+                    input_data=GetRequest(module_name=module, client='iLens', ilens_version=ilens_version,
+                                          release_version=release_version), db=session_obj)
+                session_obj.close()
+                image_url = module_info.get("image_url", '') if module_info else ''
                 compose_dict['services'].update({
-                    module: DockerComposeSchema().dict()
+                    module: DockerComposeSchema(image=image_url).dict()
                 })
             compose_out_file_path = os.path.join(output_path, "docker-compose.yml")
             jinja_template_file = 'docker_deployment.yml'

templates/docker_deployment.yml

-version: "2"
+{% for k,v in modules.items() %}
+x-{{ k }}-image: &{{k}}-image {{v['image']}}
+{% endfor %}
 services:
   {% for k,v in modules.items() %}
 {{k}}:
-    image: {{v['image']}}
+    image: *{{k}}-image
     ports:
       - {{v['ports']}}
     environment:
@@ -16,3 +18,8 @@ services:
       - {{volume}}
 {% endfor %}
   {% endfor %}
+
+networks:
+  default:
+    external:
+      name: "ilens"
\ No newline at end of file