address review

truenas · Dec 31, 2024 · 6613e43 · 6613e43
1 parent 0420b93
commit 6613e43
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/middlewared/middlewared/plugins/failover_/zpool_cachefile.py b/src/middlewared/middlewared/plugins/failover_/zpool_cachefile.py
@@ -4,6 +4,7 @@
 # See the file LICENSE.IX for complete terms and conditions
 
 from pathlib import Path
+from typing import Literal
 
 from middlewared.service import Service
 from middlewared.plugins.pool_.utils import ZPOOL_CACHE_FILE
@@ -17,7 +18,10 @@ class Config:
         private = True
         namespace = 'failover.zpool.cachefile'
 
-    def setup(self, event: str = "MASTER"):
+    def setup(
+        self,
+        event: Literal["MASTER", "BACKUP", "SYNC"] = "MASTER"
+    ):
         saved = Path(ZPOOL_CACHE_FILE_SAVED)
         default = Path(ZPOOL_CACHE_FILE)
         overwrite = Path(ZPOOL_CACHE_FILE_OVERWRITE)