python3 fix

omarish · Oct 20, 2015 · 73c1e84 · 73c1e84
1 parent b05f9cc
commit 73c1e84
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/django_cprofile_middleware/middleware.py b/django_cprofile_middleware/middleware.py
@@ -2,8 +2,13 @@
     import cProfile as profile
 except ImportError:
     import profile
+
+try:
+    from cStringIO import StringIO
+except:
+    from io import StringIO
+
 import pstats
-from io import StringIO
 from django.conf import settings