wi1-bot 1.4.4__py3-none-any.whl → 1.4.6__py3-none-any.whl

wi1_bot/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.4.4'
-__version_tuple__ = version_tuple = (1, 4, 4)
+__version__ = version = '1.4.6'
+__version_tuple__ = version_tuple = (1, 4, 6)

wi1_bot/arr/radarr.py

@@ -149,10 +149,10 @@ class Radarr:
         raise ValueError(f"no quality profile with the id {profile_id}")
 
     def rescan_movie(self, movie_id: int) -> None:
-        self._radarr.post_command("RescanMovie", movieId=movie_id)  # type: ignore
+        self._radarr.post_command("RescanMovie", movieId=movie_id)
 
     def refresh_movie(self, movie_id: int) -> None:
-        self._radarr.post_command("RefreshMovie", movieIds=[movie_id])  # type: ignore
+        self._radarr.post_command("RefreshMovie", movieIds=[movie_id])
 
     def search_missing(self) -> None:
         self._radarr.post_command(name="MissingMoviesSearch")

wi1_bot/scripts/rescan.py

@@ -0,0 +1,57 @@
+import argparse
+from time import sleep
+from typing import cast
+
+from wi1_bot.arr.radarr import Radarr
+from wi1_bot.arr.sonarr import Sonarr
+from wi1_bot.config import config
+
+
+def rescan_radarr() -> None:
+    radarr = Radarr(config["radarr"]["url"], config["radarr"]["api_key"])
+
+    all_movies = radarr._radarr.get_movie()
+    assert isinstance(all_movies, list)
+    all_movies.sort(key=lambda m: cast(str, m["title"]))
+
+    for movie in all_movies:
+        assert isinstance(movie, dict)
+        print(f"Rescanning {movie['title']}...")
+        radarr.refresh_movie(movie["id"])
+        sleep(3)
+
+
+def rescan_sonarr() -> None:
+    sonarr = Sonarr(config["sonarr"]["url"], config["sonarr"]["api_key"])
+
+    all_series = sonarr._sonarr.get_series()
+    assert isinstance(all_series, list)
+    all_series.sort(key=lambda s: cast(str, s["title"]))
+
+    for series in all_series:
+        assert isinstance(series, dict)
+        print(f"Rescanning {series['title']}...")
+        sonarr.rescan_series(series["id"])
+        sleep(5)
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(description="Rescan all movies/shows")
+
+    parser.add_argument(
+        "service", nargs="?", choices=["radarr", "sonarr"], help="radarr or sonarr"
+    )
+
+    args = parser.parse_args()
+
+    if args.service == "radarr":
+        rescan_radarr()
+    elif args.service == "sonarr":
+        rescan_sonarr()
+    else:
+        rescan_radarr()
+        rescan_sonarr()
+
+
+if __name__ == "__main__":
+    main()

wi1_bot/transcoder/transcoder.py

@@ -46,17 +46,10 @@ class Transcoder:
                 continue
 
             try:
-                self._do_transcode(item)
-                queue.remove(item)
-            except SignalInterrupt:
-                # retry when restarted
-                pass
-            except FileNotFoundError:
-                # don't retry
-                queue.remove(item)
-            except UnknownError:
-                # don't retry
-                queue.remove(item)
+                remove = self._do_transcode(item)
+
+                if remove:
+                    queue.remove(item)
             except Exception:
                 self.logger.warning(
                     "got exception when trying to transcode", exc_info=True
@@ -64,14 +57,14 @@ class Transcoder:
 
             sleep(3)
 
-    def _do_transcode(self, item: TranscodeItem) -> None:
+    def _do_transcode(self, item: TranscodeItem) -> bool:
         path = pathlib.Path(item.path)
 
-        self.logger.debug(f"attempting to transcode {path.name}")
+        self.logger.info(f"attempting to transcode {path.name}")
 
         if path.suffix == ".avi":
-            self.logger.debug(f"cannot transcode {path.name}: .avi not supported")
-            return
+            self.logger.info(f"cannot transcode {path.name}: .avi not supported")
+            return True
 
         # push.send(f"{basename}", title="starting transcode")
 
@@ -103,29 +96,34 @@ class Transcoder:
             tmp_log_path = tmp_folder / "wi1_bot.transcoder.log"
 
             with open(tmp_log_path, "w") as ffmpeg_log_file:
-                for line in proc.stdout:  # type: ignore
+                assert proc.stdout is not None
+                for line in proc.stdout:
                     ffmpeg_log_file.write(line)
                     last_output = line.strip()
 
             status = proc.wait()
 
             if status != 0:
-                self.logger.error(f"ffmpeg failed (status {status}): {last_output}")
+                transcode_to.unlink()
 
-                if (
-                    "No such file or directory" in last_output
-                    and "shared object file" not in last_output
-                ):
-                    self.logger.debug(
+                if "Error opening input files" in last_output:
+                    self.logger.info(
                         f"file does not exist: {path}, skipping transcoding"
                     )
-                    raise FileNotFoundError
+                    return True
 
                 if "received signal 15" in last_output:
-                    self.logger.debug(
+                    self.logger.info(
                         f"transcoding interrupted by signal: {path}, will retry"
                     )
-                    raise SignalInterrupt
+                    return False
+
+                if "cannot open shared object file" in last_output:
+                    self.logger.error(
+                        "ffmpeg error: missing shared object file, will retry"
+                    )
+                    push.send(f"ffmpeg error: {last_output}", title="ffmpeg error")
+                    return False
 
                 perm_log_path = tmp_folder / f"{path.stem}.log"
 
@@ -136,6 +134,8 @@ class Transcoder:
                     perm_log_path.parent.mkdir(parents=True, exist_ok=True)
 
                 shutil.copy(tmp_log_path, perm_log_path)
+
+                self.logger.error(f"ffmpeg failed (status {status}): {last_output}")
                 self.logger.error(f"log file: {perm_log_path}")
 
                 push.send(
@@ -143,9 +143,7 @@ class Transcoder:
                     title="transcoding error",
                 )
 
-                raise UnknownError
-
-        new_path = path.parent / transcode_to.name
+                return True
 
         if not path.exists():
             self.logger.debug(
@@ -153,8 +151,9 @@ class Transcoder:
             )
 
             transcode_to.unlink()
-            return
+            return True
 
+        new_path = path.parent / transcode_to.name
         shutil.move(transcode_to, new_path)
         path.unlink()
 
@@ -163,8 +162,9 @@ class Transcoder:
         self.logger.info(f"transcoded: {path.name} -> {new_path.name}")
         # push.send(f"{path.name} -> {new_path.name}", title="file transcoded")
 
+        return True
+
     def _rescan_content(self, item: TranscodeItem, new_path: str) -> None:
-        # FIXME: don't hardcode library paths (config)
         if item.content_id is not None:
             if new_path.startswith(config["radarr"]["root_folder"]):
                 self.radarr.rescan_movie(item.content_id)

{wi1_bot-1.4.4.dist-info → wi1_bot-1.4.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wi1-bot
-Version: 1.4.4
+Version: 1.4.6
 Summary: Discord bot for Radarr/Sonarr integration
 Home-page: https://github.com/wthueb/wi1-bot
 Author-email: William Huebner <wilhueb@gmail.com>

{wi1_bot-1.4.4.dist-info → wi1_bot-1.4.6.dist-info}/RECORD

@@ -1,5 +1,5 @@
 wi1_bot/__init__.py,sha256=11ozJKiUsqDCZ3_mcAHhGYUyGK_Unl54djVSBBExFB4,59
-wi1_bot/_version.py,sha256=xV-8Yj_iWn4cgetLWs0HBzWrymonMn9cAMsYPhJoE9k,411
+wi1_bot/_version.py,sha256=HFQuEZSl62eJNTO3Swl0voif7bfUlhX7pVxUC5X9W7g,411
 wi1_bot/config.py,sha256=AtzOXvCeat_lb-w_Wy37XyWIpV3LmKh38OGigluJ2nM,3139
 wi1_bot/push.py,sha256=-Az8c21R3IZAkJVRQmWsbNXMfvBzzpc9pFTWsDy1mJE,789
 wi1_bot/webhook.py,sha256=cVp6USg1vbfL05TPQSDIUUYPHWDjuPGb1IGsqImao6s,4713
@@ -7,7 +7,7 @@ wi1_bot/arr/__init__.py,sha256=ZIgkW24GBdS4sJmaiEIhueHOB6s2L8y2s9ahgp1USRI,86
 wi1_bot/arr/download.py,sha256=02AYFglnFdWSG8xj_TaJc6l2wjybyhUW6F97CnoyUFw,1381
 wi1_bot/arr/episode.py,sha256=j25ljy9hyTXFAEBeUQ4iozKP2YXpZyzauphaXa2oqh8,1057
 wi1_bot/arr/movie.py,sha256=9cpJZcoW3r4op6X8Fkr4Cv0BsiWfqKD8MkBfEpTcT8k,741
-wi1_bot/arr/radarr.py,sha256=I8_SyUlbD-bSM-71adsZOoc_-yFdsC_SGXpJvZ5IrbM,5743
+wi1_bot/arr/radarr.py,sha256=sZ2zVhSDir3T3THW2vvIgWPdsvoEsYZLualeY8UCIr4,5711
 wi1_bot/arr/sonarr.py,sha256=OouuorRHBMFAOy5oDeJZ5H6WZ-95-N01EHtpOTVMTCo,5970
 wi1_bot/discord/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wi1_bot/discord/bot.py,sha256=puBlprcGqoC-m98DDZuM7Q7zhfTHSsqJIw2ptNrE4_M,5080
@@ -17,14 +17,15 @@ wi1_bot/discord/cogs/movie.py,sha256=J-1wahJLdIk9WhgQlva4zp9AK__2Vw1M8JoPvhcwzbw
 wi1_bot/discord/cogs/series.py,sha256=AfUfWuU-vUlID-gW7GWu9w-GiwWzZ4Cxm49_FjgvJik,4837
 wi1_bot/scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wi1_bot/scripts/add_tag.py,sha256=mWwo8egk2Y5XRiQCpfkA11-3rcxZoD0JOJKxV0LguLk,586
+wi1_bot/scripts/rescan.py,sha256=2V7a3GSP8owLxdJ0J6nG98M-PqPWfRRhfJuDOgWQgnU,1488
 wi1_bot/scripts/start.py,sha256=vNa_iHkx10D5YWonyRW0f5nG8uE3_JtwJ-XZ-c0hWCs,2477
 wi1_bot/scripts/transcode_item.py,sha256=21MeeIZ9wIRhAY-FOEgOdPsg6YOLlSUj59r8NkTfixw,1155
 wi1_bot/transcoder/__init__.py,sha256=B4xr82UtIFc3tyy_MEZdZKMukYW0yejPnfsGowaTIM0,105
 wi1_bot/transcoder/transcode_queue.py,sha256=W7r2I7OD8QuxseCEb7_ddOvYXiBBLmKLvCs2IgJJ92I,1856
-wi1_bot/transcoder/transcoder.py,sha256=yK_F4xTiAlu9Tzg_DPyN-FtQkeFjlCBCOF2EB6k0HYY,8788
-wi1_bot-1.4.4.dist-info/LICENSE,sha256=6V4_mQoPoLJl77_WMsQRQMDG2mnIhDUdfCmMkqM9Qwc,1072
-wi1_bot-1.4.4.dist-info/METADATA,sha256=Q4NldKu_ZvXydXPTmV6_aheTPUR-eG-3_1vHE1JDLB4,4530
-wi1_bot-1.4.4.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-wi1_bot-1.4.4.dist-info/entry_points.txt,sha256=G_NEW2uMHHw4pbmVwGZ6cZaBwGpdkv741jNYtuX-1vM,147
-wi1_bot-1.4.4.dist-info/top_level.txt,sha256=Q7mTnPLk80Td82YbjlBMO5tvXJoTFHhheHdmwc-c-7I,8
-wi1_bot-1.4.4.dist-info/RECORD,,
+wi1_bot/transcoder/transcoder.py,sha256=g82E_X37tG2GStbMPA9cgPpZ43kpZKi4LpR7_iKdc78,8767
+wi1_bot-1.4.6.dist-info/LICENSE,sha256=6V4_mQoPoLJl77_WMsQRQMDG2mnIhDUdfCmMkqM9Qwc,1072
+wi1_bot-1.4.6.dist-info/METADATA,sha256=laH8OH_m7-0OAAcDJHVGhTakuTNAo-9vgd_6ADNrhQ0,4530
+wi1_bot-1.4.6.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+wi1_bot-1.4.6.dist-info/entry_points.txt,sha256=pF5EawAQsUNMHs5exynpNdRq9g4dODg-RaPkx2MyYLU,184
+wi1_bot-1.4.6.dist-info/top_level.txt,sha256=Q7mTnPLk80Td82YbjlBMO5tvXJoTFHhheHdmwc-c-7I,8
+wi1_bot-1.4.6.dist-info/RECORD,,

{wi1_bot-1.4.4.dist-info → wi1_bot-1.4.6.dist-info}/entry_points.txt

@@ -1,4 +1,5 @@
 [console_scripts]
 add-tag = wi1_bot.scripts.add_tag:main
+rescan = wi1_bot.scripts.rescan:main
 transcode-item = wi1_bot.scripts.transcode_item:main
 wi1-bot = wi1_bot.scripts.start:main