diff options
author | Mike Bayer <mike_mp@zzzcomputing.com> | 2007-03-06 17:17:27 +0000 |
---|---|---|
committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2007-03-06 17:17:27 +0000 |
commit | b401cd2ea73d14f57e19bf0c7f77406d80c3293c (patch) | |
tree | 644d49b94938306eab714b8ce5ae411bdc802c42 /lib/sqlalchemy/engine/url.py | |
parent | 7b43cd5092436150a4c9648431173c9e79356090 (diff) | |
download | sqlalchemy-b401cd2ea73d14f57e19bf0c7f77406d80c3293c.tar.gz |
decorated ImportError thrown when the <database>:// module isnt found
Diffstat (limited to 'lib/sqlalchemy/engine/url.py')
-rw-r--r-- | lib/sqlalchemy/engine/url.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/lib/sqlalchemy/engine/url.py b/lib/sqlalchemy/engine/url.py index 6a0180d62..8f9d33d54 100644 --- a/lib/sqlalchemy/engine/url.py +++ b/lib/sqlalchemy/engine/url.py @@ -1,5 +1,6 @@ import re import cgi +import sys import urllib from sqlalchemy import exceptions @@ -70,8 +71,12 @@ class URL(object): def get_module(self): """Return the SQLAlchemy database module corresponding to this URL's driver name.""" - - return getattr(__import__('sqlalchemy.databases.%s' % self.drivername).databases, self.drivername) + try: + return getattr(__import__('sqlalchemy.databases.%s' % self.drivername).databases, self.drivername) + except ImportError: + if sys.exc_info()[2].tb_next is None: + raise exceptions.ArgumentError('unknown database %r' % self.drivername) + raise def translate_connect_args(self, names): """Translate this URL's attributes into a dictionary of connection arguments. |