Refactor DuckDBStore to handle JSON data and improve service retrieval logic

Author: emmanuelmathot

titiler/openeo/services/duckdb.py

@@ -1,5 +1,6 @@
 """titiler.openeo.services duckDB."""
 
+import json
 import uuid
 from typing import Any, Dict, List, Optional
 
@@ -9,7 +10,7 @@
 from .base import ServicesStore
 
 
-@define(kw_only=True, init=False)
+@define(kw_only=True)
 class DuckDBStore(ServicesStore):
     """DuckDB Service Store."""
 
@@ -45,23 +46,23 @@ def get_service(self, service_id: str) -> Optional[Dict]:
 
             return {
                 "id": result[0],
-                **result[1],
+                **json.loads(result[1]),
             }
 
     def get_services(self, **kwargs) -> List[Dict]:
         """Return All Services."""
         with duckdb.connect(self.store) as con:
             results = con.execute(
                 """
-                SELECT service_id, service
+                SELECT service_id, json(service)
                 FROM services
                 """
             ).fetchall()
 
             return [
                 {
                     "id": result[0],
-                    **result[1],
+                    **json.loads(result[1]),
                 }
                 for result in results
             ]
@@ -71,20 +72,17 @@ def get_user_services(self, user_id: str, **kwargs) -> List[Dict]:
         with duckdb.connect(self.store) as con:
             results = con.execute(
                 """
-                SELECT service_id, service
+                SELECT service_id, service::JSON
                 FROM services
                 WHERE user_id = ?
                 """,
                 [user_id],
             ).fetchall()
 
-            if not results:
-                raise ValueError(f"Could not find service for user: {user_id}")
-
             return [
                 {
                     "id": result[0],
-                    **result[1],
+                    **json.loads(result[1]),
                 }
                 for result in results
             ]