aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJoseph Hunkeler <jhunkeler@gmail.com>2016-07-03 21:02:52 -0400
committerJoseph Hunkeler <jhunkeler@gmail.com>2016-07-03 21:02:52 -0400
commita5276b9056bedc1b78a6b809f68c9f77daf6ee3f (patch)
treeb94e1aeb5958c9269526ed0495340bcdb1847c9d
parent75d3c99573e2459579afed8c540beadec8011de7 (diff)
downloadsteuermann-a5276b9056bedc1b78a6b809f68c9f77daf6ee3f.tar.gz
Fix StringIO python 3k compat
-rw-r--r--steuermann/nodes.py6
-rw-r--r--steuermann/report.py6
2 files changed, 8 insertions, 4 deletions
diff --git a/steuermann/nodes.py b/steuermann/nodes.py
index aaf42d6..521db2d 100644
--- a/steuermann/nodes.py
+++ b/steuermann/nodes.py
@@ -237,7 +237,11 @@ class node(object) :
# debug - make a string representation of all the nodes
def show_nodes( node_index ) :
- import cStringIO as StringIO
+ try:
+ from io import StringIO
+ except ImportError:
+ from cStringIO import StringIO
+
s = StringIO.StringIO()
for x in sorted( [ x for x in node_index ] ) :
x = node_index[x]
diff --git a/steuermann/report.py b/steuermann/report.py
index afdf650..f27d84d 100644
--- a/steuermann/report.py
+++ b/steuermann/report.py
@@ -11,7 +11,7 @@ import pandokia.common
try:
from io import StringIO
except ImportError:
- import StringIO
+ from StringIO import StringIO
# maybe the output is html 3.2 - in any case, it is way simpler than
# more recent standards.
@@ -170,7 +170,7 @@ def report_text( db, run_name, info_callback = info_callback_status ) :
raw = raw_report( db, run_name, info_callback )
- s = StringIO.StringIO()
+ s = StringIO()
table_list = get_table_list(db, run_name)
@@ -196,7 +196,7 @@ def report_text( db, run_name, info_callback = info_callback_status ) :
def report_html( db, run_name, info_callback = info_callback_status, hlevel=1 ) :
raw = raw_report( db, run_name, info_callback )
- s = StringIO.StringIO()
+ s = StringIO()
s.write('<h%d>%s</h%d>\n'%(hlevel,run_name,hlevel))
hlevel = hlevel + 1