## PyQt5 自定义可拖拽列表组件：实现动态排序、上下移动与删除功能
写在前面：

在本篇教程中，将实现一个基于 **PyQt5** 的自定义列表组件，这个组件支持拖拽排序、点击按钮上下移动、以及点击删除的功能。

实现效果：


![预览.gif](https://blog.lc044.love/static/img/82ba082dc03977ae0b3331c2427d57cb.%C3%A9%C2%A2%C2%84%C3%A8%C2%A7%C2%88.gif)

:::success
为什么不用Qt自带的`QListWidget` 和 `QListView`？

1. `QListWidget` 支持拖拽，但仅限于默认的文字图标样式的item，不支持item上显示自定义widget（拖拽之后会出现空白）
2. **自定义拖拽图像**：默认的 `QListWidget` 直接拖动列表项的文字或图标，而我们希望通过截图方式，让拖拽过程中显示小部件的实际样式。
3. **控制拖拽行为**：在自定义组件中，可以细粒度地控制拖拽行为，甚至动态调整拖拽图像、拖拽位置、以及拖拽结果的处理逻辑。例如，我们可以在拖拽时调整列表项的大小，或者设置复杂的拖拽区域判断。

:::

### 环境准备
首先，请确保你的环境中安装了 **PyQt5**，可以使用以下命令安装：

```python
pip install PyQt5
```

### 项目需求分析
实现一个自定义列表组件，要求如下：

1. 每个列表项由一个标签和三个操作按钮组成（上移、下移、删除）。
2. 支持拖拽排序，使用户可以自由地重新排列列表项的顺序。
3. 能够通过按钮快速将项上下移动或删除。

### 代码详解
#### 1. 项目结构
首先我们创建以下几个类：

+ `CustomWidget`: 定义列表项样式，包括标签和按钮。
+ `FileInfo`: 存储每个列表项的信息。
+ `FileItem`: 继承 `QStandardItem`，保存自定义数据。
+ `FileListView`: 主列表视图类，实现拖拽、排序和按钮操作的核心逻辑。

#### 2. 自定义列表项 `CustomWidget`
`CustomWidget` 负责定义每个列表项的外观，包括：

+ 标签显示项的名称。
+ `↑` 按钮（上移）、`↓` 按钮（下移）、以及 `删除` 按钮。
+ 鼠标悬停效果。

代码实现如下：

```python
class CustomWidget(QWidget):
    """自定义列表项，带有标签和删除按钮"""

    def __init__(self, text):
        super(CustomWidget, self).__init__()
        self.is_selected = False
        self.label = QLabel(os.path.basename(text))
        self.btn_up = QPushButton("↑")
        self.btn_down = QPushButton("↓")
        self.btn_delete = QPushButton("删除")

        # 创建布局
        layout = QHBoxLayout()
        layout.addWidget(self.label)
        spacerItem = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)
        layout.addItem(spacerItem)
        layout.addWidget(self.btn_up)
        layout.addWidget(self.btn_down)
        layout.addWidget(self.btn_delete)
        layout.setContentsMargins(0, 0, 10, 0)
        self.setLayout(layout)
        self.setStyleSheet("QWidget,QLabel{background: rgb(238,244,249);}")
```

通过 `QLabel` 显示标签信息，`QPushButton` 添加操作按钮，并设置 `QHBoxLayout` 布局，使标签和按钮水平排列。

#### 3. 数据结构 `FileInfo` 和 `FileItem`
`FileInfo` 类保存每个列表项的具体信息，如文件名、路径和排序所需的页码。

```python
class FileInfo:
    def __init__(self, file, index):
        self.filename = os.path.basename(file)
        self.filepath = file
        self.index = index
        self.selected = True
```

`FileItem` 继承自 `QStandardItem`，在每项中存储 `FileInfo` 数据。

```python
class FileItem(QStandardItem):
    def __init__(self, fileinfo):
        super(FileItem, self).__init__()
        self.setData(fileinfo, Qt.UserRole)  # 存储自定义文本数据
```

#### 4. 列表视图 `FileListView`
`FileListView` 是组件的核心部分，定义了以下功能：

+ **拖拽排序**：通过 `mousePressEvent` 和 `startDrag` 实现拖拽效果。
+ **列表操作**：调用 `add_item` 添加列表项，通过 `remove_item` 删除项。

##### 关键功能实现
+ **拖拽**：

在 `FileListView` 的 `startDrag` 方法中，生成自定义小部件的截图作为拖拽图像，并设置拖拽热点，确保拖拽开始时图像与鼠标对齐。

```python
def startDrag(self, supportedActions):
    index = self.currentIndex()
    if index.isValid():
        item_widget = self.indexWidget(index)
        if item_widget:
            pixmap = item_widget.grab()  
            drag = QDrag(self)
            drag.setPixmap(pixmap)
            drag.setHotSpot(self.start_drag_pos - item_widget.pos())
            drag.exec_(supportedActions)
```

+ **上下移动**：

按钮 `↑` 和 `↓` 实现上下移动功能。通过获取目标行号和源行号，进行插入和删除操作。

```python
def move_item_up(self, item):
    row = item.row()
    if row > 0:
        new_item = FileItem(item.data(Qt.UserRole).copy())
        self.model.removeRow(item.row())
        self.model.insertRow(row - 1, new_item)
        self.set_item_widget(new_item)
```

+ **删除**：

当点击删除按钮时，调用 `remove_item` 方法，将指定项从模型中移除。

```python
def remove_item(self, item):
    self.model.removeRow(item.row())
```

#### 5. 初始化组件
在 `main` 函数中，我们创建一个 `FileListView` 实例，并添加几个示例项，展示组件的完整效果。

```python
if __name__ == '__main__':
    app = QApplication(sys.argv)
    view = FileListView()
    for i in range(5):
        view.add_item(f'Item {i + 1}', i)
    view.show()
    sys.exit(app.exec_())
```

#### 效果预览
运行后，窗口会显示一个列表，每个项可以通过拖拽排序、点击按钮上移或下移，或者删除，方便用户进行自由操作。

![](https://blog.lc044.love/static/img/82ba082dc03977ae0b3331c2427d57cb.é¢è§.gif)

### 总结
通过本教程，我们实现了一个功能丰富的 PyQt5 自定义列表组件，主要包括以下要点：

1. 自定义小部件，实现项目内容的布局和交互。
2. 使用 `QStandardItem` 和 `QStandardItemModel` 维护列表数据。
3. 通过自定义拖拽和按钮操作，实现上下移动、删除和排序功能。

本组件适用于需要动态排序的项目管理类应用，如文件列表、任务管理等。


## 完整代码
[https://github.com/LC044/pyqt_component_library](https://github.com/LC044/pyqt_component_library)



使用 PyQt5 实现自定义可拖拽列表组件


在日常工作中，处理和合并多个 PDF 文件是一个常见需求，尤其是在需要将大量文件整理成一个完整文档时。本文将详细介绍如何使用 Python 的 `PyMuPDF` 库来实现批量 PDF 文件合并，并提供针对大文件优化的解决方案。

## 安装 PyMuPDF
要使用 PyMuPDF，需要先安装这个库。可以通过以下命令安装：

```bash
pip install pymupdf
```

安装完成后，`PyMuPDF` 将作为 `fitz` 模块导入。

## 基础方法：合并两个 PDF 文件
`PyMuPDF` 中的 `Document.insert_pdf()` 方法支持将一个 PDF 文档的内容插入到另一个文档的指定位置，这样我们就能非常轻松地实现两个 PDF 文件的合并。例如：

```python
import fitz  # 导入 PyMuPDF

def merge_two_pdf(pdf1_path: str, pdf2_path: str, output_pdf: str):
    pdf1 = fitz.open(pdf1_path)
    pdf2 = fitz.open(pdf2_path)
    pdf1.insert_pdf(pdf2)
    pdf1.save(output_pdf, garbage=4)  # 保存结果到指定路径
    pdf1.close()
    pdf2.close()
```

这个函数 `merge_two_pdf` 接受两个 PDF 文件路径，并将它们合并为一个新的 PDF 文件。`garbage=4` 参数可以在保存时移除不必要的对象，从而减小输出文件的大小。

## 批量合并多个 PDF 文件
如果需要合并的 PDF 文件不止两个，可以创建一个循环来遍历文件列表，将每个文件依次添加到目标 PDF 中：

```python
def merge_pdfs(input_files: list, output_file: str):
    final_pdf = fitz.open()  # 创建空的 PDF 对象
    for temp_file in input_files:
        with fitz.open(temp_file) as temp_pdf:
            final_pdf.insert_pdf(temp_pdf)  # 插入每个 PDF 文件
    final_pdf.save(output_file, garbage=4)  # 保存最终合并结果
    final_pdf.close()
    print(f"合并完成，输出文件为：{output_file}")
```

在 `merge_pdfs` 函数中，`input_files` 是一个包含多个 PDF 文件路径的列表，该函数将所有文件合并到一个目标 PDF `output_file` 中。

### 注意事项
当文件数量多、单个 PDF 文件特别大时，上述方法可能会导致内存占用过高。在这种情况下，可以使用以下优化策略来提高内存和时间效率。

## 针对大文件的优化策略
如果需要合并的 PDF 文件很多，或者单个文件特别大，直接合并会占用大量内存。以下是一些优化策略：

1. **分批处理并写入**：大量文件时可以分批加载和合并，避免一次性加载所有文件。例如，处理一组文件后就保存一次部分结果并释放内存。
2. **避免逐页插入**：对于多页文件，可以直接将整个 PDF 插入到目标文档中，而不是逐页插入，从而减少 `insert_pdf` 的调用次数。
3. **多次保存分批结果，避免中间缓存**：每处理完一组文件后保存为一个中间文件，继续合并下一组文件。最终将所有中间文件合并。
4. **利用磁盘而非内存**：采用文件缓存方式，将中间文件写入磁盘。合并完成后，删除不再需要的中间文件。

### 优化代码示例
以下代码展示了如何分批合并 PDF 文件，适合大文件和文件数非常多的情况：

```python
import os
import fitz  # 导入 PyMuPDF

def merge_two_pdf(pdf1_path: str, pdf2_path: str, output_pdf: str):
    """
    合并 pdf1 和 pdf2 输出到 output_pdf，并删除临时文件
    """
    pdf1 = fitz.open(pdf1_path)
    pdf2 = fitz.open(pdf2_path)
    pdf1.insert_pdf(pdf2)
    pdf1.save(output_pdf, garbage=4)  # 保存结果
    pdf1.close()
    pdf2.close()
    if pdf1_path.endswith('.tmp'):
        os.remove(pdf1_path)
    if pdf2_path.endswith('.tmp'):
        os.remove(pdf2_path)

def merge_pdfs(input_files: list, output_file: str):
    tmp_count = 0
    while len(input_files) > 2:
        merged_files = []
        for i in range(0, len(input_files), 2):
            if i + 1 < len(input_files):
                tmp_file = f"{output_file}{tmp_count}.tmp"
                merge_two_pdf(input_files[i], input_files[i + 1], tmp_file)
                merged_files.append(tmp_file)
                tmp_count += 1
            else:
                merged_files.append(input_files[i])
        input_files = merged_files
    merge_two_pdf(input_files[0], input_files[1], output_file)  # 保存最终结果
```

### 代码解析
1. `merge_two_pdf` 函数负责合并两个 PDF 文件，并在合并后删除临时文件。
2. `merge_pdfs` 函数使用循环批量合并文件。每次迭代中，它将文件成对合并为一个临时文件，添加到列表 `merged_files`，以便下一次迭代使用。
3. 最终，合并后的文件保存在 `output_file` 中。

### 示例调用
假设我们有一组 PDF 文件 `file1.pdf`, `file2.pdf`, `file3.pdf` 等：

```python
input_files = ["file1.pdf", "file2.pdf", "file3.pdf", "file4.pdf"]
output_file = "merged_output.pdf"
merge_pdfs(input_files, output_file)
```

## 总结
本文介绍了使用 `PyMuPDF` 库批量合并多个 PDF 文件的方法，并针对大文件或大量文件的情况提供了优化方案。

Python批量合并多个PDF

<div style="display:none;" id='thumbnail'><img src="/img/thumbnail.png" alt=""></div>

# Pyqt实现微信气泡消息

最近想做个界面展示聊天记录，奈何一直没有找到用pyqt实现的类似微信的聊天界面，于是就想着自己做一个。

## 特性

1. 自适应大小、支持缩放: 通过重写QLabel并添加合适的布局，实现消息框的自适应大小，并支持缩放。
2. 支持图片显示: 项目支持在聊天框中显示图片消息，为用户提供更加丰富的聊天体验。
3. 支持更换头像: 用户可以轻松更换头像，个性化定制聊天界面。

效果


![result.gif](http://blog.lc044.love/static/img/4b01f97d7a0daeb0e87721b2bff15655.result.gif)

## 实现原理，重写QLabel，添加布局

### 聊天框布局
聊天框由四部分组成：最左侧是头像，右侧是聊天内容，中间绘制一个三角形指向头像，最右侧放置一个弹簧控件来调整消息的长度。


![layout.png](http://blog.lc044.love/static/img/eebcfbf6b312f91ede9df2db1ffffaab.layout.webp)

### 头像的实现
头像通过继承QLabel实现，用户可以传入头像路径或QPixmap对象。头像会根据需要进行缩放，并设置固定的大小。

```python
class Avatar(QLabel):
    def __init__(self, avatar, parent=None):
        super().__init__(parent)
        if isinstance(avatar, str):
            self.setPixmap(QPixmap(avatar).scaled(45, 45))
            self.image_path = avatar
        elif isinstance(avatar, QPixmap):
            self.setPixmap(avatar.scaled(45, 45))
        self.setFixedSize(QSize(45, 45))
```
### 绘制三角形
通过继承QLabel实现一个三角形，用于指向头像的方向。根据消息是发送还是接收，绘制不同方向的三角形。

```python
class Triangle(QLabel):
    def __init__(self, Type, is_send=False, parent=None):
        super().__init__(parent)
        self.Type = Type
        self.is_send = is_send
        self.setFixedSize(6, 45)

    def paintEvent(self, a0: QtGui.QPaintEvent) -> None:
        super(Triangle, self).paintEvent(a0)
        if self.Type == MessageType.Text:
            painter = QPainter(self)
            triangle = QPolygon()
            if self.is_send:
                painter.setPen(QColor('#b2e281'))
                painter.setBrush(QColor('#b2e281'))
                triangle.setPoints(0, 20, 0, 35, 6, 25)
            else:
                painter.setPen(QColor('white'))
                painter.setBrush(QColor('white'))
                triangle.setPoints(0, 25, 6, 20, 6, 35)
            painter.drawPolygon(triangle)
```

### 文字消息控件
通过继承QLabel实现文字消息控件，支持自动换行、文字选择和不同样式的边框。

```python
class TextMessage(QLabel):
    heightSingal = pyqtSignal(int)

    def __init__(self, text, is_send=False, parent=None):
        super(TextMessage, self).__init__(text, parent)
        font = QFont('微软雅黑', 12)
        self.setFont(font)
        self.setWordWrap(True)
        self.setMaximumWidth(800)
        self.setMinimumWidth(100)
        self.setMinimumHeight(45)
        self.setTextInteractionFlags(Qt.TextSelectableByMouse)
        self.setSizePolicy(QSizePolicy.Ignored, QSizePolicy.Ignored)
        if is_send:
            self.setAlignment(Qt.AlignCenter | Qt.AlignRight)
            self.setStyleSheet(
                '''
                background-color:#b2e281;
                border-radius:10px;
                padding:10px;
                '''
            )
        else:
            self.setStyleSheet(
                '''
                background-color:white;
                border-radius:10px;
                padding:10px;
                '''
            )
        font_metrics = QFontMetrics(font)
        rect = font_metrics.boundingRect(text)
        self.setMaximumWidth(rect.width()+30)
```

核心是边框样式的设置，self.setWordWrap(True)设置文字自动换行，self.setTextInteractionFlags(Qt.TextSelectableByMouse)设置文字可选中，self.setAlignment(Qt.AlignCenter | Qt.AlignRight)设置文字对齐方式

```css
{
    background-color:#b2e281; // 边框颜色
    border-radius:10px;  // 边框圆角半径
    padding:10px;
}
```

### 聊天框组件

最后，将头像、三角形和消息控件组合在一起，通过水平布局排列，形成完整的聊天框组件。

```python
class BubbleMessage(QWidget):
    def __init__(self, str_content, avatar, Type, is_send=False, parent=None):
        super().__init__(parent)
        self.isSend = is_send
        # self.set
        self.setStyleSheet(
            '''
            border:none;
            '''
        )
        layout = QHBoxLayout()
        layout.setSpacing(0)
        layout.setContentsMargins(0, 5, 5, 5)
        # self.resize(QSize(200, 50))
        self.avatar = Avatar(avatar)
        triangle = Triangle(Type, is_send)
        if Type == MessageType.Text:
            self.message = TextMessage(str_content, is_send)
            # self.message.setMaximumWidth(int(self.width() * 0.6))
        elif Type == MessageType.Image:
            self.message = ImageMessage(str_content)
        else:
            raise ValueError("未知的消息类型")

        self.spacerItem = QSpacerItem(45 + 6, 45, QSizePolicy.Expanding, QSizePolicy.Minimum)
        if is_send:
            layout.addItem(self.spacerItem)
            layout.addWidget(self.message, 1)
            layout.addWidget(triangle, 0, Qt.AlignTop | Qt.AlignLeft)
            layout.addWidget(self.avatar, 0, Qt.AlignTop | Qt.AlignLeft)
        else:
            layout.addWidget(self.avatar, 0, Qt.AlignTop | Qt.AlignRight)
            layout.addWidget(triangle, 0, Qt.AlignTop | Qt.AlignRight)
            layout.addWidget(self.message, 1)
            layout.addItem(self.spacerItem)
        self.setLayout(layout)
```

## 最终效果


![demo.png](http://blog.lc044.love/static/img/1bb12bf62d1784f5b12f7f207450c2d5.demo.webp)

应用到项目中去:[微信聊天记录备份](https://github.com/LC044/WeChatMsg)


![chat_.png](http://blog.lc044.love/static/img/592dab276ec6101e0d972c1780572e13.chat_.webp)

# 开源地址

[https://github.com/LC044/pyqt_component_library](https://github.com/LC044/pyqt_component_library)

PyQt实现类似微信气泡消息

看什么看

👨🏻‍💻这是一个充满知识💬的链接，快点进来看看你的好友给你分享了什么干货吧❣️