Now works with multiple files as arguments

This commit is contained in:
admin 2024-05-19 13:33:16 -04:00
parent 545279d6c6
commit 79351568a7

View File

@ -45,25 +45,27 @@ def clean_directory_name(directory_name):
return directory_name.rsplit("[", 1)[0] return directory_name.rsplit("[", 1)[0]
def main(): def main():
if len(sys.argv) != 2: if len(sys.argv) < 2:
print("Usage: python script.py <folder_path>") print("Usage: python script.py <folder_path>")
sys.exit(1) sys.exit(1)
directory_name = sys.argv[1] args = sys.argv[1:]
suffix = count_folder_files(directory_name) for arg in args:
# If [num] already exists at the end of the string, remove it so we can update it directory_name = arg
directory_name_cleaned = clean_directory_name(directory_name) 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): if check_directory_exists(directory_name):
print(f"Directory '{directory_name}' exists.") print(f"Directory '{directory_name}' exists.")
#If folder was counted already, remove the old count. Ex: dirname[1] -> dirname #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)) os.rename(directory_name, os.path.join(os.path.dirname(directory_name), directory_name_cleaned))
directory_name = directory_name_cleaned directory_name = directory_name_cleaned
append_to_folder_name(directory_name, suffix) append_to_folder_name(directory_name, suffix)
else: else:
print(f"Directory '{directory_name}' does not exist.") print(f"Directory '{directory_name}' does not exist.")
if __name__ == "__main__": if __name__ == "__main__":
main() main()