Merge pull request #168 from benoit-dubreuil/feature/#166/export-deco…

…rator-for-public-api Feature/#166/export decorator for public api
benoit-dubreuil · Apr 18, 2021 · e990885 · e990885
2 parents 55a7d83 + 21cfce8
commit e990885
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