Convert Flask path variables to OpenAPI path parameters

bjmc · bjmc · commit 3ef0c6adcfa7 · 2019-06-12T12:16:55.000+01:00
diff --git a/openapi_core/wrappers/flask.py b/openapi_core/wrappers/flask.py
@@ -1,9 +1,16 @@
 """OpenAPI core wrappers module"""
+import re
+
 from openapi_core.wrappers.base import BaseOpenAPIRequest, BaseOpenAPIResponse
 
+# http://flask.pocoo.org/docs/1.0/quickstart/#variable-rules
+PATH_PARAMETER_PATTERN = r'<(?:(?:string|int|float|path|uuid):)?(\w+)>'
+
 
 class FlaskOpenAPIRequest(BaseOpenAPIRequest):
 
+    path_regex = re.compile(PATH_PARAMETER_PATTERN)
+
     def __init__(self, request):
         self.request = request
 
@@ -24,7 +31,7 @@ def path_pattern(self):
         if self.request.url_rule is None:
             return self.path
 
-        return self.request.url_rule.rule
+        return self.path_regex.sub(r'{\1}', self.request.url_rule.rule)
 
     @property
     def parameters(self):
