From e74a3ddfa12d91b768f9686bca83246b66ca8e05 Mon Sep 17 00:00:00 2001 From: kyokukou Date: Thu, 23 May 2024 13:18:20 -0700 Subject: [PATCH] add surrogate key outputs a headers object --- arxiv/integration/fastly/headers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/arxiv/integration/fastly/headers.py b/arxiv/integration/fastly/headers.py index bc5539ac..36498922 100644 --- a/arxiv/integration/fastly/headers.py +++ b/arxiv/integration/fastly/headers.py @@ -1,6 +1,7 @@ from typing import Dict, List +from werkzeug.datastructures import Headers -def add_surrogate_key(headers: Dict[str,str], keys:List[str])-> Dict[str, str]: +def add_surrogate_key(headers: Dict[str,str], keys:List[str])-> Headers: """adds surrogate key(s) to a response header, will update the header dictionary with new keys while retaining the rest of the header information""" old_keys=f' {headers.get("Surrogate-Key","").strip()} '