workspace.py

import fitz  # PyMuPDF
from PySide6.QtCore import Qt
from PySide6.QtGui import QImage, QPixmap


class Workspace:
    def __init__(self):
        self.pdf_files = []  # 存储PDF文件列表

    def add_pdf(self, file_path):
        pdf_file = PDFFile(file_path)
        pdf_file.load_pages()
        self.pdf_files.append(pdf_file)

    def clear(self):
        self.pdf_files = []  # 清空工作区中的PDF文件

    def remove_pdf_by_name(self, pdf_name):
        """
        根据PDF文件名删除工作区中的PDF文件及其页面。

        Args:
            pdf_name (str): 要删除的PDF文件名。
        """
        self.pdf_files = [
            pdf for pdf in self.pdf_files if not pdf.file_path.endswith(pdf_name)
        ]

    def update(self):
        """
        更新工作区，检查每个PDF文件的页面数量。
        如果某个PDF文件的页面数量为0，则将其从工作区中移除。
        """
        self.pdf_files = [pdf for pdf in self.pdf_files if pdf.pages]


class PDFFile:
    def __init__(self, file_path):
        self.file_path = file_path
        self.pages = []

    def load_pages(self):
        doc = fitz.open(self.file_path)
        for i in range(doc.page_count):
            page = doc.load_page(i)
            pix = page.get_pixmap(alpha=False)
            image = QImage(
                pix.samples, pix.width, pix.height, pix.stride, QImage.Format_RGB888
            )
            thumbnail = QPixmap.fromImage(image).scaled(
                180, 240, Qt.KeepAspectRatio, Qt.SmoothTransformation
            )
            self.pages.append(PDFPage(i + 1, thumbnail, self.file_path))


class PDFPage:
    def __init__(self, page_number, thumbnail, parent_pdf_path):
        self.page_number = page_number
        self.thumbnail = thumbnail
        self.parent_pdf_path = parent_pdf_path
        self.rotation = 0

    def update_thumbnail(self):
        """
        根据当前旋转角度重新生成缩略图。
        """
        doc = fitz.open(self.parent_pdf_path)
        page = doc.load_page(self.page_number - 1)
        mat = fitz.Matrix(1, 1).prerotate(self.rotation)  # 修正方法名为prerotate
        pix = page.get_pixmap(matrix=mat, alpha=False)
        image = QImage(
            pix.samples, pix.width, pix.height, pix.stride, QImage.Format_RGB888
        )
        self.thumbnail = QPixmap.fromImage(image).scaled(
            180, 240, Qt.KeepAspectRatio, Qt.SmoothTransformation
        )
        doc.close()