working to download rule zip locally

Author: cyphronix

aws_sra_examples/solutions/genai/bedrock_org/lambda/src/sra_s3.py

@@ -122,3 +122,32 @@ def stage_code_to_s3(self, directory_path, bucket_name, s3_path):
                     self.LOGGER.info("Credentials not available")
                     return
                 self.LOGGER.info(f"Uploaded {local_path} to {bucket_name} {s3_file_path}")
+
+    def download_s3_file(self, rule_name, bucket_name):
+        """
+        Downloads the rule code from the staging S3 bucket.
+
+        :param rule_name: Name of the rule
+        :param bucket_name: Name of the S3 bucket
+        """
+        self.LOGGER.info(f"Downloading {rule_name} rule code from s3...")
+        s3_key_template = 'rules/{rule_name}/{rule_name}.zip'
+        local_base_path = '/tmp/sra_staging_upload'
+        
+        s3_key = s3_key_template.format(rule_name=rule_name)
+        local_file_path = os.path.join(local_base_path, 'rules', rule_name, f'{rule_name}.zip')
+        
+        # Ensure local directories exist
+        os.makedirs(os.path.dirname(local_file_path), exist_ok=True)
+        
+        try:
+            # Download the file from S3
+            self.S3_CLIENT.download_file(bucket_name, s3_key, local_file_path)
+        except NoCredentialsError:
+            self.LOGGER.info("Credentials not available")
+            return
+        # Handle other exceptions as needed
+        except Exception as e:
+            self.LOGGER.info(f"Error downloading file: {e}")
+        
+        self.LOGGER.info(f"File downloaded successfully to {local_file_path}")