Total Complexity | 4 |
Total Lines | 31 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | import inspect |
||
2 | |||
3 | from typing import Any |
||
4 | |||
5 | |||
6 | def export(target: Any) -> Any: |
||
7 | """ |
||
8 | Mark a module-level object as exported. |
||
9 | |||
10 | Simplifies tracking of objects available via wildcard imports. |
||
11 | |||
12 | """ |
||
13 | frm = inspect.stack()[1] |
||
14 | mod = inspect.getmodule(frm[0]) |
||
15 | |||
16 | __all__ = getattr(mod, '__all__', None) |
||
17 | |||
18 | if __all__ is None: |
||
19 | __all__ = [] |
||
20 | setattr(mod, '__all__', __all__) |
||
21 | |||
22 | elif not isinstance(__all__, list): |
||
23 | __all__ = list(__all__) |
||
24 | setattr(mod, '__all__', __all__) |
||
25 | |||
26 | target_name = target.__name__ |
||
27 | if target_name not in __all__: |
||
28 | __all__.append(target_name) |
||
29 | |||
30 | return target |
||
31 |