From 79351568a7a384ba69fec01ef9f8c63f5842738d Mon Sep 17 00:00:00 2001 From: admin Date: Sun, 19 May 2024 13:33:16 -0400 Subject: [PATCH] Now works with multiple files as arguments --- CountFolderFiles.py | 28 +++++++++++++++------------- 1 file changed, 15 insertions(+), 13 deletions(-) diff --git a/CountFolderFiles.py b/CountFolderFiles.py index b9420ef..3cc2219 100644 --- a/CountFolderFiles.py +++ b/CountFolderFiles.py @@ -45,25 +45,27 @@ def clean_directory_name(directory_name): return directory_name.rsplit("[", 1)[0] def main(): - if len(sys.argv) != 2: + if len(sys.argv) < 2: print("Usage: python script.py ") sys.exit(1) - directory_name = sys.argv[1] - suffix = count_folder_files(directory_name) - # If [num] already exists at the end of the string, remove it so we can update it - directory_name_cleaned = clean_directory_name(directory_name) + args = sys.argv[1:] + for arg in args: + directory_name = arg + suffix = count_folder_files(directory_name) + # If [num] already exists at the end of the string, remove it so we can update it + directory_name_cleaned = clean_directory_name(directory_name) - if check_directory_exists(directory_name): - print(f"Directory '{directory_name}' exists.") + if check_directory_exists(directory_name): + print(f"Directory '{directory_name}' exists.") - #If folder was counted already, remove the old count. Ex: dirname[1] -> dirname - os.rename(directory_name, os.path.join(os.path.dirname(directory_name), directory_name_cleaned)) - directory_name = directory_name_cleaned + #If folder was counted already, remove the old count. Ex: dirname[1] -> dirname + os.rename(directory_name, os.path.join(os.path.dirname(directory_name), directory_name_cleaned)) + directory_name = directory_name_cleaned - append_to_folder_name(directory_name, suffix) - else: - print(f"Directory '{directory_name}' does not exist.") + append_to_folder_name(directory_name, suffix) + else: + print(f"Directory '{directory_name}' does not exist.") if __name__ == "__main__": main()