AWS S3 Management with Python and Boto3: A Class-based Approach

# Import the MenuGerenciadorS3 class from the correct module
from classes.MenuGerenciadorS3 import MenuGerenciadorS3

# Replace with the name of your bucket
bucket_name = "aulaclasses"

# Instantiate the MenuGerenciadorS3 class with the bucket name as an argument
menu = MenuGerenciadorS3(bucket_name)

# Execute the 'executar' method to start the menu
menu.executar()

This code snippet demonstrates the use of a Python class, “MenuGerenciadorS3”, to interact with an Amazon S3 bucket using the Boto3 library. The class is instantiated with the name of an S3 bucket, and the “executar” method is then called to execute the menu-driven operations for managing the specified S3 bucket. The code aims to provide a modular and structured approach to S3 management using Python and Boto3.

The code below should be inside the MenuManager.py class file. This code will create the menu and call the class to interact with the menu.

from classes.GerenciarS3 import GerenciarS3

class MenuGerenciadorS3:
    def __init__(self, bucket_name):
        # Initialize the menu with the provided bucket name
        self.manager = GerenciarS3(bucket_name)

    def display_menu(self):
        # Display the menu options
        print("--- MENU ---")
        print("1. List files")
        print("2. Upload file")
        print("3. Download file")
        print("4. Delete file")
        print("0. Exit")

    def list_files(self):
        # Call the manager's method to list files in the bucket
        self.manager.list_files()

    def upload_file(self):
        # Prompt user for file path and optional file name
        file_path = input("Enter the file path: ")
        file_name = input("Enter the file name (optional): ")
        # Call the manager's method to upload the file
        self.manager.upload_file(file_path, file_name)

    def download_file(self):
        # Prompt user for file name and save path
        file_name = input("Enter the file name: ")
        save_path = input("Enter the path to save the file: ")
        # Call the manager's method to download the file
        self.manager.download_file(file_name, save_path)

    def delete_file(self):
        # Prompt user for the name of the file to delete
        file_name = input("Enter the name of the file to delete: ")
        # Call the manager's method to delete the file
        self.manager.delete_file(file_name)

    def execute(self):
        # Main execution loop
        while True:
            # Display the menu
            self.display_menu()
            # Prompt user for option
            option = input("Enter the desired option: ")
            # Process the user's choice
            if option == "1":
                self.list_files()
            elif option == "2":
                self.upload_file()
            elif option == "3":
                self.download_file()
            elif option == "4":
                self.delete_file()
            elif option == "0":
                # Exit the program if the user chooses
                print("Exiting the program.")
                break
            else:
                # Handle invalid input
                print("Invalid option. Please enter again.")

Now in the file ManageS3.py put the code below. This code will interact with S3. Don’t forget to create your bucket first.

import os
import boto3

class GerenciarS3:
    def __init__(self, bucket_name):
        # Initialize the S3 manager with the provided bucket name
        self.bucket_name = bucket_name
        self.s3 = boto3.client('s3')

    def list_files(self):
        # List files in the specified S3 bucket
        try:
            response = self.s3.list_objects_v2(Bucket=self.bucket_name)

            if 'Contents' in response:
                # Extract and print file names if files exist
                files = [obj['Key'] for obj in response['Contents']]
                print("Files in S3:")
                for file in files:
                    print(file)
            else:
                print("No files found in S3.")
        except Exception as e:
            print(f"Error listing files in S3: {e}")

    def upload_file(self, file_path, file_name=None):
        # Upload a file to the specified S3 bucket
        if file_name is None:
            file_name = os.path.basename(file_path)

        try:
            self.s3.upload_file(file_path, self.bucket_name, file_name)
            print(f"File {file_name} successfully uploaded to S3.")
        except Exception as e:
            print(f"Error uploading file to S3: {e}")

    def download_file(self, file_name, save_path):
        # Download a file from the specified S3 bucket
        try:
            full_path = os.path.join(save_path, file_name)
            self.s3.download_file(self.bucket_name, file_name, full_path)
            print(f"File {file_name} successfully downloaded from S3.")
        except Exception as e:
            print(f"Error downloading file from S3: {e}")

    def delete_file(self, file_name):
        # Delete a file from the specified S3 bucket
        try:
            self.s3.delete_object(Bucket=self.bucket_name, Key=file_name)
            print(f"File {file_name} successfully deleted from S3.")
        except Exception as e:
            print(f"Error deleting file from S3: {e}")

To install the boto3 library, type the following command into your terminal.

pip install boto3

Boto3 is the official Amazon Web Services (AWS) Software Development Kit (SDK) for Python. It provides a simple and Pythonic interface to interact with various AWS services, making it easier for developers to integrate their Python applications with AWS cloud resources.