Improving the algorithm

gil9red · gil9red · commit 1228372e8c85 · 2020-09-26T03:17:07.000+05:00
diff --git a/EscapePyPromptLineString/main.py b/EscapePyPromptLineString/main.py
@@ -104,7 +104,7 @@ def input_text_changed(self):
             new_out_text = []
 
             for line in out_text.splitlines():
-                if line.startswith('>>> ') or line.startswith('... '):
+                if line.startswith('>>> ') or line.startswith('... ') or line == '...':
                     line = line[4:]
 
                 new_out_text.append(line)
