Workaround for quick shutdown, fixed ongoing season logic

rivenmedia · Dec 27, 2023 · a141777 · a141777
1 parent a877fd9
commit a141777
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 5 deletions.
diff --git a/backend/program/media/item.py b/backend/program/media/item.py
@@ -201,7 +201,7 @@ def _determine_state(self):
                 return Symlink()
             if all(episode.file and episode.folder for episode in self.episodes):
                 return Download()
-            if self.is_scraped():
+            if self.is_scraped() or any(episode.state == Scrape for episode in self.episodes):
                 return Scrape()
             if any(episode.state == Content for episode in self.episodes):
                 return Content()

diff --git a/backend/program/media/state.py b/backend/program/media/state.py
@@ -34,7 +34,11 @@ def perform_action(self):
             scrape.run(self.context)
         if self.context.type == "show":
             for season in self.context.seasons:
-                season.state.perform_action()
+                if season.aired_at:
+                    season.state.perform_action()
+                else:
+                    for episode in season.episodes:
+                        episode.state.perform_action()
 
 
 class Scrape(MediaItemState):
@@ -46,7 +50,11 @@ def perform_action(self):
             debrid.run(self.context)
         if self.context.type == "show":
             for season in self.context.seasons:
-                season.state.perform_action()
+                if season.aired_at:
+                    season.state.perform_action()
+                else:
+                    for episode in season.episodes:
+                        episode.state.perform_action()
         if self.context.type == "season":
             self.context.state.perform_action()
 

diff --git a/backend/program/realdebrid.py b/backend/program/realdebrid.py
@@ -83,7 +83,7 @@ def _is_downloaded(self, item):
 
     def _update_torrent_info(self, item):
         info = self.get_torrent_info(item.get("active_stream")["id"])
-        item.active_stream["name"] = info.filename
+        item.active_stream["name"] = info.original_filename
 
     def _download_item(self, item):
         request_id = self.add_magnet(item)

diff --git a/backend/utils/utils.py b/backend/utils/utils.py
@@ -33,7 +33,11 @@ def save(self) -> None:
     def run(self):
         while self.running:
             self.save()
-            time.sleep(10)
+            # workaround for quick shutdown, we should use threading.Event instead
+            for i in range(10):
+                if not self.running:
+                    break
+                time.sleep(i)
 
 
 class Parser: