X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=test%2Fhtml2txt.py;h=6b47f77b4d3c80d94645e5ef6bae6feecf4a6718;hb=HEAD;hp=4fa49f8cb931067cee9c43fe4f43c949d26c1bed;hpb=059486f5e3d34dd43c49e8ec16cb2429e76e39af;p=mimedecode.git

diff --git a/test/html2txt.py b/test/html2txt.py
index 4fa49f8..6b47f77 100755
--- a/test/html2txt.py
+++ b/test/html2txt.py
@@ -4,11 +4,21 @@ from __future__ import print_function
 import sys
 from m_lib.net.www.html import HTMLFilter
 
-with open(sys.argv[1], 'r') as f:
-    html = f.read()
+PY2 = sys.version_info[0] < 3
+if PY2:
+    with open(sys.argv[1], 'r') as f:
+        html = f.read()
+else:
+    with open(sys.argv[1], 'r', encoding='utf-8') as f:
+        html = f.read()
 
 filter = HTMLFilter()
 filter.feed(html)
 filter.close()
 
-print(filter.accumulator)
+if PY2:
+    print(filter.accumulator)
+else:
+    if not isinstance(filter.accumulator, bytes):
+        filter.accumulator = filter.accumulator.encode('utf-8')
+    sys.stdout.buffer.write(filter.accumulator + b'\n')