Merge pull request #28 from 31December99/work

Duplicate add info_hash
31December99 · Aug 20, 2024 · 6143b64 · 6143b64
2 parents 6c53b30 + dc74dc1
commit 6143b64
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 13 deletions.
diff --git a/README.md b/README.md
@@ -16,19 +16,11 @@ It performs the following tasks:
 - Seeding in qbittorrent
 
 ___
-### User preferences.cfg:
-
-duplicate_on is used to activate a check for titles already present in the tracker
-
-number_of_screenshots determines the number of screenshots generated
-
-torrent_archive set an optional path to move the .torrent file to a desired folder
-
 ## Join Us
 
 The program is functional but still in the early stages, undergoing constant modifications.
 You can help improve this project or just hang out with us on
-[Discord](https://discord.gg/CJ32Vwyu)
+[forum](https://itatorrents.xyz/forums/topics/414?page=1#post-1497) 
 
 Thank you
 

diff --git a/unit3dup/duplicate.py b/unit3dup/duplicate.py
@@ -140,6 +140,7 @@ def _view_data(self, data: dict) -> bool:
             if "attributes" in key:
                 name = value["name"]
                 poster = value["meta"]["poster"]
+                info_hash = value["info_hash"]
 
                 # Size in GB
                 size = round(value["size"] / (1024**3), 2)
@@ -154,14 +155,17 @@ def _view_data(self, data: dict) -> bool:
                 size_width = 6
                 name_width = 20
                 poster_width = 20
+                info_hash_width = 20
                 formatted_tmdb_id = f"{tmdb_id:>{tmdb_id_width}}"
                 formatted_size = f"{size:>{size_width}.2f} GB"
                 formatted_name = f"{name:<{name_width}}"
                 formatted_poster = f"{poster:<{poster_width}}"
+                formatted_info_hash = f"{info_hash:<{info_hash_width}}"
 
                 if already:
                     console.log(
-                        f"[TMDB-ID {formatted_tmdb_id}] [{formatted_size}] '{formatted_name}' '{formatted_poster}'"
+                        f"[TMDB-ID {formatted_tmdb_id}] [{formatted_size}] '[HASH {formatted_info_hash}]' "
+                        f"{formatted_name}' '{formatted_poster}'"
                     )
 
                     self.flag_already = True

diff --git a/unit3dup/media.py b/unit3dup/media.py
@@ -55,12 +55,12 @@ def process(self, mode="man"):
             console.rule(content.file_name)
 
             if (
-                content.category == self.movie_category
-                or content.category == self.serie_category
+                    content.category == self.movie_category
+                    or content.category == self.serie_category
             ):
                 video_manager = VideoManager(content=content)
 
-                if config.DUPLICATE:
+                if config.DUPLICATE == 'True':
                     results = video_manager.check_duplicate()
                     if results:
                         console.log(