Source code for wolframclient.utils.importutils

from __future__ import absolute_import, print_function, unicode_literals

import os
from importlib import import_module

from wolframclient.utils import six


[docs]def module_path(module, *args): if isinstance(module, six.string_types): try: module = import_module(module) except ImportError: return None return os.path.join(os.path.dirname(os.path.realpath(module.__file__)), *args)
[docs]def import_string(dotted_path): """ Import a dotted module path and return the attribute/class designated by the last name in the path. Raise ImportError if the import failed. """ if not isinstance(dotted_path, six.string_types): return dotted_path try: module_path, class_name = dotted_path.rsplit(".", 1) except ValueError: raise ImportError("%s doesn't look like a module path" % dotted_path) module = import_module(module_path) if class_name == "__module__": return module try: return getattr(module, class_name) except AttributeError: raise ImportError( 'Module "%s" does not define a "%s" attribute/class' % (module_path, class_name) )
[docs]def safe_import_string(f): if isinstance(f, (list, tuple)): for path in f: try: return import_string(path) except ImportError: pass raise ImportError("Cannot import %s" % (f,)) if isinstance(f, six.string_types): return import_string(f) return f
[docs]def safe_import_string_and_call(f, *args, **kw): return safe_import_string(f)(*args, **kw)
[docs]class API(object): def __init__(self, importer=safe_import_string, **mapping): self.__dict__["importer"] = importer self.__dict__["mapping"] = mapping self.__dict__["imports"] = {} def __getattr__(self, value): key = self.__dict__["mapping"][value] try: return self.__dict__["imports"][key] except KeyError: self.__dict__["imports"][key] = self.__dict__["importer"](key) return self.__dict__["imports"][key] def __getitem__(self, key): try: return getattr(self, key) except AttributeError: raise KeyError("unregistred module %s" % key) def __len__(self): return len(self.__dict__["mapping"]) def __bool__(self): return bool(self.__dict__["mapping"])
[docs] def keys(self): return iter(self)
[docs] def values(self): for key in self: yield self[key]
[docs] def items(self): return zip(self.keys(), self.values())
def __repr__(self): return "<%s %s>" % (self.__class__.__name__, ", ".join(sorted(self))) def __dir__(self): return list(self) def __iter__(self): return iter(self.__dict__["mapping"].keys())