Introduce max_iterations for _get_all_iterative_for_uris()

openannotation · Feb 4, 2015 · d53fcfa · d53fcfa
1 parent 389dc0d
commit d53fcfa
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/annotator/document.py b/annotator/document.py
@@ -24,6 +24,7 @@
         }
     }
 }
+MAX_ITERATIONS = 5
 
 
 class Document(es.Model):
@@ -93,8 +94,9 @@ def _get_all_iterative_for_uris(cls, uris):
         documents = {}
         all_uris = set(uris)
         new_uris = list(uris)
+        iterations = 0
 
-        while len(new_uris):
+        while len(new_uris) and iterations < MAX_ITERATIONS:
             docs = cls._get_all_by_uris(new_uris)
             new_uris = []
             for doc in docs:
@@ -105,6 +107,7 @@ def _get_all_iterative_for_uris(cls, uris):
                         if uri not in all_uris:
                             new_uris.append(uri)
                             all_uris.add(uri)
+            iterations += 1
 
         return list(documents.values())