StringIO comes in different modules when in Python 2.x or 3.x

fe346e37 · Javier D. Garcia-Lasheras · 44783c06 · fe346e37
Commit fe346e37 authored Aug 16, 2016 by Javier D. Garcia-Lasheras
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

configparser.py hdlmake/manifest_parser/configparser.py +5 -2

No files found.
--- a/hdlmake/manifest_parser/configparser.py
+++ b/hdlmake/manifest_parser/configparser.py
@@ -26,7 +26,10 @@ from __future__ import absolute_import
 import logging
 import os
 import sys
-import StringIO
+if not sys.version[0] is "2":
+    from io import StringIO
+else:
+    from StringIO import StringIO
 import contextlib


@@ -36,7 +39,7 @@ def stdout_io(stdout=None):
    the executed Manifest.py files"""
    old = sys.stdout
    if stdout is None:
-        stdout = StringIO.StringIO()
+        stdout = StringIO()
    sys.stdout = stdout
    yield stdout
    sys.stdout = old