image_cutout/module/module_resize_pic.py

# -*- coding: utf-8 -*-
import time
import sys
import os
from UI.resize_photos import Ui_Form as ResizePicUI
from PySide6.QtWidgets import *
from PySide6.QtCore import *
from PIL import Image


class ResizePic(QWidget, ResizePicUI):
    signal_data = Signal(dict)

    def __init__(self, windows=None):
        super().__init__()
        self.windows = windows
        self.setupUi(self)
        self.setFixedSize(self.width(), self.height())
        self.allow_modified = True
        self.pic_width_target = 10000
        self.state = 0
        self.show_img_dir = ""
        self.change_state(self.state)
        self.init()
        self.setWindowModality(Qt.ApplicationModal)
        self.show()

    def change_state(self, index=0):
        if index == 0:
            self.progressBar.hide()
            self.label_5.hide()
            self.pushButton.setText("执行压缩")
            self.pushButton.setEnabled(True)
        if index == 1:
            self.progressBar.show()
            self.progressBar.setValue(0)
            self.textBrowser_2.setText("")
            self.label_5.show()
            self.pushButton.setText("进行中")
            self.pushButton.setEnabled(False)

    def init(self):
        self.label_4.setText("")
        self.label_8.hide()
        self.label_3.mousePressEvent = self.change_img_dir
        self.label_8.mousePressEvent = self.open_dir
        self.signal_data.connect(self.show_info)
        self.pushButton.clicked.connect(self.run)

    def open_dir(self, *args, **kwargs):
        if self.show_img_dir:
            os.startfile(self.show_img_dir)

    def run(self, is_sure=False):
        if not self.lineEdit.text():
            QMessageBox.question(self, '确认', '请输入目标像素', QMessageBox.Yes)
            return
        else:
            try:
                self.pic_width_target = int(self.lineEdit.text())
                if self.pic_width_target < 200 or self.pic_width_target > 2000:
                    QMessageBox.question(self, '确认', '建议像素在800px左右', QMessageBox.Yes)
                    return
            except:
                QMessageBox.question(self, '确认', '像素必须是整数', QMessageBox.Yes)
                return

        if not is_sure:
            if not self.show_img_dir:
                QMessageBox.question(self, '确认', '请选择图片文件夹', QMessageBox.Yes)
                return

            a = QMessageBox.question(self, '确认', '是否确认继续执行，软件会自动处理所有子文件夹，请自行备份数据，执行后无法恢复',
                                     QMessageBox.Yes | QMessageBox.No)
            if a != QMessageBox.Yes:
                return

            # 计算待处理图片数量
            total_pic = 0
            _Type = ['.png', '.PNG', '.jpg', '.JPG', '.gif', '.GIF', '.jpeg', '.JPEG']
            for dirpath, dirnames, filenames in os.walk(self.show_img_dir):
                for file in filenames:
                    if os.path.splitext(file)[1] in _Type:
                        total_pic += 1

            if not total_pic:
                QMessageBox.question(self, '确认', '没有任何图片', QMessageBox.Yes)
                return

            a = QMessageBox.question(self, '确认', '待处理图片数：{}，是否继续'.format(total_pic),
                                     QMessageBox.Yes | QMessageBox.No)
            if a == QMessageBox.Yes:
                self.run(is_sure=True)
        else:
            self.deal_pic()

    def show_info(self, data):
        if data["_type"] == "show_schedule":
            self.progressBar.setValue(data["data"])

        if data["_type"] == "show_text":
            self.textBrowser_2.append(data["data"])

        if data["_type"] == "change_state":
            self.change_state(data["data"])

    def deal_pic(self):
        total_pic = 0
        _Type = ['.png', '.PNG', '.jpg', '.JPG', '.gif', '.GIF', '.jpeg', '.JPEG']
        for dirpath, dirnames, filenames in os.walk(self.show_img_dir):
            for file in filenames:
                if os.path.splitext(file)[1] in _Type:
                    total_pic += 1

        self.signal_data.emit({"_type": "change_state",
                               "data": 1})

        self.signal_data.emit({"_type": "show_schedule",
                               "data": 0})

        n = 0
        _Type = ['.png', '.PNG', '.jpg', '.JPG', '.gif', '.GIF', '.jpeg', '.JPEG']
        for dirpath, dirnames, filenames in os.walk(self.show_img_dir):
            for file in filenames:
                if os.path.splitext(file)[1] in _Type:
                    n += 1
                    file_path = dirpath + "/" + file
                    self.signal_data.emit({"_type": "show_schedule",
                                           "data": (n / total_pic * 100)})
                    self.signal_data.emit({"_type": "show_text",
                                           "data": "正在处理：{}".format(file)})
                    print(file_path)
                    pic = Picture(file_path)
                    if pic.x > self.pic_width_target:
                        pic.resize(self.pic_width_target)
                        pic.save_img(file_path)
        self.signal_data.emit({"_type": "show_text",
                               "data": "已完成"})
        self.signal_data.emit({"_type": "change_state",
                               "data": 0})

    def change_img_dir(self, *args):
        folder = QFileDialog.getExistingDirectory(self, "选取文件夹", "./")
        print(folder)
        self.show_img_dir = folder
        self.label_4.setText(self.show_img_dir)
        if self.show_img_dir:
            self.label_8.show()
        else:
            self.label_8.hide()


class Picture:

    def __init__(self, in_path):
        self.im = Image.open(in_path)
        self.x, self.y = self.im.size
        # print(self.x, self.y)

    def save_img(self, outpath, quality=90):
        if outpath.endswith("jpg"):
            if self.im.mode != 'RGB':
                self.im = self.im.convert("RGB")
                self.im.save(outpath,format="JPEG")
                return

        if outpath.endswith("png"):
            if self.im.mode != 'RGBA':
                self.im = self.im.convert("RGBA")
            self.im.save(outpath)
            return
        self.im.save(outpath)

    def resize(self, width):
        re_x = int(width)
        re_y = int(self.y * re_x / self.x)
        self.im = self.im.resize((re_x, re_y), Image.Resampling.LANCZOS)
        self.x, self.y = self.im.size

    def resize_regular(self, width, high):
        self.im = self.im.resize((width, high), Image.Resampling.LANCZOS)