Add METS download link to AIP page. (#46)

Adds a METS download link to the AIP detail page.

AIPscan/Reporter/templates/aip.html

@@ -22,6 +22,8 @@
   Originals: {{ original_file_count }}
   <br>
   Preservation copies: {{ preservation_file_count }}
+  <br>
+  <a href="{{ url_for('reporter.download_mets', aip_id=aip.id) }}">Download METS</a>
 </div>
 
 <h3>Original files</h3>

AIPscan/Reporter/views.py

@@ -7,7 +7,8 @@
 
 from datetime import datetime
 
-from flask import jsonify, make_response, render_template, request, session
+from flask import jsonify, make_response, render_template, request, session, Response
+import requests
 
 from AIPscan.models import (
     AIP,
@@ -38,6 +39,7 @@
     sort_puids,
 )
 from AIPscan.Reporter.helpers import get_premis_xml_lines
+from AIPscan.Aggregator.task_helpers import get_mets_url
 
 
 def _get_storage_service(storage_service_id):
@@ -272,3 +274,25 @@ def update_dates():
         req = request.get_json()
         session["end_date"] = request.json.get("end_date")
         return make_response(jsonify(req), 200)
+
+
+@reporter.route("/download_mets/<aip_id>", methods=["GET"])
+def download_mets(aip_id):
+    aip = AIP.query.get(aip_id)
+    storage_service = StorageService.query.get(aip.storage_service_id)
+
+    api_url = {
+        "baseUrl": storage_service.url,
+        "userName": storage_service.user_name,
+        "apiKey": storage_service.api_key,
+    }
+
+    mets_response = requests.get(
+        get_mets_url(api_url, aip.uuid, f"{aip.transfer_name}-{aip.uuid}/data/METS.{aip.uuid}.xml")
+    )
+
+    headers = {
+        "Content-Disposition": f'attachment; filename="METS-{aip.uuid}.xml"',
+        "Content-length": len(mets_response.content)
+    }
+    return Response(mets_response.content, mets_response.status_code, headers)