improves url grouping

napster0125 · Mar 1, 2014 · 9c7daf8 · 9c7daf8
1 parent e45a97c
commit 9c7daf8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/rest_framework_swagger/urlparser.py b/rest_framework_swagger/urlparser.py
@@ -36,8 +36,9 @@ def get_apis(self, patterns=None, filter_path=None, exclude_namespaces=[]):
     def get_filtered_apis(self, apis, filter_path):
         filtered_list = []
 
+        match = "/{f}/".format(f=filter_path)
         for api in apis:
-            if filter_path in api['path'].strip('/'):
+            if api['path'].startswith(match):
                 filtered_list.append(api)
 
         return filtered_list
@@ -70,7 +71,7 @@ def __filter_top_level_apis__(self, root_paths):
         filtered_paths = set()
         base_path = self.__get_base_path__(root_paths)
         for path in root_paths:
-            resource = path.lstrip(base_path).split('/')[0]
+            resource = path.replace(base_path, '').split('/')[0]
             filtered_paths.add(base_path + resource)
 
         return list(filtered_paths)