Fix for key collision

encode · Oct 9, 2017 · 30553f7 · 30553f7
1 parent f1f657d
commit 30553f7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/rest_framework/schemas/generators.py b/rest_framework/schemas/generators.py
@@ -70,11 +70,17 @@ def __init__(self):
         self.methods_counter = Counter()
         super(LinkNode, self).__init__()
 
-    def get_next_key(self, method):
+    def _generate_next_key(self, method):
         current_val = self.methods_counter[method]
         self.methods_counter[method] += 1
         return '{}_{}'.format(method, current_val)
 
+    def get_next_key(self, method):
+        while True:
+            key = self._generate_next_key(method)
+            if key not in self:
+                return key
+
 
 def insert_into(target, keys, value):
     """