Small fix to truncation

jsmnbom · jsmnbom · commit cc436d8ed404 · 2018-10-24T21:21:00.000+02:00
diff --git a/bot/utils.py b/bot/utils.py
@@ -172,7 +172,7 @@ def __iter__(self):
                         'type': 'EndTag',
                         'name': tag
                     }
-                    yield from iter(self.truncated_message)
+                yield from iter(self.truncated_message)
                 break
             if token['type'] in ('Characters', 'SpaceCharacters'):
                 if (current_length + len(token['data'])) > self.max_length:
@@ -185,8 +185,7 @@ def __iter__(self):
                             'type': 'EndTag',
                             'name': tag
                         }
-                    for token2 in iter(self.truncated_message):
-                        yield token2
+                    yield from iter(self.truncated_message)
                     break
                 else:
                     current_length += len(token['data'])
@@ -200,8 +199,7 @@ def __iter__(self):
 
             yield token
 
-        for token in iter(self.suffix):
-            yield token
+        yield from iter(self.suffix)
 
 
 def truncate(html, truncated_message, suffix):
