Fixed import for Python3

zalando · Dec 5, 2016 · 1afd06c · 1afd06c
1 parent 3cabb1b
commit 1afd06c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pg_view/helpers.py b/pg_view/helpers.py
@@ -1,9 +1,14 @@
-import ConfigParser
 import subprocess
+import sys
 
 from pg_view import consts
 from pg_view.exceptions import InvalidConnectionParamError
 
+if sys.hexversion >= 0x03000000:
+    import configparser as ConfigParser
+else:
+    import ConfigParser
+
 BYTES_IN_MB = 1048576
 SECTORS_IN_MB = 2048
 KB_IN_MB = 1024