Fix header processing

materialsproject · Oct 13, 2023 · da8371e · da8371e
1 parent 879284b
commit da8371e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/src/maggma/api/resource/read_resource.py b/src/maggma/api/resource/read_resource.py
@@ -173,7 +173,10 @@ def get_by_key(
                 response = Response(orjson.dumps(response, default=serialization_helper))  # type: ignore
 
             if self.header_processor is not None:
-                self.header_processor.process_header(response, request)
+                if self.disable_validation:
+                    self.header_processor.process_header(response, request)
+                else:
+                    self.header_processor.process_header(temp_response, request)
 
             return response
 
@@ -269,7 +272,10 @@ def search(**queries: Dict[str, STORE_PARAMS]) -> Union[Dict, Response]:
                 response = Response(orjson.dumps(response, default=serialization_helper))  # type: ignore
 
             if self.header_processor is not None:
-                self.header_processor.process_header(response, request)
+                if self.disable_validation:
+                    self.header_processor.process_header(response, request)
+                else:
+                    self.header_processor.process_header(temp_response, request)
 
             return response
 

diff --git a/src/maggma/api/resource/s3_url.py b/src/maggma/api/resource/s3_url.py
@@ -123,7 +123,10 @@ def get_by_key(
                 response = Response(orjson.dumps(response, default=serialization_helper))  # type: ignore
 
             if self.header_processor is not None:
-                self.header_processor.process_header(response, request)
+                if self.disable_validation:
+                    self.header_processor.process_header(response, request)
+                else:
+                    self.header_processor.process_header(temp_response, request)
 
             return response