diff --git a/dataset/persistence/database.py b/dataset/persistence/database.py index 6abada5..8b1a9b5 100644 --- a/dataset/persistence/database.py +++ b/dataset/persistence/database.py @@ -1,8 +1,13 @@ import logging import threading import re -from urlparse import parse_qs -from urllib import urlencode + +try: + from urllib.parse import urlencode + from urllib.parse import parse_qs +except ImportError: + from urllib import urlencode + from urlparse import parse_qs from sqlalchemy import create_engine from migrate.versioning.util import construct_engine