fix: register list & dict value data

DHARPA-Project · Apr 17, 2024 · 17261e7 · 17261e7
1 parent 61ec4f3
commit 17261e7
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 4 deletions.
diff --git a/src/kiara/data_types/included_core_types/__init__.py b/src/kiara/data_types/included_core_types/__init__.py
@@ -403,7 +403,14 @@ def parse_python_obj(self, data: Any) -> KiaraDict:
                 dict_data = orjson.loads(data)
                 schema = {"title": "dict", "type": "object"}
             except Exception:
-                pass
+                try:
+                    from kiara.utils.cli import dict_from_cli_args
+
+                    tokens = data.split(",")
+                    dict_data = dict_from_cli_args(*tokens)
+                    schema = {"title": "dict", "type": "object"}
+                except Exception:
+                    pass
 
         if dict_data is None or schema is None:
             raise Exception(f"Invalid data for value type 'dict': {data}")

diff --git a/src/kiara/registries/data/__init__.py b/src/kiara/registries/data/__init__.py
@@ -1014,6 +1014,7 @@ def _create_value(
 
         data_type: Union[None, DataType] = None
         if reuse_existing and not isinstance(data, (Value, uuid.UUID, SpecialValue)):
+
             data_type = self._kiara.type_registry.retrieve_data_type(
                 data_type_name=schema.type, data_type_config=schema.type_config
             )

diff --git a/src/kiara/registries/data/data_store/sqlite_store.py b/src/kiara/registries/data/data_store/sqlite_store.py
@@ -309,10 +309,13 @@ def _find_values_with_hash(
             raise NotImplementedError()
 
         sql = text(
-            f"SELECT value_id FROM {TABLE_NAME_DATA_METADATA} WHERE value_hash = ?"
+            f"SELECT value_id FROM {TABLE_NAME_DATA_METADATA} WHERE value_hash = :value_hash"
         )
+        params = {
+            "value_hash": value_hash,
+        }
         with self.sqlite_engine.connect() as conn:
-            cursor = conn.execute(sql, (value_hash,))
+            cursor = conn.execute(sql, parameters=params)
             result = cursor.fetchall()
             return {uuid.UUID(x[0]) for x in result}
 

diff --git a/src/kiara/registries/jobs/__init__.py b/src/kiara/registries/jobs/__init__.py
@@ -387,7 +387,11 @@ def get_job_record(self, job_id: uuid.UUID) -> JobRecord:
             if r.job_id == job_id:
                 return r
 
-        raise NotImplementedError()
+        # raise a FailedJobException if the job is in the failed jobs list
+        self.get_active_job(job_id=job_id)
+
+        # this should never happen
+        raise KiaraException("Can't find job record with id: {job_id}")
 
     def find_job_records(self, matcher: JobMatcher) -> Mapping[uuid.UUID, JobRecord]: