Make syntax highlighting for XML lazy, to only call pygments when debug logs are enabled.

Makes poezio about 11% faster when sending/receiving messages.

1 files changed, 8 insertions, 1 deletions

diff --git a/slixmpp/xmlstream/tostring.py b/slixmpp/xmlstream/tostring.py
index 521ee460..041346cb 100644
--- a/slixmpp/xmlstream/tostring.py
+++ b/slixmpp/xmlstream/tostring.py
@@ -169,7 +169,14 @@ def _get_highlight():
         LEXER = get_lexer_by_name('xml')
         FORMATTER = Terminal256Formatter()
 
-        return lambda x: highlight(x, LEXER, FORMATTER)
+        class Highlighter:
+            __slots__ = ['string']
+            def __init__(self, string):
+                self.string = string
+            def __str__(self):
+                return highlight(str(self.string).strip(), LEXER, FORMATTER)
+
+        return Highlighter
     except ImportError:
         return lambda x: x