使用Python结合Tkinter和PyAutoGUI开发精确截图工具

import tkinter as tk
from tkinter import ttk
import pyautogui
import numpy as np
from PIL import Image, ImageTk, ImageGrab
import os
from datetime import datetime
import time

class ImprovedScreenshotTool:
    def __init__(self):
        # Create the main window
        self.root = tk.Tk()
        self.root.title("精确截图工具")
        self.root.geometry("400x200")
        self.root.resizable(False, False)
        
        # Center the window
        self.center_window()
        
        # Create a frame for the controls
        control_frame = ttk.Frame(self.root)
        control_frame.pack(pady=10, fill=tk.X)
        
        # Create and place the screenshot button
        self.screenshot_btn = ttk.Button(
            control_frame, 
            text="开始截图",
            width=20,
            command=self.prepare_screenshot
        )
        self.screenshot_btn.pack(pady=10)
        
        # Status label
        self.status_label = ttk.Label(control_frame, text="就绪")
        self.status_label.pack(pady=5)
        
        # Preview frame
        self.preview_frame = ttk.LabelFrame(self.root, text="最近截图预览")
        self.preview_frame.pack(pady=10, padx=10, fill=tk.BOTH, expand=True)
        
        self.preview_label = ttk.Label(self.preview_frame)
        self.preview_label.pack(fill=tk.BOTH, expand=True)
        
        # Variables for region selection
        self.start_x = 0
        self.start_y = 0
        self.end_x = 0
        self.end_y = 0
        
        # Save directory
        pictures_dir = os.path.join(os.path.expanduser("~"), "Pictures")
        if os.path.exists(pictures_dir):
            self.save_dir = pictures_dir
        else:
            self.save_dir = os.path.join(os.path.expanduser("~"), "Desktop")
            
        # Make sure the directory exists
        if not os.path.exists(self.save_dir):
            os.makedirs(self.save_dir)
            
        # Last saved file
        self.last_saved_file = None
        
    def center_window(self):
        # Get screen width and height
        screen_width = self.root.winfo_screenwidth()
        screen_height = self.root.winfo_screenheight()
        
        # Calculate position
        x = (screen_width - 400) // 2
        y = (screen_height - 200) // 2
        
        # Set window position
        self.root.geometry(f"400x200+{x}+{y}")
            
    def prepare_screenshot(self):
        # Update status
        self.status_label.config(text="准备截图...")
        self.root.update()  # Force UI update
        
        # Minimize window
        self.root.withdraw()
        
        # Wait a moment for UI to update
        self.root.after(500, self.take_screenshot)
    
    def take_screenshot(self):
        # Create overlay window for selection
        overlay = ScreenshotOverlay(self)
        self.root.wait_window(overlay.overlay)
        
        # If cancelled, just return to normal
        if not hasattr(self, 'screenshot_region') or not self.screenshot_region:
            self.root.deiconify()
            self.status_label.config(text="已取消截图")
            return
            
        # Get the selection coordinates
        x1, y1, x2, y2 = self.screenshot_region
        
        # Convert to proper coordinates (top-left, bottom-right)
        left = min(x1, x2)
        top = min(y1, y2)
        right = max(x1, x2)
        bottom = max(y1, y2)
        
        # Ensure minimum size
        if right - left < 5 or bottom - top < 5:
            self.root.deiconify()
            self.status_label.config(text="选择区域太小")
            return
        
        # Take screenshot
        try:
            # Wait a moment to ensure overlay is gone
            time.sleep(0.3)
            
            # Capture the screen region
            screenshot = pyautogui.screenshot(region=(left, top, right-left, bottom-top))
            
            # Generate filename
            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
            filename = os.path.join(self.save_dir, f"screenshot_{timestamp}.png")
            
            # Save the screenshot
            screenshot.save(filename)
            self.last_saved_file = filename
            
            # Update preview
            self.update_preview(screenshot)
            
            # Update status
            self.status_label.config(text=f"截图已保存: {os.path.basename(filename)}")
            
        except Exception as e:
            self.status_label.config(text=f"截图错误: {str(e)}")
        
        # Show the main window again
        self.root.deiconify()
    
    def update_preview(self, image):
        # Resize for preview if needed
        preview_width = 360
        preview_height = 100
        
        width, height = image.size
        ratio = min(preview_width/width, preview_height/height)
        new_size = (int(width * ratio), int(height * ratio))
        
        resized = image.resize(new_size, Image.LANCZOS)
        
        # Convert to PhotoImage
        photo = ImageTk.PhotoImage(resized)
        
        # Update label
        self.preview_label.config(image=photo)
        self.preview_label.image = photo  # Keep a reference
    
    def run(self):
        self.root.mainloop()


class ScreenshotOverlay:
    def __init__(self, parent):
        self.parent = parent
        
        # Create fullscreen overlay window
        self.overlay = tk.Toplevel()
        self.overlay.attributes('-fullscreen', True)
        self.overlay.attributes('-alpha', 0.3)
        self.overlay.attributes('-topmost', True)
        
        # Make it semi-transparent with dark background
        self.overlay.configure(bg='black')
        
        # Add a canvas for drawing
        self.canvas = tk.Canvas(
            self.overlay,
            bg='#1a1a1a',
            highlightthickness=0,
            cursor="crosshair"
        )
        self.canvas.pack(fill=tk.BOTH, expand=True)
        
        # Variables for tracking
        self.start_x = None
        self.start_y = None
        self.rect_id = None
        self.magnifier_id = None
        self.coords_text_id = None
        
        # Instructions text
        self.canvas.create_text(
            self.overlay.winfo_screenwidth() // 2,
            50,
            text="单击并拖动鼠标选择截图区域 | 按ESC取消",
            fill="white",
            font=("Arial", 16)
        )
        
        # Bind events
        self.canvas.bind("<ButtonPress-1>", self.on_press)
        self.canvas.bind("<B1-Motion>", self.on_drag)
        self.canvas.bind("<ButtonRelease-1>", self.on_release)
        self.overlay.bind("<Escape>", self.on_cancel)
        
        # Take a full screenshot for the magnifier
        self.screen_image = pyautogui.screenshot()
        self.screen_array = np.array(self.screen_image)
    
    def on_press(self, event):
        # Record start position
        self.start_x = event.x
        self.start_y = event.y
        
        # Create rectangle
        self.rect_id = self.canvas.create_rectangle(
            self.start_x, self.start_y, 
            self.start_x, self.start_y,
            outline="#00ff00", width=2, fill=""
        )
        
        # Create magnifier circle
        self.magnifier_id = self.canvas.create_oval(
            event.x - 50, event.y - 50,
            event.x + 50, event.y + 50,
            outline="#ffffff", width=2, fill="#333333"
        )
        
        # Create coordinate display
        self.coords_text_id = self.canvas.create_text(
            event.x, event.y - 60,
            text=f"({event.x}, {event.y})",
            fill="#ffffff",
            font=("Arial", 10)
        )
    
    def on_drag(self, event):
        # Update rectangle
        self.canvas.coords(
            self.rect_id,
            self.start_x, self.start_y,
            event.x, event.y
        )
        
        # Update magnifier position
        self.canvas.coords(
            self.magnifier_id,
            event.x - 50, event.y - 50,
            event.x + 50, event.y + 50
        )
        
        # Update coordinate display
        self.canvas.coords(
            self.coords_text_id,
            event.x, event.y - 60
        )
        self.canvas.itemconfig(
            self.coords_text_id,
            text=f"({event.x}, {event.y}) | 大小: {abs(event.x - self.start_x)}x{abs(event.y - self.start_y)}"
        )
        
        # Draw selection area with fill (using a valid color format)
        self.canvas.itemconfig(
            self.rect_id, 
            fill="#22ff22"  # Changed to a valid semi-transparent green
        )
    
    def on_release(self, event):
        # Store the selection coordinates in the parent
        self.parent.screenshot_region = (
            self.start_x, self.start_y, 
            event.x, event.y
        )
        
        # Close the overlay
        self.overlay.destroy()
    
    def on_cancel(self, event):
        # Reset parent's screenshot region
        self.parent.screenshot_region = None
        
        # Close overlay
        self.overlay.destroy()


if __name__ == "__main__":
    app = ImprovedScreenshotTool()
    app.run()

import tkinter as tk
from tkinter import ttk
import pyautogui
import numpy as np
from PIL import Image, ImageTk
import os
from datetime import datetime
import time

class ImprovedScreenshotTool:
    def __init__(self):
        self.root = tk.Tk()
        self.root.title("精确截图工具")
        self.root.geometry("400x200")
        self.root.resizable(False, False)
        self.center_window()
        
        self.create_widgets()
        self.setup_save_directory()

    def center_window(self):
        screen_width = self.root.winfo_screenwidth()
        screen_height = self.root.winfo_screenheight()
        x = (screen_width - 400) // 2
        y = (screen_height - 200) // 2
        self.root.geometry(f"400x200+{x}+{y}")

    def create_widgets(self):
        control_frame = ttk.Frame(self.root)
        control_frame.pack(pady=10, fill=tk.X)
        
        self.screenshot_btn = ttk.Button(control_frame, text="开始截图", width=20, command=self.prepare_screenshot)
        self.screenshot_btn.pack(pady=10)
        
        self.status_label = ttk.Label(control_frame, text="就绪")
        self.status_label.pack(pady=5)
        
        self.preview_frame = ttk.LabelFrame(self.root, text="最近截图预览")
        self.preview_frame.pack(pady=10, padx=10, fill=tk.BOTH, expand=True)
        
        self.preview_label = ttk.Label(self.preview_frame)
        self.preview_label.pack(fill=tk.BOTH, expand=True)

    def prepare_screenshot(self):
        self.status_label.config(text="准备截图...")
        self.root.update()
        self.root.withdraw()
        self.root.after(500, self.take_screenshot)

    def take_screenshot(self):
        overlay = ScreenshotOverlay(self)
        self.root.wait_window(overlay.overlay)
        
        if not hasattr(self, 'screenshot_region') or not self.screenshot_region:
            self.root.deiconify()
            self.status_label.config(text="已取消截图")
            return
        
        x1, y1, x2, y2 = self.screenshot_region
        left, top = min(x1, x2), min(y1, y2)
        right, bottom = max(x1, x2), max(y1, y2)
        
        if right - left < 5 or bottom - top < 5:
            self.root.deiconify()
            self.status_label.config(text="选择区域太小")
            return
        
        time.sleep(0.3)
        screenshot = pyautogui.screenshot(region=(left, top, right-left, bottom-top))
        filename = os.path.join(self.save_dir, f"screenshot_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png")
        screenshot.save(filename)
        self.last_saved_file = filename
        
        self.update_preview(screenshot)
        self.status_label.config(text=f"截图已保存: {os.path.basename(filename)}")
        self.root.deiconify()

class ScreenshotOverlay:
    def __init__(self, parent):
        self.parent = parent
        self.overlay = tk.Toplevel()
        self.overlay.attributes('-fullscreen', True)
        self.overlay.attributes('-alpha', 0.3)
        self.overlay.attributes('-topmost', True)
        self.overlay.configure(bg='black')
        
        self.canvas = tk.Canvas(self.overlay, bg='#1a1a1a', highlightthickness=0, cursor="crosshair")
        self.canvas.pack(fill=tk.BOTH, expand=True)
        
        self.canvas.bind("<ButtonPress-1>", self.on_press)
        self.canvas.bind("<B1-Motion>", self.on_drag)
        self.canvas.bind("<ButtonRelease-1>", self.on_release)
        self.overlay.bind("<Escape>", self.on_cancel)