Fix import bug with StringIO and Python 3.

054066d5 · William Tisäter · b76bdd49 · 054066d5
Commit 054066d5 authored Sep 20, 2012 by William Tisäter
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletions

pygeoip/__init__.py
+5 -1

No files found.
--- a/pygeoip/__init__.py
+++ b/pygeoip/__init__.py
@@ -34,7 +34,11 @@ import socket
 import mmap
 import gzip
 import codecs
-from StringIO import StringIO
+
+try:
+    from StringIO import StringIO
+except ImportError:
+    from io import StringIO

 from . import const
 from .util import ip2long