enchancement

Samsung · Jul 17, 2024 · 0402379 · 0402379
1 parent 722757e
commit 0402379
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/credsweeper/deep_scanner/bzip2_scanner.py b/credsweeper/deep_scanner/bzip2_scanner.py
@@ -24,9 +24,11 @@ def data_scan(
         candidates = []
         try:
             file_path = Path(data_provider.file_path)
-            new_path = file_path.as_posix()[:-4] if ".bz2" == file_path.suffix else file_path.as_posix()
+            new_path = file_path.as_posix()
+            if ".bz2" == file_path.suffix:
+                new_path = new_path[:-4]
             bzip2_content_provider = DataContentProvider(data=bz2.decompress(data_provider.data),
-                                                         file_path=file_path.as_posix(),
+                                                         file_path=new_path,
                                                          file_type=Util.get_extension(new_path),
                                                          info=f"{data_provider.info}|BZIP2|{new_path}")
             new_limit = recursive_limit_size - len(bzip2_content_provider.data)

diff --git a/credsweeper/deep_scanner/gzip_scanner.py b/credsweeper/deep_scanner/gzip_scanner.py
@@ -26,9 +26,11 @@ def data_scan(
         try:
             with gzip.open(io.BytesIO(data_provider.data)) as f:
                 file_path = Path(data_provider.file_path)
-                new_path = file_path.as_posix()[:-3] if ".gz" == file_path.suffix else file_path.as_posix()
+                new_path = file_path.as_posix()
+                if ".gz" == file_path.suffix:
+                    new_path = new_path[:-3]
                 gzip_content_provider = DataContentProvider(data=f.read(),
-                                                            file_path=file_path.as_posix(),
+                                                            file_path=new_path,
                                                             file_type=Util.get_extension(new_path),
                                                             info=f"{data_provider.info}|GZIP|{new_path}")
                 new_limit = recursive_limit_size - len(gzip_content_provider.data)