基于Python制作一个图片色卡提取器

pip install PyQt5 -i http://pypi.tuna.tsinghua.edu.cn/simple/

pip install extcolors -i http://pypi.tuna.tsinghua.edu.cn/simple/

# It's a module that allows you to print the stack trace of an exception.
import traceback

# It's a module that allows you to print the stack trace of an exception.
import extcolors

# It's importing all the classes from the QtWidgets module.
from PyQt5.QtWidgets import *

# It's importing all the classes from the QtGui module.
from PyQt5.QtGui import *

# It's importing all the classes from the QtCore module.
from PyQt5.QtCore import *

# It's importing the sys module.
import sys

# It's importing the os module.
import os

class ColorUI(QWidget):
    def __init__(self):
        """
        A constructor. It is called when an object is created from a class and it allows the class to initialize the
        attributes of a class.
        """
        super(ColorUI, self).__init__()
        self.init_ui()

    def init_ui(self):
        """
        This function initializes the UI.
        """
        self.setWindowTitle('图片颜色提取器 公众号：Python 集中营')
        self.setWindowIcon(QIcon('color.ico'))
        self.resize(500, 300)

        self.image_label = QLabel()
        self.image_label.setMinimumWidth(300)
        self.image_label.setMaximumHeight(300)
        self.image_label.setText('公众号：Python 集中营')
        self.image_label.setAlignment(Qt.AlignCenter)
        self.image_label.setStyleSheet('font-size:20px;color:blue;')
        self.image_label.setScaledContents(True)

        self.image_path_in = QLineEdit()
        self.image_path_in.setPlaceholderText('源图片路径')
        self.image_path_in.setReadOnly(True)

        self.image_path_btn = QPushButton()
        self.image_path_btn.setText('加载源图片')
        self.image_path_btn.clicked.connect(self.image_path_btn_click)

        self.set_color_num_label = QLabel()
        self.set_color_num_label.setText('设置提取色卡数量：')

        self.set_color_num_in = QLineEdit()
        self.set_color_num_in.setPlaceholderText('例如：10')

        self.start_btn = QPushButton()
        self.start_btn.setText('开始提取颜色')
        self.start_btn.clicked.connect(self.start_btn_click)

        self.brower = QTextBrowser()
        self.brower.setReadOnly(True)
        self.brower.setFont(QFont('宋体', 8))
        self.brower.setPlaceholderText('日志处理过程区域...')
        self.brower.ensureCursorVisible()

        hbox = QHBoxLayout()

        left_box = QVBoxLayout()
        right_box = QVBoxLayout()

        left_box.addWidget(self.image_label)
        right_form_box = QFormLayout()
        right_form_box.addRow(self.image_path_in, self.image_path_btn)
        right_form_box.addRow(self.set_color_num_label, self.set_color_num_in)
        right_form_box.addRow(self.start_btn)
        right_box.addLayout(right_form_box)
        right_box.addWidget(self.brower)

        hbox.addLayout(left_box)
        hbox.addLayout(right_box)

        self.thread_ = ColorWork(self)
        self.thread_.message.connect(self.show_message)

        self.setLayout(hbox)

    def show_message(self, text):
        """
        It shows a message

        :param text: The text to be displayed
        """
        cursor = self.brower.textCursor()
        cursor.movePosition(QTextCursor.End)
        self.brower.append(text)
        self.brower.setTextCursor(cursor)
        self.brower.ensureCursorVisible()

    def start_btn_click(self):
        """
        A function that is called when the start button is clicked.
        """
        self.thread_.start()

    def image_path_btn_click(self):
        """
        It opens a file dialog box to select the image file.
        """
        path = QFileDialog.getOpenFileName(self, "选取文件", os.getcwd(), "Image File (*.jpg);;Image File (*.png)")
        self.image_path_in.setText(path[0])
        pixmap = QPixmap(path[0])
        self.image_label.clear()
        self.image_label.setPixmap(pixmap)

class ColorWork(QThread):
    message = pyqtSignal(str)

    def __init__(self, parent=None):
        """
        A constructor that initializes the class.

        :param parent: The parent widget
        """
        super(ColorWork, self).__init__(parent)
        self.working = True
        self.parent = parent

    def __del__(self):
        """
        A destructor. It is called when the object is destroyed.
        """
        self.working = False

    def run(self):
        """
        *|CURSOR_MARCADOR|*
        """
        try:
            image_path_in = self.parent.image_path_in.text().strip()
            set_color_num_in = self.parent.set_color_num_in.text().strip()
            if image_path_in == '' or set_color_num_in == '':
                self.message.emit('系统参数设置不能为空，请检查参数设置！')
                return
            colors_x = extcolors.extract_from_path(image_path_in, tolerance=12, limit=int(set_color_num_in))

            for turple_ in colors_x[0]:
                rgb_ = turple_[0]
                color_16 = ('{:02X}' * 3).format(rgb_[0], rgb_[1], rgb_[2])
                color_16 = ('#' + color_16)
                self.message.emit(color_16)
        except:
            traceback.print_exc()
            self.message.emit('系统运行出现错误，请检查相关参数是否正确！')

if __name__ == '__main__':
    app = QApplication(sys.argv)
    main = ColorUI()
    main.show()
    sys.exit(app.exec_())