Fix for key collision

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