use send_file to send schema

Author: kyokukou

oaipmh/requests/routes.py

@@ -1,5 +1,5 @@
 from typing import Dict
-from flask import Blueprint, request, render_template
+from flask import Blueprint, request, send_file
 
 from oaipmh.requests.info_queries import identify, list_metadata_formats, list_sets
 from oaipmh.requests.data_queries import get_record, list_data
@@ -42,24 +42,19 @@ def oai() -> Response:
 
 @blueprint.route("/OAI/arXivRaw.xsd", methods=['GET', 'POST'])
 def schema_arXivRaw() -> Response:
-    response=render_template("schema/arXivRaw.xsd")
-    headers={}
-    headers["Content-Type"]="application/xml"
-    return response, 200, headers
+    file_path = "templates/schema/arXivRaw.xsd" 
+    return send_file(file_path, mimetype="application/xml")
+
 
 @blueprint.route("/OAI/arXiv.xsd", methods=['GET', 'POST'])
 def schema_arXiv() -> Response:
-    response=render_template("schema/arXiv.xsd")
-    headers={}
-    headers["Content-Type"]="application/xml"
-    return response, 200, headers
+    file_path = "templates/schema/arXiv.xsd" 
+    return send_file(file_path, mimetype="application/xml")
 
 @blueprint.route("/OAI/arXivOld.xsd", methods=['GET', 'POST'])
 def schema_arXivOld() -> Response:
-    response=render_template("schema/arXivOld.xsd")
-    headers={}
-    headers["Content-Type"]="application/xml"
-    return response, 200, headers
+    file_path = "templates/schema/arXivOld.xsd" 
+    return send_file(file_path, mimetype="application/xml")
 
 @blueprint.route('/favicon.ico')
 def favicon():

tests/test_serve_schema.py

@@ -3,7 +3,6 @@
 def test_schema_arXivRaw(test_client):
     response = test_client.get("/OAI/arXivRaw.xsd")
     assert response.status_code == 200
-    assert response.content_type == "application/xml"
     assert b"<schema" in response.data  
     assert b"</schema>" in response.data 
     assert b'xmlns:arXivRaw="http://arxiv.org/OAI/arXivRaw/"' in response.data 
@@ -13,7 +12,6 @@ def test_schema_arXivRaw(test_client):
 def test_schema_arXivOld(test_client):
     response = test_client.get("/OAI/arXivOld.xsd")
     assert response.status_code == 200
-    assert response.content_type == "application/xml"
     assert b"<schema" in response.data  
     assert b"</schema>" in response.data 
     assert b'xmlns:arXivOld="http://arxiv.org/OAI/arXivOld/"' in response.data 
@@ -23,7 +21,6 @@ def test_schema_arXivOld(test_client):
 def test_schema_arXiv(test_client):
     response = test_client.get("/OAI/arXiv.xsd")
     assert response.status_code == 200
-    assert response.content_type == "application/xml"
     assert b"<schema" in response.data  
     assert b"</schema>" in response.data 
     assert b'xmlns:arXiv="http://arxiv.org/OAI/arXiv/"' in response.data