mmar-mapi 1.0.10__py3-none-any.whl → 1.0.11__py3-none-any.whl

mmar_mapi/__init__.py

@@ -17,7 +17,7 @@ from .models.tracks import TrackInfo, DomainInfo
 from .models.widget import Widget
 from .utils import make_session_id, chunked
 from .xml_parser import XMLParser
-
+from .utils_import import load_main_objects
 
 __all__ = [
     "AIMessage",
@@ -44,6 +44,7 @@ __all__ = [
     "Widget",
     "XMLParser",
     "chunked",
+    "load_main_objects",
     "make_content",
     "make_session_id",
 ]

mmar_mapi/utils_import.py

@@ -0,0 +1,42 @@
+from pkgutil import iter_modules
+from importlib import import_module
+from types import ModuleType
+
+from loguru import logger
+
+def convert_snake_to_pascal(name: str) -> str:
+    """ snake_to_pascal -> SnakeToPascal """
+    return "".join(map(str.capitalize, name.split("_")))
+
+
+def get_main_object_name(module: type[ModuleType]) -> str:
+    module_name = module.__name__.rsplit(".", 1)[-1]
+    main_object_name = convert_snake_to_pascal(module_name)
+    return main_object_name
+
+
+def get_main_object(module: type[ModuleType], obj_type):
+    """tries to find SomeObject in object src.some_object"""
+    main_object_name = get_main_object_name(module)
+    try:
+        res = getattr(module, main_object_name)
+        if not isinstance(res, obj_type) and not issubclass(res, obj_type):
+            logger.error(f"Failed to load {module}.{main_object_name}: expected {obj_type} but found {type(res)}")
+            return None
+        return res
+    except AttributeError as ex:
+        logger.error(f"Failed to load {module}.{main_object_name}: {ex}")
+        return None
+
+
+def load_modules(package_name: str) -> list[type[ModuleType]]:
+    package = import_module(package_name)
+    res = [import_module(module_name) for _, module_name, _ in iter_modules(package.__path__, package_name + ".")]
+    return res
+
+
+def load_main_objects(package_name: str, obj_type: type) -> dict[str, object]:
+    modules = load_modules(package_name)
+    main_objects = [get_main_object(m, obj_type) for m in modules]
+    res = {obj.__name__: obj for obj in main_objects}
+    return res

{mmar_mapi-1.0.10.dist-info → mmar_mapi-1.0.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mmar-mapi
-Version: 1.0.10
+Version: 1.0.11
 Summary: Common pure/IO utilities for multi-modal architectures team
 Keywords: 
 Author: Eugene Tagin
@@ -20,6 +20,7 @@ Classifier: Topic :: Software Development
 Classifier: Topic :: Utilities
 Classifier: Typing :: Typed
 Requires-Dist: pydantic~=2.11.7
+Requires-Dist: loguru~=0.7.3
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 

{mmar_mapi-1.0.10.dist-info → mmar_mapi-1.0.11.dist-info}/RECORD

@@ -1,4 +1,4 @@
-mmar_mapi/__init__.py,sha256=NxCsA3aMKjxIZ2f4pQTJ1LPpg7t2yz7PsiDeb_HrAJE,1109
+mmar_mapi/__init__.py,sha256=Q2WFyNje0mDzP3eK7k-900W4Uty82BlulKv9-FmkcZs,1177
 mmar_mapi/api.py,sha256=C9Sr8dISvf51xfEznPjccI_odaG4coQE3HI_0jVpjMQ,1677
 mmar_mapi/file_storage.py,sha256=GbahBabBdAKjlAnv1MszERUxxZyA9HGMiR9tz2a9dgY,4409
 mmar_mapi/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -10,8 +10,9 @@ mmar_mapi/models/tracks.py,sha256=HKDp-BX1p7AlDfSEKfOKCu0TRSK9cD4Dmq1vJt8oRjw,30
 mmar_mapi/models/widget.py,sha256=ue5o4AkN8SG09aA8eQLOOxbwah-mkZir0ZGL6OA8S9Q,1355
 mmar_mapi/type_union.py,sha256=diwmzcnbqkpGFckPHNw9o8zyQ955mOGNvhTlcBJ0RMI,1905
 mmar_mapi/utils.py,sha256=oVrEAHGcfFBprwdTEzSbFCpdLtkDRVzWtOnYpS439os,394
+mmar_mapi/utils_import.py,sha256=47qNpd41ZvzJQsPtielqTlikO-N3xUXlj5LnoGrbM5g,1562
 mmar_mapi/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
-mmar_mapi-1.0.10.dist-info/licenses/LICENSE,sha256=2A90w8WjhOgQXnFuUijKJYazaqZ4_NTokYb9Po4y-9k,1061
-mmar_mapi-1.0.10.dist-info/WHEEL,sha256=Pi5uDq5Fdo_Rr-HD5h9BiPn9Et29Y9Sh8NhcJNnFU1c,79
-mmar_mapi-1.0.10.dist-info/METADATA,sha256=cL99l0W1GOtgnJOUNDyflzdkbMNHNfGwNpdiLnhFIz0,915
-mmar_mapi-1.0.10.dist-info/RECORD,,
+mmar_mapi-1.0.11.dist-info/licenses/LICENSE,sha256=2A90w8WjhOgQXnFuUijKJYazaqZ4_NTokYb9Po4y-9k,1061
+mmar_mapi-1.0.11.dist-info/WHEEL,sha256=Pi5uDq5Fdo_Rr-HD5h9BiPn9Et29Y9Sh8NhcJNnFU1c,79
+mmar_mapi-1.0.11.dist-info/METADATA,sha256=JGGIso4M27c1Rf48uIx5OtdTGQ5-PQwzZ2Kr89kZm74,944
+mmar_mapi-1.0.11.dist-info/RECORD,,