Python删除指定目录下所有文件及目录,复制指定目录文件到指定目录

前言

加密文件夹下面的图片,不被盗用

功能:

1、删除指定目录所有文件,可选项删除指定目录下其它所有子目录

2、随机复制指定目录下10张图片到指定目录下

以下是完整代码:

 1 import os, random, shutil
 2 
 3 ##先删除目录下所有文件及文件夹
 4 os.chdir('Z:\\游戏菜单\\WIN7WEB') #进入要清空的目录
 5 ds = list(os.listdir()) #获得该目录下所有文件或文件夹列表
 6 for d in ds: #遍历该列表
 7     if os.path.isfile(d): #如果列表项是文件
 8         os.remove(d) #直接删除
 9 #    else: #如果不是文件,肯定是文件夹
10 #        shutil.rmtree(d) #也直接删除
11 
12 ##复制文件代码        
13 def copyFile(fileDir):
14         pathDir = os.listdir(fileDir)    #取图片的原始路径
15         filenumber=len(pathDir)
16         rate=0.01    #自定义抽取图片的比例,比方说100张抽10张,那就是0.1
17         picknumber=int(filenumber*rate) #按照rate比例从文件夹中取一定数量图片
18         sample = random.sample(pathDir, picknumber)  #随机选取picknumber数量的样本图片
19         print (sample)
20         for name in sample:
21                 shutil.copy(fileDir+name, tarDir+name)
22         return
23 os.chdir('Z:\\') #进入要清空的目录
24 if __name__ == '__main__':
25     fileDir = "./必应壁纸/"    #源图片文件夹路径
26     tarDir = './游戏菜单/WIN7WEB/'    #复制到新的文件夹路径
27     copyFile(fileDir)