42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
# encoding: utf-8
|
|
"""
|
|
A simple utility to import something by its string name.
|
|
"""
|
|
# Copyright (c) IPython Development Team.
|
|
# Distributed under the terms of the Modified BSD License.
|
|
|
|
from ipython_genutils.py3compat import cast_bytes_py2
|
|
from six import string_types
|
|
|
|
def import_item(name):
|
|
"""Import and return ``bar`` given the string ``foo.bar``.
|
|
|
|
Calling ``bar = import_item("foo.bar")`` is the functional equivalent of
|
|
executing the code ``from foo import bar``.
|
|
|
|
Parameters
|
|
----------
|
|
name : string
|
|
The fully qualified name of the module/package being imported.
|
|
|
|
Returns
|
|
-------
|
|
mod : module object
|
|
The module that was imported.
|
|
"""
|
|
if not isinstance(name, string_types):
|
|
raise TypeError("import_item accepts strings, not '%s'." % type(name))
|
|
name = cast_bytes_py2(name)
|
|
parts = name.rsplit('.', 1)
|
|
if len(parts) == 2:
|
|
# called with 'foo.bar....'
|
|
package, obj = parts
|
|
module = __import__(package, fromlist=[obj])
|
|
try:
|
|
pak = getattr(module, obj)
|
|
except AttributeError:
|
|
raise ImportError('No module named %s' % obj)
|
|
return pak
|
|
else:
|
|
# called with un-dotted string
|
|
return __import__(parts[0])
|