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]
|
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()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user