fixup! mozillaGH-72 Add FormattedPolicyCSPMiddleware

DylanYoung · May 26, 2022 · a4e83ed · a4e83ed
1 parent 3bce755
commit a4e83ed
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/csp/contrib/formatted_policy.py b/csp/contrib/formatted_policy.py
@@ -24,7 +24,10 @@ def build_policy(self, request, response):
             request, response,
         ):
             format_kwargs = {
-                field_name for _, field_name, _, _ in formatter.parse(csp)
+                # field_name ::= arg_name ("." attribute_name | "[" element_index "]")*
+                field_name[: next((i for i, c in enumerate(field_name) if c in ".["), None)]
+                for _, field_name, _, _ in formatter.parse(csp)
+                if field_name is not None
             }
             if format_kwargs:
                 for name in format_kwargs: