Tests(tox): Add forgotten Python 3.13

[sqlconvert.git] / sqlconvert / print_tokens.py

diff --git a/sqlconvert/print_tokens.py b/sqlconvert/print_tokens.py
index 3e2b0d5cab08be97bba84a777fe52ceddb422fca..65441b9fac42cbf9257f48d91cca63ef98b623c4 100644 (file)
--- a/sqlconvert/print_tokens.py
+++ b/sqlconvert/print_tokens.py
@@ -4,13 +4,20 @@ import sys
 
 def print_tokens(token_list, outfile=sys.stdout, encoding=None):
     if encoding:
-        outfile = getattr(outfile, 'buffer', outfile)
+        buffer = getattr(outfile, 'buffer', outfile)
+    else:
+        buffer = outfile
     for token in token_list.flatten():
         normalized = token.normalized
         if encoding:
             normalized = normalized.encode(encoding)
-        outfile.write(normalized)
+        buffer.write(normalized)
+    if buffer is not outfile:
+        buffer.flush()
+    outfile.flush()
 
 
 def tlist2str(token_list):
-    return u''.join(token.normalized for token in token_list.flatten())
+    return u''.join(
+        token.normalized for token in token_list.flatten()
+    ).replace('  ', ' ')