darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 01:41:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #586 from GSA/purge_command

Browse Source 
write a csv bucket purge command
This commit is contained in:
Carlo Costino
2023-11-09 17:00:52 -05:00
committed by GitHub
parent e2f27cc940 9dab7cc481
commit 48b18b9366
2 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
app/aws/s3.py
View File

@@ -22,6 +22,17 @@ def get_s3_object(bucket_name, file_location, access_key, secret_key, region):
return s3.Object(bucket_name, file_location)
def purge_bucket(bucket_name, access_key, secret_key, region):
session = Session(
aws_access_key_id=access_key,
aws_secret_access_key=secret_key,
region_name=region,
)
s3 = session.resource("s3", config=AWS_CLIENT_CONFIG)
bucket = s3.Bucket(bucket_name)
bucket.objects.all().delete()
def file_exists(bucket_name, file_location, access_key, secret_key, region):
try:
# try and access metadata of object

11
app/commands.py
View File

@@ -841,3 +841,14 @@ def promote_user_to_platform_admin(user_email_address):
user.platform_admin = True
db.session.add(user)
db.session.commit()
@notify_command(name="purge-csv-bucket")
def purge_csv_bucket():
bucket_name = getenv("CSV_BUCKET_NAME")
access_key = getenv("CSV_AWS_ACCESS_KEY_ID")
secret = getenv("CSV_AWS_SECRET_ACCESS_KEY")
region = getenv("CSV_AWS_REGION")
print("ABOUT TO RUN PURGE CSV BUCKET")
s3.purge_bucket(bucket_name, access_key, secret, region)
print("RAN PURGE CSV BUCKET")