Python结合PyQt6实现屏幕截图工具(附源码)

import sys
import os
from PyQt6.QtWidgets import (QApplication, QMainWindow, QPushButton, QWidget,
                             QVBoxLayout, QLabel, QMessageBox, QDialog)
from PyQt6.QtGui import QScreen, QPixmap, QPainter, QColor, QPen
from PyQt6.QtCore import Qt, QRect, QPoint, QDateTime


class ScreenshotMask(QDialog):
    def __init__(self, full_screen_pix):
        super().__init__()
        self.full_pix = full_screen_pix
        # 全屏无边框置顶
        self.setWindowFlags(
            Qt.WindowType.FramelessWindowHint
            | Qt.WindowType.WindowStaysOnTopHint
        )
        self.setGeometry(0, 0, self.full_pix.width(), self.full_pix.height())
        self.setAttribute(Qt.WidgetAttribute.WA_TranslucentBackground)
        self.setMouseTracking(True)

        self.start_pos = QPoint()
        self.end_pos = QPoint()
        self.is_drawing = False
        self.select_rect = QRect()
        self.result_pix = None

    def paintEvent(self, event):
        painter = QPainter(self)
        # 底层原图
        painter.drawPixmap(self.rect(), self.full_pix)
        # 半透明遮罩
        painter.fillRect(self.rect(), QColor(0, 0, 0, 130))

        if self.is_drawing:
            rect = QRect(self.start_pos, self.end_pos).normalized()
            # 选区清空遮罩
            painter.setCompositionMode(QPainter.CompositionMode.CompositionMode_Clear)
            painter.fillRect(rect, Qt.GlobalColor.transparent)
            painter.setCompositionMode(QPainter.CompositionMode.CompositionMode_SourceOver)
            # 绘制边框
            pen = QPen(QColor(0, 160, 255), 2)
            painter.setPen(pen)
            painter.drawRect(rect)
            self.select_rect = rect

    def mousePressEvent(self, event):
        if event.button() == Qt.MouseButton.LeftButton:
            self.is_drawing = True
            self.start_pos = event.pos()
            self.end_pos = event.pos()
            self.update()

    def mouseMoveEvent(self, event):
        if self.is_drawing:
            self.end_pos = event.pos()
            self.update()

    def mouseReleaseEvent(self, event):
        if event.button() == Qt.MouseButton.LeftButton and self.is_drawing:
            self.is_drawing = False
            if not self.select_rect.isEmpty():
                self.result_pix = self.full_pix.copy(self.select_rect)
            self.accept()  # 关闭对话框

    def keyPressEvent(self, event):
        if event.key() == Qt.Key.Key_Escape:
            self.reject()  # 取消截图


class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.init_ui()

    def init_ui(self):
        self.setWindowTitle("PyQt6 截图工具")
        self.resize(300, 150)
        center = QWidget()
        self.setCentralWidget(center)
        layout = QVBoxLayout(center)
        layout.setSpacing(12)
        layout.setContentsMargins(30, 30, 30, 30)

        btn_full = QPushButton("全屏截图")
        btn_area = QPushButton("区域拖拽截图")
        tip = QLabel("按住鼠标拖拽选区 | ESC取消")
        tip.setAlignment(Qt.AlignmentFlag.AlignCenter)
        tip.setStyleSheet("color:#666")

        layout.addWidget(btn_full)
        layout.addWidget(btn_area)
        layout.addWidget(tip)

        btn_full.clicked.connect(self.capture_full)
        btn_area.clicked.connect(self.start_area_capture)

    def capture_full(self):
        screen = QApplication.primaryScreen()
        full_img = screen.grabWindow(0)
        time_str = QDateTime.currentDateTime().toString("yyyyMMdd_HHmmss")
        save_path = f"full_{time_str}.png"
        full_img.save(save_path, "PNG")
        QMessageBox.information(self, "保存成功", f"路径：\n{os.path.abspath(save_path)}")

    def start_area_capture(self):
        # 先隐藏主窗口，避免截图包含自身
        self.hide()
        QApplication.processEvents()

        screen = QApplication.primaryScreen()
        full_img = screen.grabWindow(0)

        mask = ScreenshotMask(full_img)
        # 正确使用QDialog的exec()，不会卡死
        ret = mask.exec()

        # 截图完成后恢复窗口
        self.show()

        if ret == 1 and mask.result_pix is not None:
            time_str = QDateTime.currentDateTime().toString("yyyyMMdd_HHmmss")
            save_path = f"region_{time_str}.png"
            mask.result_pix.save(save_path, "PNG")
            QMessageBox.information(self, "区域截图完成", f"已保存：\n{os.path.abspath(save_path)}")


if __name__ == "__main__":
    app = QApplication(sys.argv)
    win = MainWindow()
    win.show()
    sys.exit(app.exec())

import sys
from PyQt6.QtWidgets import QApplication, QMainWindow, QPushButton, QWidget, QVBoxLayout
from PyQt6.QtGui import QScreen
from PyQt6.QtCore import Qt

class ScreenshotWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.init_ui()

def init_ui(self):
    self.setWindowTitle("屏幕截图工具")
    self.resize(300, 120)
    
    # 中心部件与布局
    central_widget = QWidget()
    self.setCentralWidget(central_widget)
    layout = QVBoxLayout(central_widget)
    layout.setContentsMargins(50, 30, 50, 30)
    
    # 截图按钮
    self.btn_capture = QPushButton("点击截取全屏并保存PNG")
    self.btn_capture.clicked.connect(self.capture_screen)
    layout.addWidget(self.btn_capture, alignment=Qt.AlignmentFlag.AlignCenter)

def capture_screen(self):
    # 获取主屏幕对象
    screen: QScreen = QApplication.primaryScreen()
    # 截取整个屏幕
    pixmap = screen.grabWindow(0)
    # 保存为png文件，保存在程序同级目录
    save_path = "screenshot.png"
    pixmap.save(save_path, "PNG")
    print(f"截图已保存至: {save_path}")

def capture_all_screens(self):
    """截取所有屏幕"""
    screens = QApplication.screens()
    for i, screen in enumerate(screens):
        pixmap = screen.grabWindow(0)
        pixmap.save(f"screenshot_screen_{i}.png", "PNG")

from PyQt6.QtWidgets import QFileDialog

def capture_with_dialog(self):
    screen = QApplication.primaryScreen()
    pixmap = screen.grabWindow(0)
    
    # 弹出文件保存对话框
    file_path, _ = QFileDialog.getSaveFileName(
        self, 
        "保存截图", 
        "screenshot.png", 
        "PNG Files (*.png);;JPEG Files (*.jpg);;All Files (*)"
    )
    
    if file_path:
        pixmap.save(file_path, "PNG")
        print(f"截图已保存至: {file_path}")

def capture_region(self, x, y, width, height):
    """截取屏幕指定区域"""
    screen = QApplication.primaryScreen()
    pixmap = screen.grabWindow(0, x, y, width, height)
    return pixmap

from datetime import datetime

def capture_with_timestamp(self):
    screen = QApplication.primaryScreen()
    pixmap = screen.grabWindow(0)
    
    # 生成带时间戳的文件名
    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
    save_path = f"screenshot_{timestamp}.png"
    
    pixmap.save(save_path, "PNG")
    print(f"截图已保存至: {save_path}")

# 启用高DPI支持
if hasattr(Qt, 'AA_EnableHighDpiScaling'):
    QApplication.setAttribute(Qt.AA_EnableHighDpiScaling, True)
if hasattr(Qt, 'AA_UseHighDpiPixmaps'):
    QApplication.setAttribute(Qt.AA_UseHighDpiPixmaps, True)