tools/zipremove.py

32 lines
990 B
Python
Raw Normal View History

2024-04-06 03:57:52 +01:00
import zipfile
import os
import sys
import tempfile
from pathlib import Path
2024-04-06 03:57:52 +01:00
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))
2024-04-06 03:57:52 +01:00
os.replace(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.")
2024-04-06 03:57:52 +01:00
if __name__ == "__main__":
if len(sys.argv) != 2:
print("Usage: python script.py <directory_path>")
2024-04-06 03:57:52 +01:00
sys.exit(1)
directory_path = sys.argv[1]
process_directory(directory_path)