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]
def main():
if len(sys.argv) != 2:
if len(sys.argv) < 2:
print("Usage: python script.py <folder_path>")
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()