Python3 從一個文件夾復制文件到另一個文件夾

需求：從input文件夾中隨機獲取一定數量的jpg文件復制到output文件夾中，

如果input文件夾中有與選中的jpg文件文件名相同的ans文件則復制到output文件夾中

參考網頁地址：

https://blog.csdn.net/weixin_40769885/article/details/82869760

https://blog.csdn.net/u010167269/article/details/51084312

https://blog.csdn.net/u010454261/article/details/80903804

 

代碼：

 

#!/usr/bin/env python
# encoding: utf-8

import os, random, glob
from shutil import copyfile


#定義要處理的文件夾變量
#choseImg是隨機選中的jpg文件文件名包括后綴
choseImg = []

def copyFile(inputDir, outDir, count):
        #通過glob.glob來獲取原始路徑下，所有'.jpg'文件
        imageList1 = glob.glob(os.path.join(inputDir, '*.jpg'))
        #隨機獲取count數量的jpg文件
        imageRandom = random.sample(imageList1, count)

        #遍歷所有隨機得到的jpg文件，獲取文件名稱（包括后綴）
        for item in imageRandom:
                choseImg.append(os.path.basename(item))

        #os.path.splitext()，返回元組，為文件名稱與文件后綴格式
        for item in choseImg:
                #將隨機選中的jpg文件遍歷復制到目標文件夾中
                copyfile(inputDir+'/'+item, outDir+'/'+item)

                #更改jpg文件后綴為ans
                (temp1, temp2) = os.path.splitext(item)
                temp = temp1 + '.ans'
                if os.path.exists(inputDir+'/'+temp):
                        #將ans文件遍歷復制到目標文件夾中
                        copyfile(inputDir+'/'+temp, outDir+'/'+temp)    
        return
    
if __name__ == '__main__':
        inputfile = input('請輸入原始路徑（如：E:/input/）：')
        outfile = input('請輸入目標路徑（如：E:/output/）：')
        count = int(input('請輸入隨機選取的數量（如：3）：'))
        #指定找到文件后，另存為的文件夾路徑
        outDir = os.path.abspath(outfile)
        #指定文件的原始路徑
        inputDir = os.path.abspath(inputfile)
        copyFile(inputDir, outDir, count)

 

 

出現的錯誤：

1. inconsistent use of tabs and spaces in indentation

解決辦法：重新調整代碼縮進，每次縮進都使用8個space

參考地址：https://blog.csdn.net/weixin_38501796/article/details/79225635

2.沒有判斷ans文件是否在目錄中存在

增加判斷： if os.path.exists(inputDir+'/'+temp):

參考地址:https://www.jb51.net/article/119750.htm