archive-extractor 0.1.2__py3-none-any.whl

archive_extractor-0.1.2.dist-info/METADATA

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: archive-extractor
+Version: 0.1.2
+Summary: 📦 Recursively extract ZIP and 7z archives from directory trees, with password-cracking support
+Author-email: Tiago Silva <eng.tiago.silva@gmail.com>
+License-File: LICENSE
+Requires-Python: >=3.12
+Requires-Dist: py7zr
+Requires-Dist: tqdm

archive_extractor-0.1.2.dist-info/RECORD

@@ -0,0 +1,6 @@
+main.py,sha256=qlje5PrHKnZyS1EIpD5aK2G26NdaCD0ZfHhX_Mk2kp4,5461
+archive_extractor-0.1.2.dist-info/METADATA,sha256=gbaY-yt-buB2MFKhlTP9QkVm0PbcbftK74azno5wzAQ,309
+archive_extractor-0.1.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+archive_extractor-0.1.2.dist-info/entry_points.txt,sha256=IuJfvHFM_0ztmmO8RnQ1ZrnVAhNTorGmpRT6CwDkuyM,48
+archive_extractor-0.1.2.dist-info/licenses/LICENSE,sha256=gTrdDdqFDu7VtWezebCYIHmebmc-XTVxqWTZGKvDux0,1068
+archive_extractor-0.1.2.dist-info/RECORD,,

archive_extractor-0.1.2.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

archive_extractor-0.1.2.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+archive-extractor = main:main

archive_extractor-0.1.2.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Tiago Silva
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

main.py

@@ -0,0 +1,131 @@
+import os
+import re
+import argparse
+from tqdm import tqdm
+import zipfile
+import py7zr
+import lzma
+
+def sanitize_filename(filename):
+    # Remove directories and illegal characters
+    filename = re.sub(r'[\\/*?:"<>|]', "_", filename)
+    filename = filename.replace("..", "")  # extra safety
+    return os.path.basename(filename)
+
+def find_archive_files(root_path):
+    """Recursively yield paths to all .zip and .7z files under root_path."""
+    for dirpath, _, filenames in os.walk(root_path):
+        for fname in filenames:
+            if fname.lower().endswith('.zip') or fname.lower().endswith('.7z'):
+                yield os.path.join(dirpath, fname)
+
+def load_passwords(password_file):
+    """Load passwords from a file, one per line, stripping whitespace."""
+    with open(password_file, 'r', encoding='utf-8') as f:
+        return [line.strip() for line in f if line.strip()]
+
+def extract_zip(zip_file, output_dir, passwords=None):
+    if not os.path.exists(output_dir):
+        os.makedirs(output_dir)
+
+    with zipfile.ZipFile(zip_file, 'r') as zf:
+        members = zf.infolist()
+        extracted = False
+        if not passwords:
+            # No passwords provided, extract directly
+            for member in tqdm(members, desc=f"Extracting {os.path.basename(zip_file)}"):
+                if member.is_dir():
+                    continue
+                safe_member_path = os.path.normpath(member.filename)
+                if os.path.isabs(safe_member_path) or safe_member_path.startswith(".."):
+                    continue
+                out_path = os.path.join(output_dir, safe_member_path)
+                out_dir = os.path.dirname(out_path)
+                if not os.path.exists(out_dir):
+                    os.makedirs(out_dir)
+                with open(out_path, 'wb') as f:
+                    f.write(zf.read(member))
+            extracted = True
+        else:
+            # Try each password for the whole zip
+            for pwd in passwords:
+                try:
+                    for member in tqdm(members, desc=f"Extracting {os.path.basename(zip_file)}", leave=False):
+                        if member.is_dir():
+                            continue
+                        safe_member_path = os.path.normpath(member.filename)
+                        if os.path.isabs(safe_member_path) or safe_member_path.startswith(".."):
+                            continue
+                        out_path = os.path.join(output_dir, safe_member_path)
+                        out_dir = os.path.dirname(out_path)
+                        if not os.path.exists(out_dir):
+                            os.makedirs(out_dir)
+                        with open(out_path, 'wb') as f:
+                            f.write(zf.read(member, pwd.encode('utf-8')))
+                    extracted = True
+                    break  # Stop trying passwords after success
+                except RuntimeError:
+                    # Wrong password, try next
+                    continue
+                except zipfile.BadZipFile:
+                    continue
+        if not extracted:
+            print(f"❌ Could not extract '{zip_file}': no valid password found.")
+        else:
+            print(f"✅ Extracted {len(members)} items to '{output_dir}'.")
+
+def extract_7z(archive_file, output_dir, passwords=None):
+    if not os.path.exists(output_dir):
+        os.makedirs(output_dir)
+
+    extracted = False
+    if not passwords:
+        try:
+            with py7zr.SevenZipFile(archive_file, mode='r') as archive:
+                archive.extractall(path=output_dir)
+            extracted = True
+        except (py7zr.exceptions.PasswordRequired, py7zr.exceptions.Bad7zFile, lzma.LZMAError):
+            pass
+        except Exception:
+            pass
+    else:
+        for pwd in passwords:
+            try:
+                with py7zr.SevenZipFile(archive_file, mode='r', password=pwd) as archive:
+                    archive.extractall(path=output_dir)
+                extracted = True
+                break
+            except (py7zr.exceptions.PasswordRequired, py7zr.exceptions.Bad7zFile, lzma.LZMAError):
+                continue
+            except Exception:
+                continue
+    if not extracted:
+        print(f"❌ Could not extract '{archive_file}': no valid password found or archive is corrupt.")
+    else:
+        print(f"✅ Extracted '{archive_file}' to '{output_dir}'.")
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Recursively extract all files from .zip and .7z archives under a given path."
+    )
+    parser.add_argument(
+        "path",
+        help="Root directory or file to search for .zip/.7z files"
+    )
+    parser.add_argument(
+        "--passwords",
+        help="Path to a file containing passwords (one per line) to try for encrypted archives"
+    )
+    args = parser.parse_args()
+    root_path = args.path
+    passwords = load_passwords(args.passwords) if args.passwords else None
+    for archive_path in find_archive_files(root_path):
+        archive_dir = os.path.splitext(archive_path)[0]
+        ext = os.path.splitext(archive_path)[1].lower()
+        if ext == ".zip":
+            extract_zip(archive_path, archive_dir, passwords)
+        elif ext == ".7z":
+            extract_7z(archive_path, archive_dir, passwords)
+
+if __name__ == "__main__":
+    main()