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

diff --git a/test/html2txt.py b/test/html2txt.py
index a15c538..6b47f77 100755
--- a/test/html2txt.py
+++ b/test/html2txt.py
@@ -1,13 +1,24 @@
 #! /usr/bin/env python
+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')