Flake fixes

truenas · Jul 12, 2024 · cc86cd3 · cc86cd3
1 parent bb10b0f
commit cc86cd3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/truenas_installer/__main__.py b/truenas_installer/__main__.py
@@ -20,13 +20,13 @@ def main():
 
     with open("/etc/version") as f:
         version = f.read().strip()
-    
+
     try:
         with open("/data/.vendor") as f:
             vendor = json.loads(f.read()).get("name", "TrueNAS")
     except FileNotFoundError:
         vendor = "TrueNAS"
-        
+
     dmi = parse_dmi()
 
     installer = Installer(version, dmi, vendor)

diff --git a/truenas_installer/installer_menu.py b/truenas_installer/installer_menu.py
@@ -36,6 +36,7 @@ async def _install_upgrade(self):
 
     async def _install_upgrade_internal(self):
         disks = await list_disks()
+        vendor = self.installer.vendor
 
         if not disks:
             await dialog_msgbox("Choose Destination Media", "No drives available")
@@ -45,7 +46,7 @@ async def _install_upgrade_internal(self):
             destination_disks = await dialog_checklist(
                 "Choose Destination Media",
                 (
-                    f"Install {self.installer.vendor} to a drive. If desired, select multiple drives to provide redundancy. {self.installer.vendor} "
+                    f"Install {vendor} to a drive. If desired, select multiple drives to provide redundancy. {vendor} "
                     "installation drive(s) are not available for use in storage pools. Use arrow keys to navigate "
                     "options. Press spacebar to select."
                 ),