Source code
Revision control
Copy as Markdown
Other Tools
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
import inspect
import os
def find_object(path):
"""
Find a Python object given a path of the form <modulepath>:<objectpath>.
Conceptually equivalent to
def find_object(modulepath, objectpath):
import <modulepath> as mod
return mod.<objectpath>
"""
if path.count(":") != 1:
raise ValueError(f'python path {path!r} does not have the form "module:object"')
modulepath, objectpath = path.split(":")
obj = __import__(modulepath)
for a in modulepath.split(".")[1:]:
obj = getattr(obj, a)
for a in objectpath.split("."):
obj = getattr(obj, a)
return obj
def import_sibling_modules(exceptions=None):
"""
Import all Python modules that are siblings of the calling module.
Args:
exceptions (list): A list of file names to exclude (caller and
__init__.py are implicitly excluded).
"""
frame = inspect.stack()[1]
mod = inspect.getmodule(frame[0])
name = os.path.basename(mod.__file__) # type: ignore
excs = {"__init__.py", name}
if exceptions:
excs.update(exceptions)
modpath = mod.__name__ # type: ignore
if not name.startswith("__init__.py"):
modpath = modpath.rsplit(".", 1)[0]
for f in os.listdir(os.path.dirname(mod.__file__)): # type: ignore
if f.endswith(".py") and f not in excs:
__import__(modpath + "." + f[:-3])