container-manager-mcp 1.0.2__py3-none-any.whl → 1.0.4__py3-none-any.whl

container_manager_mcp/container_manager_mcp.py

@@ -39,6 +39,29 @@ def to_boolean(string):
         raise ValueError(f"Cannot convert '{string}' to boolean")
 
 
+def parse_image_string(image: str, default_tag: str = "latest") -> tuple[str, str]:
+    """
+    Parse a container image string into image and tag components.
+
+    Args:
+        image: Input image string (e.g., 'registry.arpa/ubuntu/ubuntu:latest' or 'nginx')
+        default_tag: Fallback tag if none is specified (default: 'latest')
+
+    Returns:
+        Tuple of (image, tag) where image includes registry/repository, tag is the tag or default_tag
+    """
+    # Split on the last ':' to separate image and tag
+    if ":" in image:
+        parts = image.rsplit(":", 1)
+        image_name, tag = parts[0], parts[1]
+        # Ensure tag is valid (not a port or malformed)
+        if "/" in tag or not tag:
+            # If tag contains '/' or is empty, assume no tag was provided
+            return image, default_tag
+        return image_name, tag
+    return image, default_tag
+
+
 environment_silent = os.environ.get("SILENT", False)
 environment_log_file = os.environ.get("LOG_FILE", None)
 environment_container_manager_type = os.environ.get("CONTAINER_MANAGER_TYPE", None)
@@ -171,8 +194,13 @@ async def list_images(
     tags={"container_management"},
 )
 async def pull_image(
-    image: str = Field(description="Image name to pull"),
-    tag: str = Field(description="Image tag", default="latest"),
+    image: str = Field(
+        description="Image name to pull (e.g., nginx, registry.arpa/ubuntu/ubuntu:latest)."
+    ),
+    tag: str = Field(
+        description="Image tag (overridden if tag is included in image string)",
+        default="latest",
+    ),
     platform: Optional[str] = Field(
         description="Platform (e.g., linux/amd64)", default=None
     ),
@@ -191,12 +219,14 @@ async def pull_image(
     ),
 ) -> Dict:
     logger = logging.getLogger("ContainerManager")
+    # Parse image string to separate image and tag
+    parsed_image, parsed_tag = parse_image_string(image, tag)
     logger.debug(
-        f"Pulling image {image}:{tag} for {manager_type}, silent: {silent}, log_file: {log_file}"
+        f"Pulling image {parsed_image}:{parsed_tag} for {manager_type}, silent: {silent}, log_file: {log_file}"
     )
     try:
         manager = create_manager(manager_type, silent, log_file)
-        return manager.pull_image(image, tag, platform)
+        return manager.pull_image(parsed_image, parsed_tag, platform)
     except Exception as e:
         logger.error(f"Failed to pull image: {str(e)}")
         raise RuntimeError(f"Failed to pull image: {str(e)}")
@@ -241,6 +271,44 @@ async def remove_image(
         raise RuntimeError(f"Failed to remove image: {str(e)}")
 
 
+@mcp.tool(
+    annotations={
+        "title": "Prune Images",
+        "readOnlyHint": False,
+        "destructiveHint": True,
+        "idempotentHint": True,
+        "openWorldHint": False,
+    },
+    tags={"container_management"},
+)
+async def prune_images(
+    all: bool = Field(description="Prune all unused images", default=False),
+    manager_type: Optional[str] = Field(
+        description="Container manager: docker, podman (default: auto-detect)",
+        default=environment_container_manager_type,
+    ),
+    silent: Optional[bool] = Field(
+        description="Suppress output", default=environment_silent
+    ),
+    log_file: Optional[str] = Field(
+        description="Path to log file", default=environment_log_file
+    ),
+    ctx: Context = Field(
+        description="MCP context for progress reporting", default=None
+    ),
+) -> Dict:
+    logger = logging.getLogger("ContainerManager")
+    logger.debug(
+        f"Pruning images for {manager_type}, all: {all}, silent: {silent}, log_file: {log_file}"
+    )
+    try:
+        manager = create_manager(manager_type, silent, log_file)
+        return manager.prune_images(all=all)
+    except Exception as e:
+        logger.error(f"Failed to prune images: {str(e)}")
+        raise RuntimeError(f"Failed to prune images: {str(e)}")
+
+
 @mcp.tool(
     annotations={
         "title": "List Containers",
@@ -414,6 +482,43 @@ async def remove_container(
         raise RuntimeError(f"Failed to remove container: {str(e)}")
 
 
+@mcp.tool(
+    annotations={
+        "title": "Prune Containers",
+        "readOnlyHint": False,
+        "destructiveHint": True,
+        "idempotentHint": True,
+        "openWorldHint": False,
+    },
+    tags={"container_management"},
+)
+async def prune_containers(
+    manager_type: Optional[str] = Field(
+        description="Container manager: docker, podman (default: auto-detect)",
+        default=environment_container_manager_type,
+    ),
+    silent: Optional[bool] = Field(
+        description="Suppress output", default=environment_silent
+    ),
+    log_file: Optional[str] = Field(
+        description="Path to log file", default=environment_log_file
+    ),
+    ctx: Context = Field(
+        description="MCP context for progress reporting", default=None
+    ),
+) -> Dict:
+    logger = logging.getLogger("ContainerManager")
+    logger.debug(
+        f"Pruning containers for {manager_type}, silent: {silent}, log_file: {log_file}"
+    )
+    try:
+        manager = create_manager(manager_type, silent, log_file)
+        return manager.prune_containers()
+    except Exception as e:
+        logger.error(f"Failed to prune containers: {str(e)}")
+        raise RuntimeError(f"Failed to prune containers: {str(e)}")
+
+
 @mcp.tool(
     annotations={
         "title": "Get Container Logs",
@@ -609,6 +714,44 @@ async def remove_volume(
         raise RuntimeError(f"Failed to remove volume: {str(e)}")
 
 
+@mcp.tool(
+    annotations={
+        "title": "Prune Volumes",
+        "readOnlyHint": False,
+        "destructiveHint": True,
+        "idempotentHint": True,
+        "openWorldHint": False,
+    },
+    tags={"container_management"},
+)
+async def prune_volumes(
+    all: bool = Field(description="Remove all volumes (dangerous)", default=False),
+    manager_type: Optional[str] = Field(
+        description="Container manager: docker, podman (default: auto-detect)",
+        default=environment_container_manager_type,
+    ),
+    silent: Optional[bool] = Field(
+        description="Suppress output", default=environment_silent
+    ),
+    log_file: Optional[str] = Field(
+        description="Path to log file", default=environment_log_file
+    ),
+    ctx: Context = Field(
+        description="MCP context for progress reporting", default=None
+    ),
+) -> Dict:
+    logger = logging.getLogger("ContainerManager")
+    logger.debug(
+        f"Pruning volumes for {manager_type}, all: {all}, silent: {silent}, log_file: {log_file}"
+    )
+    try:
+        manager = create_manager(manager_type, silent, log_file)
+        return manager.prune_volumes(all=all)
+    except Exception as e:
+        logger.error(f"Failed to prune volumes: {str(e)}")
+        raise RuntimeError(f"Failed to prune volumes: {str(e)}")
+
+
 @mcp.tool(
     annotations={
         "title": "List Networks",
@@ -723,6 +866,82 @@ async def remove_network(
         raise RuntimeError(f"Failed to remove network: {str(e)}")
 
 
+@mcp.tool(
+    annotations={
+        "title": "Prune Networks",
+        "readOnlyHint": False,
+        "destructiveHint": True,
+        "idempotentHint": True,
+        "openWorldHint": False,
+    },
+    tags={"container_management"},
+)
+async def prune_networks(
+    manager_type: Optional[str] = Field(
+        description="Container manager: docker, podman (default: auto-detect)",
+        default=environment_container_manager_type,
+    ),
+    silent: Optional[bool] = Field(
+        description="Suppress output", default=environment_silent
+    ),
+    log_file: Optional[str] = Field(
+        description="Path to log file", default=environment_log_file
+    ),
+    ctx: Context = Field(
+        description="MCP context for progress reporting", default=None
+    ),
+) -> Dict:
+    logger = logging.getLogger("ContainerManager")
+    logger.debug(
+        f"Pruning networks for {manager_type}, silent: {silent}, log_file: {log_file}"
+    )
+    try:
+        manager = create_manager(manager_type, silent, log_file)
+        return manager.prune_networks()
+    except Exception as e:
+        logger.error(f"Failed to prune networks: {str(e)}")
+        raise RuntimeError(f"Failed to prune networks: {str(e)}")
+
+
+@mcp.tool(
+    annotations={
+        "title": "Prune System",
+        "readOnlyHint": False,
+        "destructiveHint": True,
+        "idempotentHint": True,
+        "openWorldHint": False,
+    },
+    tags={"container_management"},
+)
+async def prune_system(
+    force: bool = Field(description="Force prune", default=False),
+    all: bool = Field(description="Prune all unused resources", default=False),
+    manager_type: Optional[str] = Field(
+        description="Container manager: docker, podman (default: auto-detect)",
+        default=environment_container_manager_type,
+    ),
+    silent: Optional[bool] = Field(
+        description="Suppress output", default=environment_silent
+    ),
+    log_file: Optional[str] = Field(
+        description="Path to log file", default=environment_log_file
+    ),
+    ctx: Context = Field(
+        description="MCP context for progress reporting", default=None
+    ),
+) -> Dict:
+    logger = logging.getLogger("ContainerManager")
+    logger.debug(
+        f"Pruning system for {manager_type}, force: {force}, all: {all}, silent: {silent}, log_file: {log_file}"
+    )
+    try:
+        manager = create_manager(manager_type, silent, log_file)
+        return manager.prune_system(force, all)
+    except Exception as e:
+        logger.error(f"Failed to prune system: {str(e)}")
+        raise RuntimeError(f"Failed to prune system: {str(e)}")
+
+
 # Swarm-specific tools
 
 

{container_manager_mcp-1.0.2.dist-info → container_manager_mcp-1.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: container-manager-mcp
-Version: 1.0.2
+Version: 1.0.4
 Summary: Container Manager manage Docker, Docker Swarm, and Podman containers as an MCP Server
 Author-email: Audel Rouhi <knucklessg1@gmail.com>
 License: MIT
@@ -48,7 +48,7 @@ Dynamic: license-file
 ![PyPI - Wheel](https://img.shields.io/pypi/wheel/container-manager-mcp)
 ![PyPI - Implementation](https://img.shields.io/pypi/implementation/container-manager-mcp)
 
-*Version: 1.0.2*
+*Version: 1.0.4*
 
 Container Manager MCP Server provides a robust interface to manage Docker and Podman containers, networks, volumes, and Docker Swarm services through a FastMCP server, enabling programmatic and remote container management.
 

container_manager_mcp-1.0.4.dist-info/RECORD

@@ -0,0 +1,10 @@
+container_manager_mcp/__init__.py,sha256=N3bhKd_oh5YmBBl9N1omfZgaXhJyP0vOzH4VKxs68_g,506
+container_manager_mcp/__main__.py,sha256=zic5tX336HG8LfdzQQ0sDVx-tMSOsgOZCtaxHWgJ4Go,134
+container_manager_mcp/container_manager.py,sha256=sHxM17dj7luzp1Ak2aQJ3JLqHd3POO3S2_W32HQwWog,89167
+container_manager_mcp/container_manager_mcp.py,sha256=E3FOTFILyYYoJF3Erkpl5zapkiDi2pKlJlkZByc-8QE,47361
+container_manager_mcp-1.0.4.dist-info/licenses/LICENSE,sha256=Z1xmcrPHBnGCETO_LLQJUeaSNBSnuptcDVTt4kaPUOE,1060
+container_manager_mcp-1.0.4.dist-info/METADATA,sha256=XhTi2pEQeO1cfnrVgVViq_LiwGqLqUyhsogD-jXUFys,8238
+container_manager_mcp-1.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+container_manager_mcp-1.0.4.dist-info/entry_points.txt,sha256=I23pXcCgAShlfYbENzs3kbw3l1lU9Gy7lODPfRqeeiA,156
+container_manager_mcp-1.0.4.dist-info/top_level.txt,sha256=B7QQLOd9mBdu0lsPKqyu4T8-zUtbqKzQJbMbtAzoozU,22
+container_manager_mcp-1.0.4.dist-info/RECORD,,

container_manager_mcp-1.0.2.dist-info/RECORD

@@ -1,10 +0,0 @@
-container_manager_mcp/__init__.py,sha256=N3bhKd_oh5YmBBl9N1omfZgaXhJyP0vOzH4VKxs68_g,506
-container_manager_mcp/__main__.py,sha256=zic5tX336HG8LfdzQQ0sDVx-tMSOsgOZCtaxHWgJ4Go,134
-container_manager_mcp/container_manager.py,sha256=9TPpZw2gB_rz4oLsu45jBeixkG9B-OYppsGM4ctqb5I,72495
-container_manager_mcp/container_manager_mcp.py,sha256=PSKbw4NuoGWuZZJdTOstiIu_2mq2wZdNNM1gDQdZMNU,39788
-container_manager_mcp-1.0.2.dist-info/licenses/LICENSE,sha256=Z1xmcrPHBnGCETO_LLQJUeaSNBSnuptcDVTt4kaPUOE,1060
-container_manager_mcp-1.0.2.dist-info/METADATA,sha256=IyGwMPvMD3xjEeffWzuWO9wTOQTAIoNLQmhaJCTBxQw,8238
-container_manager_mcp-1.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-container_manager_mcp-1.0.2.dist-info/entry_points.txt,sha256=I23pXcCgAShlfYbENzs3kbw3l1lU9Gy7lODPfRqeeiA,156
-container_manager_mcp-1.0.2.dist-info/top_level.txt,sha256=B7QQLOd9mBdu0lsPKqyu4T8-zUtbqKzQJbMbtAzoozU,22
-container_manager_mcp-1.0.2.dist-info/RECORD,,