Merge pull request #45 from gratipay/import-tweak

use cleaner test for Python major version
liberapay · Mar 11, 2015 · e818e10 · e818e10
2 parents bb4e7b4 + 22015bc
commit e818e10
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/postgres/__init__.py b/postgres/__init__.py
@@ -157,7 +157,7 @@
 from __future__ import absolute_import, division, print_function, unicode_literals
 
 import sys
-try:                    # Python 2
+if sys.version_info[0] == 2:    # Python 2
     import urlparse
 
     # "Note: In Python 2, if you want to uniformly receive all your database
@@ -169,7 +169,7 @@
     psycopg2.extensions.register_type(psycopg2.extensions.UNICODE)
     psycopg2.extensions.register_type(psycopg2.extensions.UNICODEARRAY)
 
-except ImportError:     # Python 3
+else:                           # Python 3
     import urllib.parse as urlparse
 from collections import namedtuple