darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-09-02 17:48:50 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add logging for streaming contents

Browse Source
This commit is contained in:
Imdad Ahad
2017-03-13 14:53:19 +00:00
parent 4f2756ee90
commit 276a072a54
3 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
app/main/views/jobs.py
View File

@@ -13,7 +13,8 @@ from flask import (
url_for,
current_app,
redirect,
Response
Response,
stream_with_context
)
from flask_login import login_required
from werkzeug.datastructures import MultiDict
@@ -96,7 +97,6 @@ def view_jobs(service_id):
@user_has_permissions('view_activity', admin_override=True)
def view_job(service_id, job_id):
job = job_api_client.get_job(service_id, job_id)['data']
if job['job_status'] == 'cancelled':
abort(404)
@@ -144,19 +144,22 @@ def view_job_csv(service_id, job_id):
filter_args['status'] = _set_status_filters(filter_args)
return Response(
generate_notifications_csv(
service_id=service_id,
job_id=job_id,
status=filter_args.get('status'),
page=request.args.get('page', 1),
page_size=5000
stream_with_context(
generate_notifications_csv(
service_id=service_id,
job_id=job_id,
status=filter_args.get('status'),
page=request.args.get('page', 1),
page_size=5000
)
),
mimetype='text/csv',
headers={
'Content-Disposition': 'inline; filename="{} - {}.csv"'.format(
template['name'],
format_datetime_short(job['created_at'])
)}
)
}
)