diff --git a/truenas_installer/__main__.py b/truenas_installer/__main__.py index 8dba5d9..1874451 100644 --- 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 index 1060b11..7100e42 100644 --- 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." ),