From 2689dd180ef4ea28a5de9c10f5cc6c46c0ed501e Mon Sep 17 00:00:00 2001
From: Kristofers Solo <kristians.cagulis@pm.me>
Date: Wed, 8 Nov 2023 13:02:06 +0200
Subject: [PATCH] feat(gui): update `update` method

---
 src/ui/app.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/ui/app.py b/src/ui/app.py
index ee997b5..fd83e95 100644
--- a/src/ui/app.py
+++ b/src/ui/app.py
@@ -64,6 +64,9 @@ class App(ctk.CTk):
         submit_button = ctk.CTkButton(popup, text="Submit", command=submit)
         submit_button.grid(row=5, column=0, padx=self.PADX, pady=self.PADY)
 
-    def update(self):
-        self.data = self.inventory.list_all()
+    def update(self, data=None) -> None:
+        if data:
+            self.data = data
+        else:
+            self.data = self.inventory.list_all()
         self.display_table()