tristan/tools
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor zipremove.py to handle both directories and ZIP files as input

Browse source
This commit is contained in:
Tristan Smith 2024-04-05 23:57:45 -04:00
parent 7004c40071
commit 069c88fc45
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
zipremove.py
View file

@ -27,8 +27,14 @@ def process_directory(directory_path):
if __name__ == "__main__": if __name__ == "__main__":
if len(sys.argv) != 2: if len(sys.argv) != 2:
print("Usage: python script.py <directory_path>") print("Usage: python script.py <path_to_directory_or_zip>")
sys.exit(1) sys.exit(1)
directory_path = sys.argv[1] path = sys.argv[1]
process_directory(directory_path) if os.path.isdir(path):
process_directory(path)
elif os.path.isfile(path) and path.endswith('.zip'):
print(f"Processing {path}")
filter_and_repack_zip(path)
else:
print("The path is not a directory or a ZIP file.")