container-manager-mcp 1.0.3__py3-none-any.whl → 1.2.0__py3-none-any.whl

container_manager_mcp/__init__.py

@@ -1,26 +1,32 @@
 #!/usr/bin/env python
 # coding: utf-8
 
-from container_manager_mcp.container_manager import (
-    container_manager,
-    create_manager,
-    ContainerManagerBase,
-    DockerManager,
-    PodmanManager,
-)
-from container_manager_mcp.container_manager_mcp import main
+import importlib
+import inspect
+
+# List of modules to import from
+MODULES = [
+    "container_manager_mcp.container_manager",
+    "container_manager_mcp.container_manager_mcp",
+    "container_manager_mcp.container_manager_a2a",
+]
+
+# Initialize __all__ to expose all public classes and functions
+__all__ = []
+
+# Dynamically import all classes and functions from the specified modules
+for module_name in MODULES:
+    module = importlib.import_module(module_name)
+    for name, obj in inspect.getmembers(module):
+        # Include only classes and functions, excluding private (starting with '_')
+        if (inspect.isclass(obj) or inspect.isfunction(obj)) and not name.startswith(
+            "_"
+        ):
+            globals()[name] = obj
+            __all__.append(name)
 
 """
 container-manager
 
 Manage your containers using docker, podman, compose, or docker swarm!
 """
-
-__all__ = [
-    "main",
-    "create_manager",
-    "container_manager",
-    "ContainerManagerBase",
-    "DockerManager",
-    "PodmanManager",
-]

container_manager_mcp/__main__.py

@@ -1,6 +1,6 @@
 #!/usr/bin/python
 # coding: utf-8
-from container_manager_mcp.container_manager_mcp import main
+from container_manager_mcp.container_manager_mcp import container_manager_mcp
 
 if __name__ == "__main__":
-    main()
+    container_manager_mcp()