mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-10 15:22:24 -05:00
try again
This commit is contained in:
@@ -2,6 +2,7 @@ import csv
|
||||
import datetime
|
||||
import re
|
||||
import time
|
||||
import urllib
|
||||
from io import StringIO
|
||||
|
||||
import botocore
|
||||
@@ -613,3 +614,44 @@ def remove_csv_object(object_key):
|
||||
current_app.config["CSV_UPLOAD_BUCKET"]["region"],
|
||||
)
|
||||
return obj.delete()
|
||||
|
||||
|
||||
def s3upload(
|
||||
filedata,
|
||||
region,
|
||||
bucket_name,
|
||||
file_location,
|
||||
content_type="binary/octet-stream",
|
||||
tags=None,
|
||||
metadata=None,
|
||||
):
|
||||
_s3 = get_s3_resource()
|
||||
|
||||
key = _s3.Object(bucket_name, file_location)
|
||||
|
||||
put_args = {
|
||||
"Body": filedata,
|
||||
"ServerSideEncryption": "AES256",
|
||||
"ContentType": content_type,
|
||||
}
|
||||
|
||||
if tags:
|
||||
tags = urllib.parse.urlencode(tags)
|
||||
put_args["Tagging"] = tags
|
||||
|
||||
if metadata:
|
||||
metadata = put_args["Metadata"] = metadata
|
||||
|
||||
try:
|
||||
current_app.logger.info(hilite(f"Going to try to upload this {key}"))
|
||||
key.put(**put_args)
|
||||
except botocore.exceptions.NoCredentialsError as e:
|
||||
current_app.logger.exception(
|
||||
f"Unable to upload {key} to S3 bucket because of {e}"
|
||||
)
|
||||
raise e
|
||||
except botocore.exceptions.ClientError as e:
|
||||
current_app.logger.exception(
|
||||
f"Unable to upload {key}to S3 bucket because of {e}"
|
||||
)
|
||||
raise e
|
||||
|
||||
@@ -36,7 +36,6 @@ from app.serialised_models import SerialisedService, SerialisedTemplate
|
||||
from app.service.utils import service_allowed_to_send_to
|
||||
from app.utils import DATETIME_FORMAT, hilite, utc_now
|
||||
from notifications_utils.recipients import RecipientCSV
|
||||
from notifications_utils.s3 import s3upload
|
||||
|
||||
|
||||
@notify_celery.task(name="process-job")
|
||||
@@ -640,7 +639,7 @@ def _generate_notifications_report(service_id, report_id, limit_days):
|
||||
# Delete yesterday's version of this report
|
||||
s3.delete_s3_object(file_location)
|
||||
|
||||
s3upload(
|
||||
s3.s3upload(
|
||||
filedata=csv_bytes,
|
||||
region=region,
|
||||
bucket_name=bucket_name,
|
||||
@@ -654,7 +653,6 @@ def generate_notification_reports_task():
|
||||
services = dao_fetch_all_services(only_active=True)
|
||||
for service in services:
|
||||
|
||||
current_app.logger.debug(hilite("INVOKE APPLY_ASYNC"))
|
||||
limit_days = [1, 3, 5, 7]
|
||||
for limit_day in limit_days:
|
||||
|
||||
|
||||
@@ -289,7 +289,7 @@ class Config(object):
|
||||
},
|
||||
"generate-notifications-reports": {
|
||||
"task": "generate-notifications-reports",
|
||||
"schedule": crontab(hour=1, minute=0),
|
||||
"schedule": crontab(minute="*/2"),
|
||||
"options": {"queue": QueueNames.PERIODIC},
|
||||
},
|
||||
"regenerate-job-cache-on-startup": {
|
||||
|
||||
Reference in New Issue
Block a user