add EPG warning messages

DYefremov · Feb 15, 2024 · 274abec · 274abec
1 parent 77a5f55
commit 274abec
Showing 1 changed file with 23 additions and 7 deletions.
diff --git a/app/ui/epg/epg.py b/app/ui/epg/epg.py
@@ -119,10 +119,13 @@ def update_epg_data(self) -> bool: pass
     def get_current_event(self, service_name) -> EpgEvent: pass
 
     @abc.abstractmethod
-    def get_current_events(self, service_name) -> list[EpgEvent]: pass
+    def get_current_events(self, service_name) -> list: pass
 
     @staticmethod
     def get_gz_file_name(url, path):
+        if not url:
+            return f"{path}epg{os.sep}epg.gz"
+
         f_sha1 = sha1(url.encode("utf-8", errors="ignore")).hexdigest()
         return f"{path}epg{os.sep}{f_sha1}_epg.gz"
 
@@ -169,12 +172,20 @@ def process():
                 else:
                     process_data()
             else:
-                if not self._canceled:
-                    task = BGTaskWidget(self._app, "Downloading EPG...", self._reader.download, process_data, )
-                    self._app.emit("add-background-task", task)
+                if not url:
+                    self._is_run = True
+                    self._app.show_info_message("The EPG source for the favorites list is not set!",
+                                                Gtk.MessageType.WARNING)
+                    GLib.idle_add(self._app.emit, "epg-cache-initialized", self)
+                else:
+                    if not self._canceled:
+                        task = BGTaskWidget(self._app, "Downloading EPG...", self._reader.download, process_data, )
+                        self._app.emit("add-background-task", task)
         elif self._src is EpgSource.DAT:
             self._reader = EPG.DatReader(f"{self._settings.profile_data_path}epg{os.sep}epg.dat")
             self._reader.download()
+        else:
+            GLib.idle_add(self._app.emit, "epg-cache-initialized", self)
 
     def reset(self) -> None:
         self.events.clear()
@@ -259,8 +270,13 @@ def load_data(self):
             else:
                 self._task = BGTaskWidget(self._app, "", self.process_data, )
         else:
-            self._task = BGTaskWidget(self._app, "Downloading EPG...", self._reader.download, self.process_data, )
-            self._app.emit("add-background-task", self._task)
+            if not self._xml_src:
+                self._app.emit("epg-cache-initialized", self)
+                self._app.show_info_message("Select a local file to load EPG data or add a web source.",
+                                            Gtk.MessageType.WARNING)
+            else:
+                self._task = BGTaskWidget(self._app, "Downloading EPG...", self._reader.download, self.process_data, )
+                self._app.emit("add-background-task", self._task)
 
     def on_cache_initialized(self, app, cache):
         if isinstance(cache, FavEpgCache):
@@ -306,7 +322,7 @@ def update_epg_data(self) -> bool:
     def get_current_event(self, service_name) -> EpgEvent:
         pass
 
-    def get_current_events(self, service_name) -> list[EpgEvent]:
+    def get_current_events(self, service_name) -> list:
         return self.events.get(service_name, [])