Python完成SVG转PNG格式——方法二

pip install cairosvg

cairosvg xx.svg -o xx.png

# -*- coding: utf-8 -*-
# @Time    : 2021/7/15 14:23
# @Author  : Elsa
# @File    : SvgToPng.py

import os
import cairosvg
from PySide2.QtCore import QStringListModel
from PySide2.QtGui import QPixmap
from PySide2.QtUiTools import QUiLoader
from PySide2.QtWidgets import QApplication, QFileDialog, QMessageBox


class Stats:
    # 初始化
    def __init__(self):
        # 动态加载ui文件
        self.ui = QUiLoader().load('SvgToPng.ui')

        # 信号与槽链接
        self.ui.btn_file.clicked.connect(self.ChangeOne)
        self.ui.btn_files.clicked.connect(self.ChangeMore)
        self.ui.listView.clicked.connect(self.ShowMore)

        # 变量定义
        self.file_name = ''  # 选择的文件/文件夹
        self.save_file_name = ''  # 最终保存的文件夹位置
        self.image_name = ''

        self.file_paths = []  # 文件列表
        self.file_index = 0  # 文件索引
        self.file_num = ''  # 文件个数

    # 选择单个文件转换
    def ChangeOne(self):
        # 选择svg格式的文件
        self.file_name = QFileDialog.getOpenFileName(None, "选择文件", "/", "*.svg")
        self.file_name = str(self.file_name).split(',')[0].strip('(').replace("'", "")
        if self.file_name == '':
            QMessageBox.warning(
                self.ui,
                '提示',
                '未打开文件！')
        # 将打开文件路径显示在文本框中
        self.ui.edit_svg.setText(self.file_name)
        # 得到源文件位置
        self.image_name = self.file_name
        # 转换格式
        self.SvgToPng()
        # 显示转换后的png文件路径
        slm = QStringListModel()
        show = [self.save_file_name, '']
        slm.setStringList(show)
        self.ui.listView.setModel(slm)
        # 预览
        self.ShowOne()

    # 选择文件夹转换
    def ChangeMore(self):
        # 选择文件夹
        self.file_name = QFileDialog.getExistingDirectory(None, '选择文件夹', '/')  # 返回选中的文件夹路径
        # 将打开的文件夹路径显示在文本框中
        self.ui.edit_svg.setText(self.file_name)
        # 读取文件夹的文件
        self.file_paths.clear()
        for root, dirs, files in os.walk(self.file_name, topdown=False):
            for file in files:
                self.file_paths.append(os.path.join(root, file))
        line_name = str(self.file_paths[0]).replace('\\', '/').split('/')[-1]
        # 读取出来的列表，第一个数据是一个系统文件，可能有可能无，所以判断一下
        if line_name == 'desktop.ini':
            self.file_paths = self.file_paths[1:]
        # 列表长度，统计有多少个文件
        self.file_num = len(self.file_paths)
        if self.file_num <= 0:
            QMessageBox.warning(
                self.ui,
                '提示',
                '文件夹为空！')
            return
        # 批量转换
        for self.file_index in range(0, self.file_num):
            self.image_name = self.file_paths[self.file_index].replace('\\', '/')
            # 处理文件
            self.SvgToPng()
            self.file_paths[self.file_index] = self.save_file_name
        # 控制右侧的列表仅展示png文件
        self.file_index = 0
        for self.file_index in range(0, int(self.file_num / 2)):
            image_name_now = self.file_paths[self.file_index].replace('\\', '/').split('/')[-1].split('.')[-1]
            if image_name_now == 'svg':
                del self.file_paths[self.file_index]

        # 去掉列表中的重复项
        self.file_num = int(len(self.file_paths) / 2)
        for i in range(0, self.file_num):
            for j in range(0, self.file_num):
                png_name_one = self.file_paths[i].replace('\\', '/').split('/')[-1].split('.')[0]
                png_name_two = self.file_paths[j].replace('\\', '/').split('/')[-1].split('.')[0]
                if png_name_one == png_name_two:
                    del self.file_paths[i]
        # 先预览最后一张
        self.ShowOne()
        # 显示图片列表
        slm = QStringListModel()
        slm.setStringList(self.file_paths)
        self.ui.listView.setModel(slm)

    # 格式转换
    def SvgToPng(self):
        # 获取当前文件的后缀名
        image_name_now = self.image_name.split('/')[-1].split('.')[-1]
        if image_name_now == 'svg':
            # 设置保存的文件名为“./名字_1.png”
            image_png_name = self.image_name.split('/')[-1]
            self.save_file_name = self.image_name.strip(image_png_name) + image_png_name.strip('.svg') + '_1.png'
            cairosvg.svg2png(file_obj=open(self.image_name, "rb"), write_to=self.save_file_name)

    # 展示图片
    def ShowOne(self):
        # 预览png图标
        pix = QPixmap(self.save_file_name)
        self.ui.label_print.setPixmap(pix)
        # 图片自适应
        self.ui.label_print.setScaledContents(True)

    # 列表展示图片
    def ShowMore(self, qModelIndex):
        self.save_file_name = self.file_paths[qModelIndex.row()]
        self.ShowOne()


if __name__ == '__main__':
    app = QApplication([])
    app.setStyle('Fusion')  # 设置窗口显示的风格
    stats = Stats()
    stats.ui.show()
    app.exec_()