Improve fixups with Job condition checks (#2451)

This commit is contained in:
Pascal Vizeli 2021-01-21 16:48:25 +01:00 committed by GitHub
parent 1a978f4762
commit dccfffd979
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 17 additions and 3 deletions

View File

@ -3,6 +3,8 @@ import logging
from typing import List
from ..coresys import CoreSys, CoreSysAttributes
from ..jobs.const import JobCondition
from ..jobs.decorator import Job
from .data import Suggestion
from .fixups.base import FixupBase
from .fixups.clear_full_snapshot import FixupClearFullSnapshot
@ -41,6 +43,7 @@ class ResolutionFixup(CoreSysAttributes):
self._store_execute_remove,
]
@Job(conditions=[JobCondition.HEALTHY])
async def run_autofix(self) -> None:
"""Run all startup fixes."""
_LOGGER.info("Starting system autofix at state %s", self.sys_core.state)

View File

@ -2,8 +2,14 @@
import logging
from typing import List, Optional
from supervisor.exceptions import ResolutionFixupError, StoreError, StoreNotFound
from ...exceptions import (
ResolutionFixupError,
ResolutionFixupJobError,
StoreError,
StoreNotFound,
)
from ...jobs.const import JobCondition
from ...jobs.decorator import Job
from ..const import ContextType, IssueType, SuggestionType
from .base import FixupBase
@ -13,6 +19,10 @@ _LOGGER: logging.Logger = logging.getLogger(__name__)
class FixupStoreExecuteReload(FixupBase):
"""Storage class for fixup."""
@Job(
conditions=[JobCondition.INTERNET_SYSTEM, JobCondition.FREE_SPACE],
on_condition=ResolutionFixupJobError,
)
async def process_fixup(self, reference: Optional[str] = None) -> None:
"""Initialize the fixup class."""
_LOGGER.info("Reload Store: %s", reference)

View File

@ -21,7 +21,8 @@ class FixupStoreExecuteReset(FixupBase):
"""Storage class for fixup."""
@Job(
conditions=[JobCondition.INTERNET_SYSTEM], on_condition=ResolutionFixupJobError
conditions=[JobCondition.INTERNET_SYSTEM, JobCondition.FREE_SPACE],
on_condition=ResolutionFixupJobError,
)
async def process_fixup(self, reference: Optional[str] = None) -> None:
"""Initialize the fixup class."""