Scrape torrent ids via HTML if not already populated

kannibalox · Sep 21, 2024 · 2a685dd · 2a685dd
1 parent 506eaf3
commit 2a685dd
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/ptpapi/movie.py b/src/ptpapi/movie.py
@@ -151,6 +151,10 @@ def load_html_data(self):
             self.data["Snatched"] = True
 
         # File list & trumpability for torrents
+        # Populate torrent IDs if not already populated
+        if "Torrents" not in self.data.keys():
+            self.data["Torrents"] = [torrent.Torrent(
+                re.match(r"torrent_(\d*)", torr['id']).group(1)) for torr in soup.find_all("tr", class_="torrent_info_row")]
         for tor in self.data["Torrents"]:
             # Get file list
             filediv = soup.find("div", id="files_%s" % tor.ID)