Fix loading torrent files

kannibalox · Oct 13, 2023 · 5d2861f · 5d2861f
1 parent d1ba384
commit 5d2861f
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/PtpUploader/Job/LoadFile.py b/src/PtpUploader/Job/LoadFile.py
@@ -12,7 +12,8 @@
 
 
 def load_json_release(path: Path):
-    data: Dict = json.load(path.open())
+    with path.open() as fh:
+        data: Dict = json.load(fh)
     release = ReleaseInfo()
     allowed_fields: List[str] = [
         "ImdbId",
@@ -36,6 +37,13 @@ def load_json_release(path: Path):
     release.save()
     path.unlink()
 
+def load_torrent_release(path: Path):
+    release = ReleaseInfo()
+    release.AnnouncementSourceName = "torrent"
+    release.SourceTorrentFilePath = path
+    release.JobRunningState = ReleaseInfo.JobState.WaitingForStart
+    release.save()
+    path.unlink()
 
 def scan_dir():
     path = Path(Settings.GetAnnouncementWatchPath())
@@ -45,7 +53,7 @@ def scan_dir():
                 load_json_release(child)
                 continue
             except json.decoder.JSONDecodeError:
-                pass
+                logger.debug("Cannot load %r as JSON, attempting .torrent check", child)
             try:
                 load_torrent_release(child)
                 continue