add max try in wait for reply method

noahzark · May 9, 2018 · 576994d · 576994d
1 parent b430f69
commit 576994d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/model/chat_robot.py b/model/chat_robot.py
@@ -147,9 +147,13 @@ def wait_for_reply(self):
     def wait_for_this_reply(self, content):
         result = self.wait_for_reply()
 
+        retry = 0
+
         content = to_str(content)
-        while to_str(result).find(content) == -1:
+        while to_str(result).find(content) == -1\
+                and retry < self.MAX_RETRY_TIMES:
             result = self.wait_for_reply()
+            retry += 1
 
         return result