#1859 - First pass at an implementation of the full reset method. Will now start testing...

src/primaite/simulator/core.py

@@ -153,6 +153,8 @@ class SimComponent(BaseModel):
     uuid: str
     """The component UUID."""
 
+    _original_state: Dict = {}
+
     def __init__(self, **kwargs):
         if not kwargs.get("uuid"):
             kwargs["uuid"] = str(uuid4())
@@ -160,6 +162,16 @@ class SimComponent(BaseModel):
         self._request_manager: RequestManager = self._init_request_manager()
         self._parent: Optional["SimComponent"] = None
 
+    # @abstractmethod
+    def set_original_state(self):
+        """Sets the original state."""
+        pass
+
+    def reset_component_for_episode(self, episode: int):
+        """Reset the original state of the SimComponent."""
+        for key, value in self._original_state.items():
+            self.__setattr__(key, value)
+
     def _init_request_manager(self) -> RequestManager:
         """
         Initialise the request manager for this component.
@@ -227,14 +239,6 @@ class SimComponent(BaseModel):
         """
         pass
 
-    def reset_component_for_episode(self, episode: int):
-        """
-        Reset this component to its original state for a new episode.
-
-        Override this method with anything that needs to happen within the component for it to be reset.
-        """
-        pass
-
     @property
     def parent(self) -> "SimComponent":
         """Reference to the parent object which manages this object.