Option to prune images when deleting template.

gns3server/api/routes/controller/templates.py

@@ -25,14 +25,15 @@ import logging
 
 log = logging.getLogger(__name__)
 
-from fastapi import APIRouter, Request, Response, HTTPException, Depends, Response, status
-from typing import List
+from fastapi import APIRouter, Request, HTTPException, Depends, Response, status
+from typing import List, Optional
 from uuid import UUID
 
 from gns3server import schemas
 from gns3server.db.repositories.templates import TemplatesRepository
 from gns3server.services.templates import TemplatesService
 from gns3server.db.repositories.rbac import RbacRepository
+from gns3server.db.repositories.images import ImagesRepository
 
 from .dependencies.authentication import get_current_active_user
 from .dependencies.database import get_repository
@@ -97,7 +98,9 @@ async def update_template(
 @router.delete("/{template_id}", status_code=status.HTTP_204_NO_CONTENT)
 async def delete_template(
         template_id: UUID,
+        prune_images: Optional[bool] = False,
         templates_repo: TemplatesRepository = Depends(get_repository(TemplatesRepository)),
+        images_repo: RbacRepository = Depends(get_repository(ImagesRepository)),
         rbac_repo: RbacRepository = Depends(get_repository(RbacRepository))
 ) -> Response:
     """
@@ -106,6 +109,8 @@ async def delete_template(
 
     await TemplatesService(templates_repo).delete_template(template_id)
     await rbac_repo.delete_all_permissions_with_path(f"/templates/{template_id}")
+    if prune_images:
+        await images_repo.prune_images()
     return Response(status_code=status.HTTP_204_NO_CONTENT)
 
 

gns3server/db/repositories/images.py

@@ -117,10 +117,11 @@ class ImagesRepository(BaseRepository):
         images_deleted = 0
         for image in images:
             try:
+                log.debug(f"Deleting image '{image.path}'")
                 os.remove(image.path)
             except OSError:
                 log.warning(f"Could not delete image file {image.path}")
             if await self.delete_image(image.filename):
                 images_deleted += 1
-        log.info(f"{images_deleted} image have been deleted")
+        log.info(f"{images_deleted} image(s) have been deleted")
         return images_deleted

gns3server/services/templates.py

@@ -159,7 +159,7 @@ class TemplatesService:
 
         image = await self._templates_repo.get_image(image_name)
         if not image or not os.path.exists(image.path):
-            raise ControllerNotFoundError(f"Image {image_name} could not be found")
+            raise ControllerNotFoundError(f"Image '{image_name}' could not be found")
         return image
 
     async def _find_images(self, template_type: str, settings: dict) -> List[models.Image]: