refactor(routes): add registrar for example images

py/routes/example_images_routes.py

@@ -1,4 +1,9 @@
 import logging
+from typing import Callable
+
+from aiohttp import web
+
+from .example_images_route_registrar import ExampleImagesRouteRegistrar
 from ..utils.example_images_download_manager import DownloadManager
 from ..utils.example_images_processor import ExampleImagesProcessor
 from ..utils.example_images_file_manager import ExampleImagesFileManager
@@ -6,22 +11,31 @@ from ..services.websocket_manager import ws_manager
 
 logger = logging.getLogger(__name__)
 
+
 class ExampleImagesRoutes:
     """Routes for example images related functionality"""
-    
+
     @staticmethod
-    def setup_routes(app):
-        """Register example images routes"""
-        app.router.add_post('/api/lm/download-example-images', ExampleImagesRoutes.download_example_images)
-        app.router.add_post('/api/lm/import-example-images', ExampleImagesRoutes.import_example_images)
-        app.router.add_get('/api/lm/example-images-status', ExampleImagesRoutes.get_example_images_status)
-        app.router.add_post('/api/lm/pause-example-images', ExampleImagesRoutes.pause_example_images)
-        app.router.add_post('/api/lm/resume-example-images', ExampleImagesRoutes.resume_example_images)
-        app.router.add_post('/api/lm/open-example-images-folder', ExampleImagesRoutes.open_example_images_folder)
-        app.router.add_get('/api/lm/example-image-files', ExampleImagesRoutes.get_example_image_files)
-        app.router.add_get('/api/lm/has-example-images', ExampleImagesRoutes.has_example_images)
-        app.router.add_post('/api/lm/delete-example-image', ExampleImagesRoutes.delete_example_image)
-        app.router.add_post('/api/lm/force-download-example-images', ExampleImagesRoutes.force_download_example_images)
+    def setup_routes(app: web.Application) -> None:
+        """Register example images routes using the registrar."""
+
+        registrar = ExampleImagesRouteRegistrar(app)
+        registrar.register_routes(ExampleImagesRoutes._route_mapping())
+
+    @staticmethod
+    def _route_mapping() -> dict[str, Callable[[web.Request], object]]:
+        return {
+            "download_example_images": ExampleImagesRoutes.download_example_images,
+            "import_example_images": ExampleImagesRoutes.import_example_images,
+            "get_example_images_status": ExampleImagesRoutes.get_example_images_status,
+            "pause_example_images": ExampleImagesRoutes.pause_example_images,
+            "resume_example_images": ExampleImagesRoutes.resume_example_images,
+            "open_example_images_folder": ExampleImagesRoutes.open_example_images_folder,
+            "get_example_image_files": ExampleImagesRoutes.get_example_image_files,
+            "has_example_images": ExampleImagesRoutes.has_example_images,
+            "delete_example_image": ExampleImagesRoutes.delete_example_image,
+            "force_download_example_images": ExampleImagesRoutes.force_download_example_images,
+        }
 
     @staticmethod
     async def download_example_images(request):
@@ -42,7 +56,7 @@ class ExampleImagesRoutes:
     async def resume_example_images(request):
         """Resume the example images download"""
         return await DownloadManager.resume_download(request)
-        
+
     @staticmethod
     async def open_example_images_folder(request):
         """Open the example images folder for a specific model"""
@@ -57,7 +71,7 @@ class ExampleImagesRoutes:
     async def import_example_images(request):
         """Import local example images for a model"""
         return await ExampleImagesProcessor.import_images(request)
-        
+
     @staticmethod
     async def has_example_images(request):
         """Check if example images folder exists and is not empty for a model"""
@@ -71,4 +85,4 @@ class ExampleImagesRoutes:
     @staticmethod
     async def force_download_example_images(request):
         """Force download example images for specific models"""
-        return await DownloadManager.start_force_download(request)
+        return await DownloadManager.start_force_download(request)