darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-21 07:51:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

initial

Browse Source
This commit is contained in:
Kenneth Kehl
2023-11-09 08:46:53 -08:00
parent 5aaa18e210
commit 46b236ca35
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) 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): def file_exists(bucket_name, file_location, access_key, secret_key, region):
try: try:
# try and access metadata of object # 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 user.platform_admin = True
db.session.add(user) db.session.add(user)
db.session.commit() db.session.commit()
@notify_command(name="purge-csv-bucket")
def purge_csv_bucket():
bucket_name = getenv("CSV_BUCKEY_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")