如果包含 zip，如何將所有 PNG 圖像從一個文件夾復制到另一個文件夾，而不是將其復制到新文件夾中

Question

我正在對圖像和 zip 文件夾進行一些腳本編寫，我從一個文件夾復制圖像並制作另一個文件夾並復制該圖像。

但是如果文件是zip文件夾，這里還有一件事，那么我會將 zip 文件夾復制到新目錄並復制到該目錄。

zip 文件夾怎么可能？ 我的圖像代碼是：

import glob
import shutil
import os

src_dir = "your/source/dir"
dst_dir = "your/destination/dir"
for jpgfile in glob.iglob(os.path.join(src_dir, "*.jpg")):
    shutil.copy(jpgfile, dst_dir)

Answer 1

如果我正確理解您的問題，您可能想要遍歷所有文件，而不僅僅是 jpg 文件。 在循環內，您可以檢查文件是圖像還是 zip 文件並采取相應措施。

from shutil import copyfile
import os

source = "your_source"
dest = "your_destination"

for file in os.listdir(source):
    if(file.endswith(".jpg")):
        copyfile(source,destination)
    elif(file.endswith(".zip")):
        destination = "new_destination"
        #do more stuff

Answer 2

謝謝大家這里是解決方案

import os 
    # Create directory
    dirName = 'new'
    zipdir = 'zip'
    src_dir = "/home/ali/Desktop/test/"
    dst_dir = "/home/ali/Desktop/test2/"
    for file in os.listdir(src_dir):
        if file.endswith(".png" or '.jpg'):
            try:
                # Create target Directory
                path = os.path.join(dst_dir, dirName)
                os.mkdir(path)
                print("Directory " , dirName ,  " Created ") 
            except FileExistsError:
                print("Directory " , dirName ,  " already exists")
            shutil.copy((src_dir+file) , path)
        if file.endswith(".zip"):
            try:
                # Create target Directory
                path = os.path.join(dst_dir, zipdir)
                os.mkdir(path)
                print("Directory " , dirName ,  " Created ") 
            except FileExistsError:
                print("Directory " , dirName ,  " already exists")
            shutil.copy((src_dir+file) , path)