Merge pull request #3299 from quay/joseph.schorr/QUAY-1233/batch-logs-download

Batch download of logs

endpoints/web.py

@@ -12,7 +12,7 @@ from flask_login import current_user
 import features
 
 from app import (app, billing as stripe, build_logs, avatar, signer, log_archive, config_provider,
-                 get_app_url, instance_keys, user_analytics)
+                 get_app_url, instance_keys, user_analytics, storage)
 from auth import scopes
 from auth.auth_context import get_authenticated_user
 from auth.basic import has_basic_auth
@@ -363,6 +363,33 @@ def buildlogs(build_uuid):
   return response
 
 
+@web.route('/exportedlogs/<file_id>', methods=['GET'])
+def exportedlogs(file_id):
+  # Only enable this endpoint if local storage is available.
+  has_local_storage = False
+  for storage_type, _ in app.config.get('DISTRIBUTED_STORAGE_CONFIG', {}).values():
+    if storage_type == 'LocalStorage':
+      has_local_storage = True
+      break
+
+  if not has_local_storage:
+    abort(404)
+
+  JSON_MIMETYPE = 'application/json'
+  exported_logs_storage_path = app.config.get('EXPORT_ACTION_LOGS_STORAGE_PATH',
+                                              'exportedactionlogs')
+  export_storage_path = os.path.join(exported_logs_storage_path, file_id)
+  if not storage.exists(storage.preferred_locations, export_storage_path):
+    abort(404)
+
+  try:
+    return send_file(storage.stream_read_file(storage.preferred_locations, export_storage_path),
+                     mimetype=JSON_MIMETYPE)
+  except IOError:
+    logger.exception('Could not read exported logs')
+    abort(403)
+
+
 @web.route('/logarchive/<file_id>', methods=['GET'])
 @route_show_if(features.BUILD_SUPPORT)
 @process_auth_or_cookie
@@ -392,6 +419,7 @@ def logarchive(file_id):
     logger.exception('Could not read archived logs')
     abort(403)
 
+
 @web.route('/receipt', methods=['GET'])
 @route_show_if(features.BILLING)
 @require_session_login