add additional events

DYefremov · Aug 29, 2024 · 384c30e · 384c30e
1 parent 05cf047
commit 384c30e
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/app/ui/main.py b/app/ui/main.py
@@ -300,8 +300,18 @@ def __init__(self, **kwargs):
                            GObject.TYPE_PYOBJECT, (GObject.TYPE_PYOBJECT,))
         GObject.signal_new("bouquet-changed", self, GObject.SIGNAL_RUN_LAST,
                            GObject.TYPE_PYOBJECT, (GObject.TYPE_PYOBJECT,))
+        GObject.signal_new("bouquet-added", self, GObject.SIGNAL_RUN_LAST,
+                           GObject.TYPE_PYOBJECT, (GObject.TYPE_PYOBJECT,))
+        GObject.signal_new("bouquet-remove", self, GObject.SIGNAL_RUN_LAST,
+                           GObject.TYPE_PYOBJECT, (GObject.TYPE_PYOBJECT,))
+        GObject.signal_new("bouquet-removed", self, GObject.SIGNAL_RUN_LAST,
+                           GObject.TYPE_PYOBJECT, (GObject.TYPE_PYOBJECT,))
         GObject.signal_new("fav-changed", self, GObject.SIGNAL_RUN_LAST,
                            GObject.TYPE_PYOBJECT, (GObject.TYPE_PYOBJECT,))
+        GObject.signal_new("fav-added", self, GObject.SIGNAL_RUN_LAST,
+                           GObject.TYPE_PYOBJECT, (GObject.TYPE_PYOBJECT,))
+        GObject.signal_new("fav-removed", self, GObject.SIGNAL_RUN_LAST,
+                           GObject.TYPE_PYOBJECT, (GObject.TYPE_PYOBJECT,))
         GObject.signal_new("fav-clicked", self, GObject.SIGNAL_RUN_LAST,
                            GObject.TYPE_PYOBJECT, (GObject.TYPE_PYOBJECT,))
         GObject.signal_new("srv-clicked", self, GObject.SIGNAL_RUN_LAST,
@@ -1391,6 +1401,7 @@ def fav_paste(self, selection):
             self.update_fav_num_column(model)
 
         self._rows_buffer.clear()
+        self.emit("fav-added", self._bq_selected)
 
     def bouquet_paste(self, selection):
         model, paths = selection.get_selected_rows()
@@ -1480,9 +1491,11 @@ def remove_favs(self, itrs, model):
                     if index % self.DEL_FACTOR == 0:
                         yield True
                 self.update_fav_num_column(model)
+                self.emit("fav-removed", self._bq_selected)
 
         self.on_model_changed(self._fav_model)
         self._wait_dialog.hide()
+
         yield True
 
     def delete_services(self, itrs, model, rows, srv_model, fav_column=Column.SRV_FAV_ID):
@@ -1524,6 +1537,8 @@ def delete_bouquets(self, itrs, model):
             self.show_error_message("This item is not allowed to be removed!")
             return
 
+        self.emit("bouquet-remove", self._bouquets)
+
         for itr in itrs:
             if len(model.get_path(itr)) < 2:
                 continue
@@ -1538,6 +1553,7 @@ def delete_bouquets(self, itrs, model):
         self._bq_name_label.set_text(self._bq_selected)
         self.on_model_changed(model)
         self._wait_dialog.hide()
+        self.emit("bouquet-removed", self._bouquets)
         yield True
 
     # ***************** Bouquets ********************* #
@@ -1596,7 +1612,9 @@ def on_new_bouquet(self, view, sub=False):
                 else:
                     it = model.insert(p_itr, int(model.get_path(itr)[1]) + 1, bq) if p_itr else model.append(itr, bq)
                     scroll_to(model.get_path(it), view, paths)
+
             self._bouquets[key] = []
+            self.emit("bouquet-added", key)
 
     def on_new_sub_bouquet(self, item=None):
         self.on_new_bouquet(self._bouquets_view, True)
@@ -2089,6 +2107,8 @@ def receive_data_to_fav(self, dst_index, fav_bouquet, itrs, model, ext_model, fa
                                      ch.fav_id, self._picons.get(ch.picon_id, None), None, None))
             fav_bouquet.insert(dst_index, ch.fav_id)
 
+        self.emit("fav-added", self._bq_selected)
+
     def on_view_press(self, view, event):
         """ Handles a mouse click (press) to view. """
         if event.get_event_type() == Gdk.EventType.BUTTON_PRESS and event.button == Gdk.BUTTON_PRIMARY: