Fix the data update record missing for sub scenrio. (#76)

DHI · Nov 26, 2024 · a595c07 · a595c07
1 parent b4c4d9d
commit a595c07
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/mikeplus/datatableaccess.py b/mikeplus/datatableaccess.py
@@ -11,6 +11,7 @@
 from DHI.Amelia.DataModule.Services.DataTables import DataTableContainer
 from DHI.Amelia.Infrastructure.Interface.UtilityHelper import GeoAPIHelper
 from DHI.Amelia.DataModule.Interface.Services import IMuGeomTable
+from DHI.Amelia.DataModule.Services.DataTables import AmlUndoRedoManager
 
 from .dotnet import as_dotnet_list
 
@@ -72,10 +73,12 @@ def open_database(self):
         datatables.SetActiveModel(data_source.ActiveModel)
         datatables.SetEumAppUnitSystem(data_source.UnitSystemOption)
         datatables.OnResetContainer(None, None)
+        datatables.UndoRedoManager = AmlUndoRedoManager()
         self._datatables = datatables
 
     def close_database(self):
         """Close database"""
+        self._datatables.UndoRedoManager.ClearUndoRedoBuffer()
         self._datatables.DataSource.CloseDatabase()
         self._datatables.Dispose()
         self._datatables = None