RenameFiles/RenameFiles.py at master · b0bdN/RenameFiles · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
# -*-coding:utf-8 -*

import os
import sys

# Initialize for another try
retry = True

# List of all files in the folder selected
def file_list(fpath):
    file_names = []
    file_names = os.listdir(fpath)
    for f in file_names:
        if os.path.isfile(os.path.join(fpath, f)):
            yield f

def rename_prefix(fpath):
    pre = input("What is the prefix to add? ")
    pre = str(pre).lstrip()
    for f in file_list(fpath):
        src = fpath + "/" + f
        dst = fpath + "/" + pre + f
        os.rename(src, dst)

def rename_suffix(fpath):
    suf = input("What is the suffix to add? ")
    suf = str(suf)
    for f in file_list(fpath):
        f_name, f_extension = os.path.splitext(f)
        src = fpath + "/" + f
        dst = fpath + "/" + f_name + suf + f_extension
        os.rename(src, dst)

# While retry = True
while retry:
    # Selection of the folder
    folder_path = input("Folder path: ")
    # print(folder_path)

    test_folder = True
    i = 0
    while test_folder:
        try:
            folder = os.listdir(folder_path)
            test_folder = False
        except FileNotFoundError:
            print("We cannot find the folder in the path specified.")
            folder_path = input("Folder path: ")
            i += 1
            if i == 2:
                sys.exit()


    # Print the list of all the files in the folder selected
    print("Here the list of all the files in the folder selected: ")
    for x in file_list(folder_path):
        print(x)

    question = input("Do you want to add a prefix (p), a suffix (s), or both (b)? - Select Q to quit.\nChoose between p, s or b: ")
    question = str(question)
    question = question.lower()
    question_bool = True

    i = 0
    while question_bool:
        if question == "p" or question == "s" or question == "b" or question == "q":
            if question == "p":
                print("You want to add a prefix.")
                question_bool = False
                rename_prefix(folder_path)
                print("Successfully rename!")
            if question == "s":
                print("You want to add a suffix.")
                question_bool = False
                rename_suffix(folder_path)
                print("Successfully rename!")
            if question == "b":
                print("You want to add a prefix and a suffix.")
                question_bool = False
                rename_prefix(folder_path)
                rename_suffix(folder_path)
                print("Successfully rename!")
            if question == "q":
                sys.exit()
        else:
            question = input("Please select between p, s, b or Q to quit: ")
            i += 1
            if i == 2:
                question_bool = False

    # Ask for another try to rename in another folder
    another = input("Do you want to rename another files? (y/n) ")
    another = str(another)
    another = another.lower()
    another_bool = True

    i = 0
    while another_bool:
        if another == "y" or another == "n":
            if another == "y":
                another_bool = False
                retry = True
            if another == "n":
                another_bool = False
                retry = False
        else:
            another = input("Select y for yes or n for no: ")
            i += 1
            if i == 2:
                sys.exit()

os.system("pause")