Python：使用 excel 文件使用 shutil 进行正则表达式文件搜索

Question

我有一个正在为内部员工编写的程序，该程序采用 CSV 文件并在文件服务器中搜索 CSV 中列出的文件，然后将每个文件复制到桌面上的一个文件夹中。 我当前代码遇到的问题是 CSV 必须包含确切的名称，但我需要使用正则表达式搜索它并复制文件名与 CSV 中的文件名类似的文件。

excel 中的文件名类似于：D6957-QR-1452

服务器上的文件名如下所示：WM_QRLabels_D6957-QR-1452_11.5x11.5_M.pdf

from tkinter import filedialog, messagebox
import openpyxl
import tkinter as tk
from pathlib import Path
import shutil
import os


desktop = Path.home() / "Desktop/Comps"

tk.messagebox.showinfo("Select a directory","Select a directory" )
folder = filedialog.askdirectory()
root = tk.Tk()


root.title("Title")


lbl = tk.Label(
    root, text="Open the excel file that includes files to search for")
lbl.pack()

frame = tk.Frame(root)
frame.pack()
scrollbar = tk.Scrollbar(frame)
scrollbar.pack(side=tk.RIGHT, fill=tk.Y)
listbox = tk.Listbox(frame, yscrollcommand=scrollbar.set)


def load_file():
    wb_path = filedialog.askopenfilename(filetypes=[('Excel files', '.xlsx')])
    wb = openpyxl.load_workbook(wb_path)
    global sheet
    sheet = wb.active

    listbox.pack()
    file_names = [cell.value for row in sheet.rows for cell in row]
    for file_name in file_names:
        listbox.insert('end', file_name)
    return file_names  # <--- return your list 
    
            



def search_folder(folder, file_name):
    # Create an empty list to store the found file paths
    found_files = []
    for root, dirs, files in os.walk(folder):
        for file in files:
            if file in file_name:
                found_files.append(os.path.join(root, file))
                shutil.copy2(file, desktop)

    return found_files


excelBtn = tk.Button(root, text="Open Excel File",
                     command=None)
excelBtn.pack()


zipBtn = tk.Button(root, text="Copy to Desktop",
                   command=search_folder(folder, load_file()))
zipBtn.pack()


root.mainloop()


Program is able to find and copy exact file names but unable to file *like* files. 

Answer 1

将您的 search_folder 方法更改为如下所示：

def search_folder(folder, file_name):
    # Create an empty list to store the found file paths
    found_files = []
    for root, dirs, files in os.walk(folder):
        for file in files:
            for file_pattern in file_name:
                if file.find(file_pattern) > -1:
                    found_files.append(os.path.join(root, file))
                    shutil.copy2(file, desktop)

    return found_files

基本上，对于每个文件，我们遍历 file_name 数组以测试所有模式。

对于测试，我们使用find方法，如果匹配，则返回找到的 substring 的 position。

也就是说，如果搜索“foo”，将返回所有这些文件：

zazfoo
foobar
zazfoobar
foo