add score in OPTIONS

av1m · Sep 11, 2022 · 166e573 · 166e573
1 parent eae0c3c
commit 166e573
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/flask_sustainable/__init__.py b/flask_sustainable/__init__.py
@@ -4,5 +4,5 @@
 
 from .extension import Sustainable
 
-__version__ = "0.0.2"
+__version__ = "0.0.3"
 __all__ = ["Sustainable"]
diff --git a/flask_sustainable/extension.py b/flask_sustainable/extension.py
@@ -15,7 +15,7 @@
 
 import flask
 
-from flask_sustainable.base import BaseIndicator, BaseScore
+from flask_sustainable.base import BaseHeader, BaseIndicator, BaseScore
 from flask_sustainable.compress import Compression
 
 logger = logging.getLogger(__name__)
@@ -90,14 +90,19 @@ def after_request(self, response: flask.Response) -> flask.Response:
         except TypeError as error:
             logger.warning("Error while compressing the response")
             logger.exception(error)
+        # Retrieve all registered headers
+        registered: list[BaseHeader] = [
+            *self._registered_indicators,
+            *self._registered_scores,
+        ]
         # Add allowed headers
         if flask.request.method == "OPTIONS":
-            headers = [x.name for x in self._registered_indicators]
+            headers = [x.name for x in registered]
             response.headers.extend(
                 {"Access-Control-Allow-Headers": ", ".join(headers)}
             )
         # Run after_request on all registered headers
-        for header in [*self._registered_indicators, *self._registered_scores]:
+        for header in registered:
             if header.should_use():
                 header.after_request(response=response)
         return response