zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-06-10 14:34:43 +02:00
Code Issues Projects Releases Packages Wiki Activity

Link bucket to generator

Browse source
This commit is contained in:
zandercymatics 2023-11-29 08:36:12 -07:00
parent f5974e00dc
commit e78a506d2b
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
3 changed files with 26 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/registrar/utility/s3_bucket.py
View file

@ -25,15 +25,10 @@ class S3ClientHelper:
except Exception as exc:
raise S3ClientError("Could not access the S3 client.") from exc
#self.bucket_name =
print("here:")
bucket = self.list_objects()
print(bucket)
def get_bucket_name(self):
"""Gets the name of our S3 Bucket"""
return settings.AWS_S3_BUCKET_NAME
def list_objects(self):
"""Returns a list of the top 1000 objects within our S3 instance"""
try:
@ -49,14 +44,16 @@ class S3ClientHelper:
except Exception as exc:
raise S3ClientError("Couldn't upload file") from exc
return response
def get_file(self, file_name, decode_to_utf=False):
"""Gets a file to our S3 instance and returns the file content"""
try:
response = self.boto_client.get_object(Bucket=self.get_bucket_name(), Key=file_name)
except Exception as exc:
raise S3ClientError("Couldn't get file") from exc
file_content = response["Body"].read()
if decode_to_utf:
return file_content.decode("utf-8")
return file_content
else:
return file_content