Update test.py

test.py

@@ -5,6 +5,66 @@ import json
 import pandas as pd
 import re
 
+----
+import boto3
+import botocore
+
+def list_files_in_bucket(bucket_name, prefix=''):
+    """
+    List all files in a given S3 bucket.
+    
+    Parameters:
+    - bucket_name (str): The name of the S3 bucket.
+    - prefix (str): The prefix to filter files (useful for listing files in a specific folder).
+
+    Returns:
+    - list: A list of file keys in the bucket.
+    """
+    s3_client = boto3.client('s3')
+    try:
+        response = s3_client.list_objects_v2(Bucket=bucket_name, Prefix=prefix)
+        files = [content['Key'] for content in response.get('Contents', [])]
+        return files
+    except botocore.exceptions.ClientError as e:
+        print(f"Error listing files: {e}")
+        return []
+
+def read_file_from_s3(bucket_name, file_key):
+    """
+    Read the contents of a file in an S3 bucket.
+    
+    Parameters:
+    - bucket_name (str): The name of the S3 bucket.
+    - file_key (str): The key of the file in the S3 bucket.
+
+    Returns:
+    - str: The content of the file as a string.
+    """
+    s3_client = boto3.client('s3')
+    try:
+        obj = s3_client.get_object(Bucket=bucket_name, Key=file_key)
+        content = obj['Body'].read().decode('utf-8')
+        return content
+    except botocore.exceptions.ClientError as e:
+        print(f"Error reading file: {e}")
+        return None
+
+# Example usage
+bucket_name = 'your-bucket-name'
+prefix = 'your/prefix/'  # Optional, if you want to list files in a specific folder
+
+# List files in the bucket
+files = list_files_in_bucket(bucket_name, prefix)
+print(f"Files in bucket '{bucket_name}': {files}")
+
+# Read the content of the first file (if it exists)
+if files:
+    file_key = files[0]
+    content = read_file_from_s3(bucket_name, file_key)
+    print(f"Content of '{file_key}':\n{content}")
+
+-----
+
 # List containing data from the snapshot
 data = [
     ['p.G6bS$ts', 'p.G6bS$ts | 8', '8 |', 'C.1994delG', 'p.G6bS$ts | 17'],