minor tweak to python_support

2016-12-06 23:09:55 -05:00
parent 1b36ca00e5
commit 0fc44efbcc
1 changed files with 3 additions and 1 deletions
--- a/mongoengine/python_support.py
+++ b/mongoengine/python_support.py
@@ -22,6 +22,8 @@ StringIO = six.BytesIO
 # Additionally for Py2, try to use the faster cStringIO, if available
 if not PY3:
    try:
-        from cStringIO import StringIO
+        import cStringIO
    except ImportError:
        pass
+    else:
+        StringIO = cStringIO.StringIO