Fixed #20167 -- Preserve the traceback of `ImportError`s in `import_by_path`.

Thanks @carljm for the review.
This commit is contained in:
Joe Friedl 2013-03-30 21:26:29 -04:00 committed by Simon Charette
parent 3ff3212713
commit 2d0db67813
2 changed files with 15 additions and 2 deletions

View File

@ -3,6 +3,7 @@ import os
import sys
from django.core.exceptions import ImproperlyConfigured
from django.utils import six
from django.utils.importlib import import_module
@ -19,8 +20,10 @@ def import_by_path(dotted_path, error_prefix=''):
try:
module = import_module(module_path)
except ImportError as e:
raise ImproperlyConfigured('%sError importing module %s: "%s"' % (
error_prefix, module_path, e))
msg = '%sError importing module %s: "%s"' % (
error_prefix, module_path, e)
six.reraise(ImproperlyConfigured, ImproperlyConfigured(msg),
sys.exc_info()[2])
try:
attr = getattr(module, class_name)
except AttributeError:

View File

@ -120,6 +120,16 @@ class ModuleImportTestCase(unittest.TestCase):
import_by_path('unexistent.module.path', error_prefix="Foo")
self.assertTrue(str(cm.exception).startswith('Foo'))
def test_import_error_traceback(self):
"""Test preserving the original traceback on an ImportError."""
try:
import_by_path('test_module.bad_module.content')
except ImproperlyConfigured:
traceback = sys.exc_info()[2]
self.assertIsNotNone(traceback.tb_next.tb_next,
'Should have more than the calling frame in the traceback.')
class ProxyFinder(object):
def __init__(self):