Add ignore dirs

Author: MDUYN

investing_algorithm_framework/cli/cli.py

@@ -172,11 +172,17 @@ def deploy_azure_function(
     help='The path to the project directory containing '
          'the Lambda function code.'
 )
+@click.option(
+    '--ignore_dirs',
+    default=None,
+    help='List of directories to ignore when deploying.'
+)
 def deploy_aws_lambda(
     lambda_function_name,
     region,
     lambda_handler,
     project_dir=None,
+    ignore_dirs=None
 ):
     """
     Command-line tool for deploying a trading bot to AWS lambda
@@ -191,6 +197,7 @@ def deploy_aws_lambda(
         project_dir (str): The path to the project directory containing the
             Lambda function code. If not provided, it defaults to
             the current directory.
+        ignore_dirs (list): List of directories to ignore when deploying.
 
     Returns:
         None
@@ -199,7 +206,8 @@ def deploy_aws_lambda(
         lambda_function_name=lambda_function_name,
         region=region,
         lambda_handler=lambda_handler,
-        project_dir=project_dir
+        project_dir=project_dir,
+        ignore_dirs=ignore_dirs
     )
 
 

investing_algorithm_framework/cli/deploy_to_aws_lambda.py

@@ -39,14 +39,15 @@ def sanitize_bucket_name(name: str) -> str:
     return name  # Enforce length limit
 
 
-def zip_code(source_dir, zip_file):
+def zip_code(source_dir, zip_file, ignore_dirs=None):
     """
     Recursively zips the contents of source_dir into zip_file,
     preserving directory structure — suitable for AWS Lambda deployment.
 
     Args:
         source_dir: str, the directory containing the Lambda function code.
         zip_file: str, the path where the zip file will be created.
+        ignore_dirs: list, directories to ignore when zipping the code.
 
     Returns:
         None
@@ -56,6 +57,16 @@ def zip_code(source_dir, zip_file):
     # Function should recursively zip all files and directories
     with zipfile.ZipFile(zip_file, 'w', zipfile.ZIP_DEFLATED) as zf:
         for root, _, files in os.walk(source_dir):
+
+            # Skip ignored directories
+            if ignore_dirs is not None:
+                relative_root = os.path.relpath(root, source_dir)
+                if any(
+                    relative_root.startswith(ignore) for ignore in ignore_dirs
+                ):
+                    click.echo(f"Ignoring directory: {relative_root}")
+                    continue
+
             for file in files:
                 click.echo(f"Adding {file} to zip")
                 file_path = os.path.join(root, file)
@@ -347,6 +358,7 @@ def command(
     region,
     lambda_handler,
     project_dir=None,
+    ignore_dirs=None
 ):
     """
     Command-line tool for deploying a trading bot to AWS Lambda.
@@ -358,6 +370,7 @@ def command(
             If None, it defaults to the current directory.
         lambda_handler: str, the name of the handler function in the code
             (default is "aws_function.lambda_handler").
+        ignore_dirs: list, directories to ignore when zipping the code.
 
     Returns:
         None