Now works with multiple files as arguments
This commit is contained in:
parent
545279d6c6
commit
79351568a7
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user