add popups with id so they can be updated /removed

1 files changed, 36 insertions, 3 deletions

diff --git a/lib/python/Tools/Notifications.py b/lib/python/Tools/Notifications.py
index 3cb107cc..85452e11 100644
--- a/lib/python/Tools/Notifications.py
+++ b/lib/python/Tools/Notifications.py
@@ -3,10 +3,43 @@ notifications = [ ]
 
 notificationAdded = [ ]
 
+
+# notifications which are currently on screen (and might be closed by similiar notifications)
+current_notifications = [ ]
+
+def __AddNotification(fnc, screen, id, *args, **kwargs):
+	notifications.append((fnc, screen, args, kwargs, id))
+	for x in notificationAdded:
+		x()
+
 def AddNotification(screen, *args, **kwargs):
 	AddNotificationWithCallback(None, screen, *args, **kwargs)
 
 def AddNotificationWithCallback(fnc, screen, *args, **kwargs):
-	notifications.append((fnc, screen, args, kwargs))
-	for x in notificationAdded:
-		x()
+	__AddNotification(fnc, screen, None, *args, **kwargs)
+
+def AddNotificationWithID(id, screen, *args, **kwargs):
+	__AddNotification(None, screen, id, *args, **kwargs)
+
+# we don't support notifications with callback and ID as this
+# would require manually calling the callback on cancelled popups.
+
+def RemovePopup(id):
+	# remove similiar notifications
+	print "RemovePopup, id =", id
+	for x in notifications:
+		if x[4] and x[4] == id:
+			print "(found in notifications)"
+			notifications.remove(x)
+
+	for x in current_notifications:
+		if x[0] == id:
+			print "(found in current notifications)"
+			x[1].close()
+
+def AddPopup(text, type, timeout, id = None):
+	if id is not None:
+		RemovePopup(id)
+	from Screens.MessageBox import MessageBox
+	print "AddPopup, id =", id
+	AddNotificationWithID(id, MessageBox, text = text, type = type, timeout = timeout)