try #2

carbonblack · Jun 23, 2017 · 1257525 · 1257525
1 parent 555df18
commit 1257525
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/src/cbapi/defense/cli.py b/src/cbapi/defense/cli.py
@@ -9,7 +9,7 @@
 import getpass
 import six
 if six.PY3:
-    from io import BytesIO as StringIO
+    from io import StringIO as StringIO
 else:
     from cStringIO import StringIO
 
@@ -49,7 +49,7 @@ def configure(opts):
     token = input("API key: ")
 
     config = RawConfigParser()
-    config.readfp(StringIO(b'[default]'))
+    config.readfp(StringIO('[default]'))
     config.set("default", "url", url)
     config.set("default", "token", "{0}/{1}".format(token, connector_id))
     config.set("default", "ssl_verify", ssl_verify)

diff --git a/src/cbapi/protection/cli.py b/src/cbapi/protection/cli.py
@@ -9,7 +9,7 @@
 import getpass
 import six
 if six.PY3:
-    from io import BytesIO as StringIO
+    from io import StringIO as StringIO
 else:
     from cStringIO import StringIO
 
@@ -56,7 +56,7 @@ def configure(opts):
     token = input("API token: ")
 
     config = RawConfigParser()
-    config.readfp(StringIO(b'[default]'))
+    config.readfp(StringIO('[default]'))
     config.set("default", "url", url)
     config.set("default", "token", token)
     config.set("default", "ssl_verify", ssl_verify)

diff --git a/src/cbapi/response/cli.py b/src/cbapi/response/cli.py
@@ -9,7 +9,7 @@
 import getpass
 import six
 if six.PY3:
-    from io import BytesIO as StringIO
+    from io import StringIO as StringIO
 else:
     from cStringIO import StringIO
 
@@ -70,7 +70,7 @@ def configure(opts):
             return 1
 
     config = RawConfigParser()
-    config.readfp(StringIO(b'[default]'))
+    config.readfp(StringIO('[default]'))
     config.set("default", "url", url)
     config.set("default", "token", token)
     config.set("default", "ssl_verify", ssl_verify)