mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-14 01:02:09 -05:00
try again
This commit is contained in:
@@ -1,4 +1,3 @@
|
|||||||
import os
|
|
||||||
import urllib
|
import urllib
|
||||||
|
|
||||||
import botocore
|
import botocore
|
||||||
@@ -6,6 +5,9 @@ from boto3 import Session
|
|||||||
from botocore.config import Config
|
from botocore.config import Config
|
||||||
from flask import current_app
|
from flask import current_app
|
||||||
|
|
||||||
|
from app.config import _s3_credentials_from_env
|
||||||
|
from app.utils import hilite
|
||||||
|
|
||||||
AWS_CLIENT_CONFIG = Config(
|
AWS_CLIENT_CONFIG = Config(
|
||||||
# This config is required to enable S3 to connect to FIPS-enabled
|
# This config is required to enable S3 to connect to FIPS-enabled
|
||||||
# endpoints. See https://aws.amazon.com/compliance/fips/ for more
|
# endpoints. See https://aws.amazon.com/compliance/fips/ for more
|
||||||
@@ -16,18 +18,17 @@ AWS_CLIENT_CONFIG = Config(
|
|||||||
max_pool_connections=50,
|
max_pool_connections=50,
|
||||||
use_fips_endpoint=True,
|
use_fips_endpoint=True,
|
||||||
)
|
)
|
||||||
|
default_regions = "us-gov-west-1"
|
||||||
|
|
||||||
default_access_key_id = os.environ.get("CSV_AWS_ACCESS_KEY_ID")
|
|
||||||
default_secret_access_key = os.environ.get("CSV_AWS_SECRET_ACCESS_KEY")
|
|
||||||
default_region = os.environ.get("CSV_AWS_REGION")
|
|
||||||
|
|
||||||
|
|
||||||
def get_s3_resource():
|
def get_s3_resource():
|
||||||
|
|
||||||
|
credentials = _s3_credentials_from_env("CSV")
|
||||||
|
current_app.logger.info(hilite(f"CREDENTIALS {credentials}"))
|
||||||
session = Session(
|
session = Session(
|
||||||
aws_access_key_id=os.environ.get("CSV_AWS_ACCESS_KEY_ID"),
|
aws_access_key_id=credentials["access_key_id"],
|
||||||
aws_secret_access_key=os.environ.get("CSV_AWS_SECRET_ACCESS_KEY"),
|
aws_secret_access_key=credentials["secret_access_key"],
|
||||||
region_name=os.environ.get("CSV_AWS_REGION"),
|
region_name=credentials["region"],
|
||||||
)
|
)
|
||||||
noti_s3_resource = session.resource("s3", config=AWS_CLIENT_CONFIG)
|
noti_s3_resource = session.resource("s3", config=AWS_CLIENT_CONFIG)
|
||||||
return noti_s3_resource
|
return noti_s3_resource
|
||||||
@@ -41,8 +42,6 @@ def s3upload(
|
|||||||
content_type="binary/octet-stream",
|
content_type="binary/octet-stream",
|
||||||
tags=None,
|
tags=None,
|
||||||
metadata=None,
|
metadata=None,
|
||||||
access_key=default_access_key_id,
|
|
||||||
secret_key=default_secret_access_key,
|
|
||||||
):
|
):
|
||||||
_s3 = get_s3_resource()
|
_s3 = get_s3_resource()
|
||||||
|
|
||||||
@@ -83,9 +82,6 @@ class S3ObjectNotFound(botocore.exceptions.ClientError):
|
|||||||
def s3download(
|
def s3download(
|
||||||
bucket_name,
|
bucket_name,
|
||||||
filename,
|
filename,
|
||||||
region=default_region,
|
|
||||||
access_key=default_access_key_id,
|
|
||||||
secret_key=default_secret_access_key,
|
|
||||||
):
|
):
|
||||||
try:
|
try:
|
||||||
s3 = get_s3_resource()
|
s3 = get_s3_resource()
|
||||||
|
|||||||
Reference in New Issue
Block a user