CH-87 Fix wrong "overrides" section generation in some cases

Author: aranega

tools/deployment-cli-tools/ch_cli_tools/skaffold.py

@@ -51,7 +51,7 @@ def build_artifact(image_name, context_path, requirements=None, dockerfile_path=
                                          in requirements]
         return artifact_spec
 
-    
+
     base_images = set()
 
     def process_build_dockerfile(dockerfile_path, root_path, global_context=False, requirements=None, app_name=None):
@@ -78,21 +78,21 @@ def process_build_dockerfile(dockerfile_path, root_path, global_context=False, r
 
         for dockerfile_path in base_dockerfiles:
             process_build_dockerfile(dockerfile_path, root_path, global_context=True)
-            
-    
+
+
     release_config = skaffold_conf['deploy']['helm']['releases'][0]
     release_config['name'] = helm_values.namespace
     release_config['namespace'] = helm_values.namespace
     release_config['artifactOverrides'][KEY_APPS] = {}
-    
+
     static_images = set()
     for root_path in root_paths:
         static_dockerfiles = find_dockerfiles_paths(
             join(root_path, STATIC_IMAGES_PATH))
 
         for dockerfile_path in static_dockerfiles:
             process_build_dockerfile(dockerfile_path, root_path)
-           
+
 
     for root_path in root_paths:
         apps_path = join(root_path, APPS_PATH)
@@ -138,22 +138,33 @@ def process_build_dockerfile(dockerfile_path, root_path, global_context=False, r
                         }
                 }
 
-            flask_main = find_file_paths(context_path, '__main__.py')
+            mains_candidates = find_file_paths(context_path, '__main__.py')
+
+            def identify_flask_main(candidates):
+                import re
+                init_flask_pattern = re.compile(r"init_flask\(")
+                for candidate in candidates:
+                    with open(f"{candidate}/__main__.py", 'r') as file:
+                        if re.search(init_flask_pattern, file.read()):
+                            return candidate
+                return None
+
+            flask_main = identify_flask_main(mains_candidates)
 
             if flask_main:
                 release_config['overrides']['apps'][app_key] = \
                     {
                         'harness': {
                             'deployment': {
                                 'command': ['python'],
-                                'args': [f'/usr/src/app/{os.path.basename(flask_main[0])}/__main__.py']
+                                'args': [f'/usr/src/app/{os.path.basename(flask_main)}/__main__.py']
                             }
                         }
                 }
-            
+
             test_config: ApplicationTestConfig = helm_values.apps[app_key].harness.test
             if test_config.unit.enabled and test_config.unit.commands:
-                    
+
                     skaffold_conf['test'].append(dict(
                         image=get_image_tag(app_name),
                         custom=[dict(command="docker run $IMAGE " + cmd) for cmd in test_config.unit.commands]
@@ -209,7 +220,7 @@ def get_image_tag(name):
                                                                                                        "/usr/src/app"),
                     }
                 })
-                
+
 
     if not os.path.exists(os.path.dirname(vscode_launch_path)):
         os.makedirs(os.path.dirname(vscode_launch_path))