AWS Automation with Python and Boto3

boto3
discord
python-dotenv

#In-Built Libraries
import os
import logging

#Third Party Libraries
import discord
import boto3
from botocore.exceptions import ClientError
from dotenv import load_dotenv

load_dotenv()

TOKEN = os.getenv('DISCORD_TOKEN')
intents = discord.Intents.all()
client = discord.Client(intents=discord.Intents.default())
region= 'ap-south-1'

@client.event
async def on_ready():
    print('We have logged in as {0.user}'.format(client))

def parse_command(message):
    # create s3 --name=test
	#Stripping input to remove whitespaces
    message = message.strip()
	#Splitting the input, placing it into an array
    message = message[1:]
    message = message.split(' ')
    cmd = message[0]
    resource = message[1]
    args = message[2::]
	#Finally the input is provided out as a JSON
    return {
        "cmd": cmd,
        "resource": resource,
        "args": args
    }

@client.event
async def on_message(message):
    if message.author == client.user:
        return
    command = parse_command(message.content)
    # print(command)
    if message.content.startswith('$'):
        if command['resource'] == "s3"  and command['cmd'] == "create":
        # Parse args
            bucket_name = command["args"][1]
            # print(bucket_name)
            aws_create_s3bucket(bucket_name)
    
        elif command['resource'] == "s3"  and command['cmd'] == "delete":
        # Parse args
            bucket_name = command["args"][1]
            aws_delete_s3bucket(bucket_name)

def aws_create_s3bucket(bucket_name):
    """
    If a region is not specified, the bucket is created in the S3 default
    region (us-east-1).
    :param bucket_name: Bucket to create
    :param region: String region to create bucket in, e.g., 'us-west-2'
    :return: True if bucket created, else False
    """
    # Create bucket
    try:
        if region is None:
            s3_client = boto3.client('s3')
            s3_client.create_bucket(Bucket=bucket_name)
        else:
            s3_client = boto3.client('s3', region_name=region)
            location = {'LocationConstraint': region}
            s3_client.create_bucket(Bucket=bucket_name,
                                    CreateBucketConfiguration=location)
            print(f"Bucket {bucket_name} has been created!")
    except Exception as e:
        logging.error(e)
        return False
    return True

def aws_delete_s3bucket(bucket_name):
    # delete bucket
    try:
        s3_client = boto3.client('s3', region_name=region)
        ##Delete bucket objects function is called during bucket deletion
        delete_all_bucket_objects([bucket_name])
        s3_client.delete_bucket(Bucket=bucket_name)
        print(f"Bucket {bucket_name} has been deleted")
    except Exception as e:
        logging.error(e)
        return False
    return True

def delete_all_bucket_objects(buckets=[]):
    s3 = boto3.resource('s3')
    print("reached delete all bucket func")
    for bucket in buckets:
        try:
            b = s3.Bucket(bucket)
            # sns_logger(message="Deleting " + bucket + "objects")
            b.objects.all().delete()
            print(f"Bucket's {bucket} objects have been deleted")
        except Exception as e:
            logging.error(e)
            print("Deleting Objects failed: ", e)
            return False

client.run(TOKEN)

Output:

2022-12-19 10:41:56 INFO     discord.client logging in using static token
2022-12-19 10:41:57 INFO     discord.gateway Shard ID None has connected to Gateway (Session ID: 7db5645c5cc94e565ece3248a71xyz).
We have logged in as blog_test_bot#12xyz

$create s3 --name blog-test-bucket-0001

Output:
Bucket blog-test-bucket-0001 has been created!

$delete s3 --name blog-test-bucket-0001

Output:
Bucket blog-test-bucket-0001 objects have been deleted
Bucket blog-test-bucket-0001 has been deleted