add docker image endpoint code

Author: SecretiveShell

src/inferadmin/docker.py

@@ -0,0 +1,22 @@
+from aiodocker import Docker
+
+class DockerManagerClass:
+    docker: Docker | None = None
+
+    async def init(self):
+        try:
+            self.docker = Docker()
+        except Exception as e:
+            raise Exception(f"Failed to initialize Docker: {e}")
+        
+    def get(self) -> Docker:
+        if self.docker is None:
+            raise Exception("Docker not initialized")
+        return self.docker
+    
+    async def close(self):
+        if self.docker is not None:
+            await self.docker.close()
+
+
+DockerManager = DockerManagerClass()

src/inferadmin/lifespan.py

@@ -1,8 +1,13 @@
 from contextlib import asynccontextmanager
 from fastapi import FastAPI
 from inferadmin.config.loader import config_manager
+from inferadmin.docker import DockerManager
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     await config_manager.load()
-    yield
+    await DockerManager.init()
+    
+    yield # run fastapi app
+    
+    await DockerManager.close()

src/inferadmin/routes/images/__init__.py

@@ -1,19 +1,46 @@
+from aiodocker import DockerError
 from fastapi import APIRouter
-from .models import GetImagesResponse, PutImageRequest, DeleteImageRequest
+
+from inferadmin.docker import DockerManager
 from inferadmin.routes.standard_models import BasicResponse
 
-router = APIRouter(
-    prefix='/images'
-)
+from .models import DeleteImageRequest, GetImagesResponse, PutImageRequest
+
+router = APIRouter(prefix="/images")
+
 
-@router.get('/')
+@router.get("/")
 async def get_images() -> GetImagesResponse:
     pass
 
-@router.put('/')
+
+@router.put("/")
 async def put_images(data: PutImageRequest) -> BasicResponse:
-    pass
+    """Pull a docker image"""
+    docker = DockerManager.get()
+
+    try:
+        await docker.images.pull(data.repo)
+    except Exception as e:
+        return BasicResponse(success=False, error_message=str(e))
 
-@router.delete('/')
+    return BasicResponse(success=True, error_message=None)
+
+
+@router.delete("/")
 async def delete_images(data: DeleteImageRequest) -> BasicResponse:
-    pass
+    """Delete a docker image"""
+    docker = DockerManager.get()
+
+    try:
+        await docker.images.get(data.id)
+    except DockerError as e:
+        if e.status == 404:
+            return BasicResponse(success=True, error_message=None)
+
+    try:
+        await docker.images.delete(data.id)
+    except Exception as e:
+        return BasicResponse(success=False, error_message=str(e))
+
+    return BasicResponse(success=True, error_message=None)