我有一个python脚本,它运行以在循环中截取屏幕截图并保存所有这些图像。但是目前,所有图像都存储在Python35文件夹中。
因此,我希望用户输入文件夹的名称,并使用此名称创建一个名称。
我知道使用makedirs函数并按newpath=r'C:/users/../'
指定路径。
但是如何将用户指定的新文件夹名称附加到路径?
(编辑)以下是我的代码: -
import openpyxl
import re
import os
import time
from PIL import ImageGrab
import webbrowser
source_excel=input('Enter the source excel file name-> ')
wb=openpyxl.load_workbook(source_excel)
sheet=wb.active
strt=int(input('Enter starting range-> '))
end=int(input('Enter ending range-> '))
#NEED THE CHANGE HERE
foldname=input("Enter a folder name to create where images will be saved")
newpath=r'C:\Users\rnair\AppData\Local\Programs\Python\Python35\'demofile
# Loop to get the link from excel and take the screenshot by opening the browser for each link
for q in range(strt,end):
getme=q,sheet.cell(row=q,column=1).value #get link from excel file
ab=getme[1]
#myopener=Myopen()
rowc=str(strt)
url=ab
webbrowser.open(url) #opens browser for the link fetched
time.sleep(7) # Delay to wait for the page to load completely
im=ImageGrab.grab()
os.chdir(newpath)
im.save("Row"+rowc+".jpg","JPEG") # Saving image to file
strt=strt+1
答案 0 :(得分:3)
如果您的意思是构建应保存文件的路径,请首先使用基本目录os.path.join
,然后使用自定义目录名称(文件名可以作为第三个参数提供)。只需要非常小心地信任任何用户输入,尤其是文件系统操作。