tools/zipremove.py
2024-04-05 23:28:55 -04:00

34 lines
1.1 KiB
Python

import zipfile
import os
import sys
import tempfile
from pathlib import Path
import shutil
def filter_and_repack_zip(zip_path):
temp_dir = tempfile.TemporaryDirectory()
new_zip_path = os.path.join(temp_dir.name, 'new_archive.zip')
with zipfile.ZipFile(zip_path, 'r') as original_zip:
with zipfile.ZipFile(new_zip_path, 'w') as new_zip:
for file in original_zip.namelist():
if "(USA" in file or "(USA, Europe)" in file:
new_zip.writestr(file, original_zip.read(file))
# Remove the original file and move the new file to the original location
os.remove(zip_path)
shutil.move(new_zip_path, zip_path)
def process_directory(directory_path):
for zip_file in Path(directory_path).glob('*.zip'):
print(f"Processing {zip_file}")
filter_and_repack_zip(zip_file)
print("All ZIP files processed.")
if __name__ == "__main__":
if len(sys.argv) != 2:
print("Usage: python script.py <directory_path>")
sys.exit(1)
directory_path = sys.argv[1]
process_directory(directory_path)