bugfix for representation context

Author: valentin-gauthier-geosiris

energyml-utils/src/energyml/utils/data/representation_context.py

@@ -21,7 +21,7 @@ def collect_graphical_info(obj: Any, workspace: EnergymlStorageInterface) -> dic
     return collect_graphical_info_from_rels(rels, obj_uuid, workspace)
 
 def collect_graphical_info_from_rels(rels: List[Relationship], obj_uuid: str, workspace: EnergymlStorageInterface) -> dict:
-    graphical_info = {}
+    graphical_info_result = {}
     # Collect graphical information entries whose target matches this representation
     for r in rels:
         if "GraphicalInformationSet" in r.target:
@@ -39,11 +39,11 @@ def collect_graphical_info_from_rels(rels: List[Relationship], obj_uuid: str, wo
                     for target_dor in target_dors:
                         target_dor_uuid = get_obj_uuid(target_dor)
                         if target_dor_uuid == obj_uuid:
-                            if graphical_info_set_uri not in graphical_info:
-                                graphical_info[graphical_info_set_uri] = []
-                            graphical_info[graphical_info_set_uri].append(graphical_info)
+                            if graphical_info_set_uri not in graphical_info_result:
+                                graphical_info_result[graphical_info_set_uri] = []
+                            graphical_info_result[graphical_info_set_uri].append(graphical_info)
                             break
-    return graphical_info
+    return graphical_info_result
 
 
 class RepresentationContext(BaseModel):
@@ -128,7 +128,7 @@ def get_default_color(self) -> ScalarRenderingInfo:
                 except Exception as exc:
                     logging.debug(f"Error reading graphical rendering info for entry {entry}: {exc}")
         # No color information found, generate a random color from uuid
-        return ScalarRenderingInfo(constant_color=RgbaColor.random_from_uuid(self.uri.uuid))
+        return ScalarRenderingInfo(target_obj_uuid=self.uri.uuid, constant_color=RgbaColor.random_from_uuid(self.uri.uuid))
 
     def get_property(self, property_uuid: str) -> Optional[Any]:
         """Return the property object with the given uuid, or None."""