Don't see all pre tags as suggestions

jsmnbom · jsmnbom · commit 227723e727a8 · 2018-11-08T23:22:39.000+01:00
diff --git a/bot/utils.py b/bot/utils.py
@@ -96,13 +96,13 @@ def __iter__(self):
         in_quote = False
         in_tag = 0
         for token in super().__iter__():
-            if token['type'] == 'StartTag' and token['name'] == 'pre':
-                if not in_tag:
-                    token['data'] = {}
-                    yield {
-                        'data': 'Suggestion:\n',
-                        'type': 'Characters'
-                    }
+            if (not in_tag and token['type'] == 'StartTag' and token['name'] == 'pre'
+                    and token['data'] and 'suggestion' in token['data'][(None, 'lang')]):
+                token['data'] = {}
+                yield {
+                    'data': 'Suggestion:\n',
+                    'type': 'Characters'
+                }
 
             if token['type'] == 'StartTag' and token['name'] == 'li':
                 if not (token['data'] and token['data'].get('class') != 'task-list-item'):
