在日常的照片管理中，我们常会遇到 EXIF 信息中的拍摄日期不准确或者缺失的情况。有些图片（截图、程序生成、去除隐私的拍摄照片）的元数据中没有图片的拍摄日期，从网盘或者其他地方下载的图片的创建日期会变成下载时候的日期，导入相册中无法正确识别到图片的日期，导致相册按时间混乱。

这些图片的文件名大都含有时间信息，这篇文章将分享一个完整的 Python 脚本，用于根据图片文件名自动提取日期，并将其写入图片的 EXIF 元数据中，便于照片管理和排序。

---

## 主要功能
这段代码实现了以下功能：

1. **提取文件名中的日期时间**：支持多种常见的日期格式，包括时间戳。
2. **校验日期有效性**：确保提取到的日期在合理范围内（如 1970 年到 2035 年之间）。
3. **修改 EXIF 元数据**：将提取的日期写入 `DateTimeOriginal` 字段，并设置其他相关字段。
4. **灵活的修改逻辑**：可选择强制使用给定日期，或者自动选择最早日期进行更新。

---

## 核心代码解析
### 1. 提取文件名中的日期
代码首先通过正则表达式从文件名中提取日期时间，支持的格式包括：

+ `YYYYMMDD_HHMMSS` 格式，例如 `20201231_200936`
+ 时间戳格式（10 位秒级或 13 位毫秒级时间戳）
+ 混合格式，例如 `YYYYMMDD-HHMMSS` 或 `YYYYMMDD_HHMMSS[附加内容]`

以下是提取日期时间的核心代码：

```python
def extract_datetime_from_filename(filename) -> datetime | None:
    patterns = [
        r"(\d{8})_(\d{6})",  # 格式：YYYYMMDD_HHMMSS
        r"(\d{13}|\d{10})",  # 格式：TIMESTAMP
        r"(\d{8})[_|-](\d{6}).*?"  # 格式：YYYYMMDD_HHMMSS[其他内容]
    ]
    for pattern in patterns:
        match = re.search(pattern, filename)
        if match:
            if len(match.groups()) == 2:
                date_part, time_part = match.groups()
                try:
                    return datetime.strptime(date_part + time_part, "%Y%m%d%H%M%S")
                except ValueError:
                    continue
            elif len(match.groups()) == 1:
                timestamp = match.group(1)
                try:
                    if len(timestamp) == 13:
                        return datetime.fromtimestamp(int(timestamp) / 1000)
                    else:
                        return datetime.fromtimestamp(int(timestamp))
                except ValueError:
                    continue
    return None
```

### 2. 校验日期的有效性
提取到的日期可能会超出合理范围（例如系统错误生成的无效时间戳）。因此，我们用以下代码确保日期在合理范围内：

```python
start_time = datetime(1970, 1, 1, 0, 0, 0)
end_time = datetime(2035, 1, 1, 0, 0, 0)

def valid_time(dt):
    if start_time <= dt <= end_time:
        return dt
    else:
        return None
```

### 3. 修改 EXIF 信息
代码使用了 [piexif](https://github.com/hMatoba/Piexif) 库来加载和修改图片的 EXIF 元数据。以下是核心逻辑：

+ 获取 `DateTimeOriginal`，与提取的日期比较，取较早者。
+ 如果 EXIF 数据不存在，则创建并插入新的日期信息。
+ 更新 EXIF 中的品牌信息（如 `Make` 和 `Model`）。

以下是主要代码段：

```python
def modify_image(image_file: ImageFile, given_date, is_force):
    try:
        exif_dict = piexif.load(image_file.file_path)
        date_time_original = get_exif_date(exif_dict, piexif.ExifIFD.DateTimeOriginal, 'Exif', None)
        if is_force:
            dates = [d for d in [date_time_original, given_date] if d is not None]
            earliest_date = min(dates)
        else:
            earliest_date = given_date

        if date_time_original != earliest_date:
            exif_dict["Exif"][piexif.ExifIFD.DateTimeOriginal] = earliest_date.strftime("%Y:%m:%d %H:%M:%S").encode("utf-8")
            exif_dict["0th"][piexif.ImageIFD.Make] = 'MemoTrace'.encode("utf-8")
            exif_dict["0th"][piexif.ImageIFD.Model] = 'EasyBox'.encode("utf-8")

        exif_bytes = piexif.dump(exif_dict)
        piexif.insert(exif_bytes, image_file.save_path)
        print(f"DateTimeOriginal 已更新为最早日期：{earliest_date}")
    except Exception as e:
        print(f"处理图片时出错：{e} {traceback.format_exc()}")
```

### 4. 示例运行
以下是代码的运行示例：

```python
if __name__ == '__main__':
    img_path = './dist/图片/Screenshot_20201231_200936.jpg'
    new_path = 'IMG_20201007_150144(3).jpg'
    image_file = ImageFile(img_path)
    image_file.save_path = new_path
    modify_image(image_file, None, False)
```

运行结果如下：

```plain
DateTimeOriginal 已更新为最早日期：2020-12-31 20:09:36
```

![image.png](https://blog.lc044.love/static/img/01192a9a40403b39489a47bcf663c3cf.image.webp)
---

## 运行环境和依赖
+ Python 版本：`>=3.9`
+ 依赖库：`piexif`

安装依赖：

```bash
pip install piexif
```

---

## 完整代码
完整代码如下：

```python
"""
根据文件名获取图片的日期，修改图片exif元数据里的拍摄日期
"""

import os.path
import re
import shutil
import traceback
from datetime import datetime

import piexif


def get_exif_date(exif_dict, field, key='Exif', default="Unknown"):
    """
    安全获取 EXIF 字段的日期，如果字段不存在或解析失败，返回默认值。
    """
    try:
        date_str = exif_dict[key].get(field, b"").decode("utf-8")
        return datetime.strptime(date_str, "%Y:%m:%d %H:%M:%S")
    except (KeyError, ValueError, UnicodeDecodeError):
        return default


start_time = datetime(1970, 1, 1, 0, 0, 0)
end_time = datetime(2035, 1, 1, 0, 0, 0)


def valid_time(dt):
    if start_time <= dt <= end_time:
        return dt
    else:
        return None


def extract_datetime_from_filename(filename) -> datetime | None:
    # 定义正则表达式，匹配常见的时间格式
    patterns = [
        r"(\d{8})_(\d{6})",  # 格式：YYYYMMDD_HHMMSS
        r"(\d{13}|\d{10})",  # 格式：TIMESTAMP (13位毫秒级时间戳)
        r"(\d{8})[_|-](\d{6}).*?"  # 格式：YYYYMMDD_HHMMSS[其他内容]
    ]
    result = None
    for pattern in patterns:
        match = re.search(pattern, filename)
        if match:
            if len(match.groups()) == 2:  # 包含日期和时间
                date_part, time_part = match.groups()
                try:
                    result = datetime.strptime(date_part + time_part, "%Y%m%d%H%M%S")
                    return valid_time(result)
                except ValueError:
                    continue
            elif len(match.groups()) == 1:  # 只包含时间戳
                timestamp = match.group(1)
                try:
                    if len(timestamp) == 13:
                        result = datetime.fromtimestamp(int(timestamp) / 1000)  # 转换为秒
                    else:
                        result = datetime.fromtimestamp(int(timestamp))  # 转换为秒
                    return valid_time(result)
                except (ValueError, OverflowError):
                    continue
    return None


class ImageFile:
    def __init__(self, file_path):
        self.save_path = file_path
        self.file_path = file_path
        self.file_name = os.path.basename(file_path)
        self.save_inplace = True
        self.save_fmt = 'auto'
        self.save_quality = 95

    def get_file_time_by_name(self):
        return extract_datetime_from_filename(self.file_name)


def modify_image(image_file: ImageFile, given_date, is_force):
    """
    修改图片的拍摄时间为given_date
    :param image_file:
    :param given_date:
    :param is_force: 是否强制设置为given_date,否：取当前图片拍摄时间和给定时间的最小值
    :return:
    """
    file_path = image_file.file_path
    if not given_date:
        given_date = image_file.get_file_time_by_name()
    try:
        # 打开图片并加载 EXIF 数据
        try:
            exif_dict = piexif.load(file_path)
            # 获取 EXIF 日期
            date_time_original = get_exif_date(exif_dict, piexif.ExifIFD.DateTimeOriginal, 'Exif', None)
            if is_force:
                modify_date = get_exif_date(exif_dict, piexif.ImageIFD.DateTime, '0th', None)
                dates = [d for d in [date_time_original, modify_date, given_date] if d is not None]
                earliest_date = min(dates)
            else:
                earliest_date = given_date
            # 如果 DateTimeOriginal 已经是最早的，则无需修改
            if date_time_original != earliest_date:
                # 更新 DateTimeOriginal
                exif_dict["Exif"][piexif.ExifIFD.DateTimeOriginal] = earliest_date.strftime("%Y:%m:%d %H:%M:%S").encode(
                    "utf-8")
                exif_dict["0th"][piexif.ImageIFD.Make] = 'MemoTrace'.encode("utf-8")
                exif_dict["0th"][piexif.ImageIFD.Model] = 'EasyBox'.encode("utf-8")
                exif_dict["0th"][piexif.ImageIFD.Software] = 'EasyBox-0.1.0'.encode("utf-8")
        except:
            earliest_date = given_date
            exif_dict = {
                '0th': {},
                'Exif': {}
            }
            if earliest_date:
                # 更新 DateTimeOriginal
                exif_dict["Exif"][piexif.ExifIFD.DateTimeOriginal] = earliest_date.strftime("%Y:%m:%d %H:%M:%S").encode(
                    "utf-8")
            exif_dict["0th"][piexif.ImageIFD.Make] = 'MemoTrace'.encode("utf-8")
            exif_dict["0th"][piexif.ImageIFD.Model] = 'EasyBox'.encode("utf-8")
            exif_dict["0th"][piexif.ImageIFD.Software] = 'EasyBox-0.1.0'.encode("utf-8")
        # 保存修改后的图片
        exif_bytes = piexif.dump(exif_dict)
        # img.save(image_file.save_path, exif=exif_bytes, quality=image_file.save_quality, subsampling=0)
        if image_file.save_path != image_file.file_path:
            shutil.copy(file_path, image_file.save_path)
        piexif.insert(exif_bytes, image_file.save_path)
        print(f"DateTimeOriginal 已更新为最早日期：{earliest_date}")
    except Exception as e:
        print(f"处理图片时出错：{e} {traceback.format_exc()}")


if __name__ == '__main__':
    img_path = './dist/图片/Screenshot_20201231_200936.jpg'
    new_path = 'IMG_20201007_150144(3).jpg'
    given_date = None
    image_file = ImageFile(img_path)
    image_file.save_path = new_path
    modify_image(image_file, None, False)

```

---

## 下载

软件下载地址 [https://github.com/LC044/EasyBox](https://github.com/LC044/EasyBox)

## 总结
这段脚本通过文件名自动提取日期，并写入图片的 EXIF 数据，简化了拍摄时间的修正流程。希望这篇文章对你管理照片有所帮助！如果你在使用过程中有任何问题或建议，欢迎留言讨论。

相关应用程序正在开发中，欢迎关注以便获取最新资源


![image.png](https://blog.lc044.love/static/img/f2756d8910cdb9d7410d0fb7df8d216a.image.webp)



基于文件名修改图片的拍摄日期

## 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实现类似微信气泡消息

看什么看

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