oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Database patch

Browse Source
This commit is contained in:
Marek Wolan
2024-01-11 15:40:37 +00:00
parent ed5591caf8
commit 842e59f596
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/primaite/simulator/system/services/database/database_service.py
View File

@@ -247,3 +247,8 @@ class DatabaseService(Service):
if timestep == 1: if timestep == 1:
self.backup_database() self.backup_database()
return super().apply_timestep(timestep) return super().apply_timestep(timestep)
def _update_patch_status(self) -> None:
super()._update_patch_status()
if self._patching_countdown is None:
self.restore_backup()

5
src/primaite/simulator/system/software.py
View File

@@ -193,8 +193,9 @@ class Software(SimComponent):
def patch(self) -> None: def patch(self) -> None:
"""Perform a patch on the software.""" """Perform a patch on the software."""
self._patching_countdown = self.patching_duration if self.health_state_actual in (SoftwareHealthState.COMPROMISED, SoftwareHealthState.GOOD):
self.set_health_state(SoftwareHealthState.PATCHING) self._patching_countdown = self.patching_duration
self.set_health_state(SoftwareHealthState.PATCHING)
def _update_patch_status(self) -> None: def _update_patch_status(self) -> None:
"""Update the patch status of the software.""" """Update the patch status of the software."""