1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
# 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
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
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])
|
