Python tkinter开发截图工具无法裁剪保存及自定义存储路径问题求助

现有代码核心问题 #

• 类定义逻辑错误：截图、cv2窗口循环的代码直接写在类的属性定义区，Python加载类定义时就会直接执行这部分代码，而非点击「Take Screenshot」按钮后触发，导致程序一启动就优先跑cv2截图逻辑，tkinter主循环root.mainloop()被阻塞，只有按q退出cv2窗口后才会显示GUI界面。
• 回调逻辑错误：takeScreenshot被定义为类，按钮command直接绑定类不会按预期触发截图逻辑，且类中的mouse_crop方法调用时self参数传递会异常。
• 裁剪区域异常：如果用户从右下向左上拖动选择区域，起止坐标会出现x_start > x_end、y_start > y_end的情况，切片得到的roi是空数组，cv2.imwrite执行失败，不会生成截图文件。

修复实现（含自定义保存路径功能） #

以下是修复后的完整代码：

import cv2
import pyautogui
import tkinter as tk
from tkinter.filedialog import asksaveasfilename
import os
import numpy as np

root = tk.Tk()
root.title("截图工具")
canvas1 = tk.Canvas(root, width=300, height=300)
canvas1.pack()

cropping = False
x_start, y_start, x_end, y_end = 0, 0, 0, 0

def take_screenshot():
    global cropping, x_start, y_start, x_end, y_end
    # 重置裁剪状态
    cropping = False
    x_start, y_start, x_end, y_end = 0, 0, 0, 0

    # 捕获全屏
    im1 = pyautogui.screenshot()
    im1.save(r"monitor-1.png")
    image = cv2.imread('monitor-1.png')
    oriImage = image.copy()
    # 清理临时文件
    try:
        os.remove('monitor-1.png')
    except:
        pass

    def mouse_crop(event, x, y, flags, param=None):
        global x_start, y_start, x_end, y_end, cropping
        if event == cv2.EVENT_LBUTTONDOWN:
            x_start, y_start, x_end, y_end = x, y, x, y
            cropping = True
        elif event == cv2.EVENT_MOUSEMOVE:
            if cropping:
                x_end, y_end = x, y
        elif event == cv2.EVENT_LBUTTONUP:
            x_end, y_end = x, y
            cropping = False
            # 坐标归一化，处理反向选框问题
            x1 = min(x_start, x_end)
            y1 = min(y_start, y_end)
            x2 = max(x_start, x_end)
            y2 = max(y_start, y_end)
            # 仅当选择区域有效时执行保存
            if x2 > x1 and y2 > y1:
                roi = oriImage[y1:y2, x1:x2]
                # 弹出自定义路径选择框
                save_path = asksaveasfilename(
                    defaultextension=".png",
                    filetypes=[("PNG图片", "*.png"), ("JPG图片", "*.jpg"), ("所有文件", "*.*")],
                    title="选择截图保存位置"
                )
                if save_path:
                    cv2.imwrite(save_path, roi)
            # 裁剪完成后自动关闭截图窗口
            cv2.destroyAllWindows()

    cv2.namedWindow("image")
    cv2.setMouseCallback("image", mouse_crop)
    while True:
        i = image.copy()
        if cropping:
            cv2.rectangle(i, (x_start, y_start), (x_end, y_end), (255, 0, 0), 2)
        cv2.imshow("image", i)
        # 按q可随时退出截图
        if cv2.waitKey(1) == ord('q'):
            break
    cv2.destroyAllWindows()

def close_window():
    root.destroy()

myQuitButton = tk.Button(text="退出", command=close_window, font=10)
canvas1.create_window(50, 50, window=myQuitButton)
myButton = tk.Button(text="开始截图", command=take_screenshot, font=10)
canvas1.create_window(150, 150, window=myButton)

root.mainloop()

功能说明 #

• 启动后直接显示tkinter GUI界面，点击「开始截图」才会弹出全屏截图选择窗口
• 拖动鼠标即可框选裁剪区域，松开左键自动弹出保存对话框，支持自定义保存路径、文件名和格式
• 按q可随时退出截图状态，不生成保存文件

内容的提问来源于stack exchange，提问作者Ly Nguyen