我的子文件夾中有多個(gè)文件需要復(fù)制到另一個(gè)具有指定子文件夾名稱和日期時(shí)間的主文件夾，意味著A1/B/C/D/D.txt復(fù)制到A2/B/C/D/datetime_username/D.txt，目錄 B、C、D 中有多個(gè)文件夾，也許是 B1、B2...用戶決定要傳輸哪個(gè)子文件夾現(xiàn)在我正在嘗試打印 D 文件夾的路徑。我嘗試的文件夾位于我的桌面上。 import os, shutil os.chdir('C:\\path') def main(): lvl1 = ['Typ','GFP2'] lvl2 = ['A0','A1'] lvl3 = ['SCAN','MBIST'] project = input("Enter project: ") #Validation for s in lvl1: if s == project: src1 = 'IDC\\{}'.format(s) stepping = input("Enter Stepping: ") for l2 in lvl2: if l2 == stepping: #shutil.copytree() src2 = src1+'\\{}'.format(l2) thirdlvl = input("Enter SCAN/MBIST: ") for l3 in lvl3: if l3 == thirdlvl: src3 = src2+'\\{}'.format(l3) print(src3) else: print("Stepping not found") break else: print("Project name not found") main()我會得到'''Enter project: TypEnter Stepping: A0Enter SCAN/MBIST: SCANIDC\Typ\A0\SCANStepping not found'''如果我想僅在未找到步進(jìn)時(shí)打印“未找到步進(jìn)”，現(xiàn)在即使存在此文件夾也會打印它。謝謝
1 回答
慕虎7371278
TA貢獻(xiàn)1802條經(jīng)驗(yàn) 獲得超4個(gè)贊
您需要break在您的for-else. 最小修改是：
def main():
lvl1 = ['Typ','GFP2']
lvl2 = ['A0','A1']
lvl3 = ['SCAN','MBIST']
project = input("Enter project: ")
#Validation
for s in lvl1:
if s == project:
src1 = 'IDC\\{}'.format(s)
stepping = input("Enter Stepping: ")
for l2 in lvl2:
if l2 == stepping:
#shutil.copytree()
src2 = src1+'\\{}'.format(l2)
thirdlvl = input("Enter SCAN/MBIST: ")
for l3 in lvl3:
if l3 == thirdlvl:
src3 = src2+'\\{}'.format(l3)
print(src3)
break
else:
print("Stepping not found")
break
else:
print("Project name not found")
for-loop您還可以通過以下方式擺脫
def main():
lvl1 = ['Typ','GFP2']
lvl2 = ['A0','A1']
lvl3 = ['SCAN','MBIST']
project = input("Enter project: ")
#Validation
if project in lvl1:
src1 = 'IDC\\{}'.format(project)
stepping = input("Enter Stepping: ")
if stepping in lvl2:
#shutil.copytree()
src2 = src1+'\\{}'.format(stepping)
thirdlvl = input("Enter SCAN/MBIST: ")
if thirdlvl in lvl3:
src3 = src2+'\\{}'.format(thirdlvl)
print(src3)
else:
print("Stepping not found")
else:
print("Project name not found")
if-else不需要多層結(jié)構(gòu)。這是一個(gè)修改。我還使用os.path.join()比字符串操作更適合處理路徑：
import os, shutil
working_dir = os.path.join("C:/", "path")
os.chdir(working_dir)
def main():
lvl1 = ['Typ','GFP2']
lvl2 = ['A0','A1']
# lvl3 = ['SCAN','MBIST']
project = input("Enter project: ")
if project not in lvl1:
print("Project name not found")
return
stepping = input("Enter Stepping: ")
if stepping not in lvl2:
print("Stepping not found")
return
thirdlvl = input("Enter SCAN/MBIST: ")
src = os.path.join(project, stepping, thirdlvl)
print("src", src)
if not os.path.exists(src):
print("Src not exists")
main()
os.listdir()您甚至可以通過為您列出所有子文件夾來進(jìn)一步自動化該過程。
import os, shutil
working_dir = os.path.join("C:/", "path")
os.chdir(working_dir)
def main():
project = input("Enter project: ")
lvl1 = os.listdir(working_dir) # list all the directories under the working directory.
if project not in lvl1:
print("Project name not found")
return # if you want to exit when not finding the project folder.
stepping = input("Enter Stepping: ") # list all the directories under the project directory.
lvl2 = os.listdir(os.path.join(working_dir, project))
if stepping not in lvl2:
print("Stepping not found")
return # if you want to exit when not finding the stepping folder.
thirdlvl = input("Enter SCAN/MBIST: ")
src = os.path.join(project, stepping, thirdlvl)
print("src", src)
if not os.path.exists(src):
print("Src not exists")
main()
添加回答
舉報(bào)
0/150
提交
取消