Merge pull request #3 from tyarkoni/master

replace variable parentheses in flask routes
getsling · Mar 9, 2015 · b79402f · b79402f
2 parents 6e48449 + 8310ee3
commit b79402f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/flask_swagger.py b/flask_swagger.py
@@ -8,6 +8,7 @@
 """
 import inspect
 import yaml
+import re
 
 
 def _sanitize(comment):
@@ -111,5 +112,8 @@ def spec():
                     parameters=params
                 )
         if len(operations):
-            paths[str(rule)] = operations
+            rule = str(rule)
+            for arg in re.findall('(<(.*?\:)?(.*?)>)', rule):
+                rule = rule.replace(arg[0], '{%s}' % arg[2])
+            paths[rule] = operations
     return output