Скопируйте папки в новую папку и добавьте

Question

Это первая часть, которая работает, в основном создала 3 новых обязательных папки (1,2,3) в главном каталоге TEST, и если пользователь хочет добавить новый, они могут это сделать , Это Python_1

import os 
root_path = r"C:\TEST" 
list_dir = [] 
while True: 
    userinput1 = raw_input("Enter the name for Folder1, Folder2, Folder3:") 
    list_dir.append(userinput1) 
    userinput2 = None 
#ask user to respond 'yes' or 'no' as to whether they want to add another directory 
    while userinput2 != "yes" and userinput2 != "no": 
     userinput2 = raw_input("Would you like to add another directory? yes/no: ") 
    if userinput2 == "no": 
     break 
for directory in list_dir: 
    os.mkdir(os.path.join(root_path, directory)) 
print 'New directories have been created' 

Теперь мне нужен новый сценарий, который делает следующее: Создает новую основную папку РАБОЧЕЕ в TEST, и скопировать все папки, созданные в Python_1 во вновь созданной рабочей папку с прилагаемым _working с именем из скопированные папки.

Любые идеи? Большое спасибо за вашу помощь!

Answer 1

import os 
import shutil 

root_path = r"C:\TEST" 
new_main_folder = 'WORKING' 

list_dir = next(os.walk(root_path))[1] # Folder1, Folder2, Folder3 from Python_1 

# Creates C:\TEST\WORKING 
new_root_path = os.path.join(root_path, new_main_folder) 
os.mkdir(new_root_path) 

for directory in list_dir: 
    src = os.path.join(root_path, directory) # C:\TEST\Folder[1,2,3] 
    dest = os.path.join(new_root_path, directory + '_working') # C:\TEST\WORKING\Folder[1,2,3]_working 
    # or 
    # renamed_folder = directory + '_' + new_main_folder.lower() 
    # dest = os.path.join(new_root_path, renamed_folder) 
    shutil.copytree(src, dest)