Update _remove_prefix method to iterativly remove prefix by regex

aws-powertools · Aug 10, 2023 · 766f525 · 766f525
1 parent 87c3456
commit 766f525
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/aws_lambda_powertools/event_handler/api_gateway.py b/aws_lambda_powertools/event_handler/api_gateway.py
@@ -715,8 +715,10 @@ def _remove_prefix(self, path: str) -> str:
         for prefix in self._strip_prefixes:
             if path == prefix:
                 return "/"
-            if self._path_starts_with(path, prefix):
-                return path[len(prefix) :]
+            path = re.sub(rf"^/?({prefix})/", r"/", path)
+
+        if not path:
+            path = "/"
 
         return path