Merge branch 'dev' into feature/#151/represent-local-env

benoit-dubreuil · Apr 18, 2021 · c700524 · c700524
2 parents ee276d4 + e990885
commit c700524
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/conf/script/src/utils/meta_prog/encapsulation/__init__.py b/conf/script/src/utils/meta_prog/encapsulation/__init__.py
@@ -0,0 +1 @@
+from .export import *
diff --git a/conf/script/src/utils/meta_prog/encapsulation/export.py b/conf/script/src/utils/meta_prog/encapsulation/export.py
@@ -0,0 +1,22 @@
+import inspect
+import typing
+
+__all__ = ['export']
+
+
+def export(func: typing.Callable):
+    attribute_name_all: typing.Final[str] = '__all__'
+    module_api: list[str]
+
+    module = inspect.getmodule(func)
+    func_api = func.__qualname__
+
+    if not hasattr(module, attribute_name_all):
+        module_api = []
+        setattr(module, attribute_name_all, module_api)
+    else:
+        module_api = getattr(module, attribute_name_all)
+
+    module_api.append(func_api)
+
+    return func