diff --git a/src/aare/gui/models/sample_queue_model.py b/src/aare/gui/models/sample_queue_model.py index 2721e29f..481a7a61 100644 --- a/src/aare/gui/models/sample_queue_model.py +++ b/src/aare/gui/models/sample_queue_model.py @@ -116,4 +116,19 @@ class SampleQueueSpreadsheet(QAbstractTableModel): def clearSamples(self): self.beginResetModel() self.samples = [] + self.endResetModel() + + def to_state(self) -> dict: + return { + "samples": [s.to_dict() for s in self.samples], + } + + def from_state(self, state: dict): + self.beginResetModel() + + self.samples = [ + SampleShortInfo.from_dict(d) + for d in state.get("samples", []) + ] + self.endResetModel() \ No newline at end of file