From 531dcaea2012e93c3ca59d8144ad94afa96d2d6b Mon Sep 17 00:00:00 2001
From: sonicaj <waqarahmedjoyia@live.com>
Date: Thu, 10 Jul 2025 21:51:38 +0500
Subject: [PATCH] Send a final event to notify that upgrade/installation
 completed (#886)

---
 truenas_install/__main__.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/truenas_install/__main__.py b/truenas_install/__main__.py
index 06b1bec..7a47e00 100644
--- a/truenas_install/__main__.py
+++ b/truenas_install/__main__.py
@@ -256,6 +256,7 @@ def main():
     input = json.loads(sys.stdin.read())
 
     old_root = input.get("old_root", None)
+    is_fresh_install = old_root is None
 
     if input.get("precheck"):
         if precheck_result := precheck(old_root):
@@ -672,6 +673,7 @@ def main():
         raise
 
     configure_system_for_zectl(pool_name)
+    write_progress(1.0, f"{'Installation' if is_fresh_install else 'Upgrade'} completed successfully")
 
 
 if __name__ == "__main__":