"""
艺术二维码生成器

@author  : zhouhuajian
@version : v1.0
"""
import time

from PySide6.QtCore import QFile
from PySide6.QtGui import QMouseEvent, Qt, QPixmap, QMovie
from PySide6.QtWidgets import QWidget, QApplication, QFileDialog, QMessageBox

from main_window_ui import Ui_MainWindow

from amzqr import amzqr


class ArtisticQrcodeGenerator(QWidget):
    """艺术二维码生成器"""

    def __init__(self):
        """初始化"""
        super().__init__()
        # 主界面UI
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        # 修改背景图片
        self.ui.backgroundImageLabel.mousePressEvent = self.changeBackgroundImage
        self.backgroundImagePath = "images/default_background.png"
        # 创建二维码
        self.ui.pushButton.clicked.connect(self.createQrcode)
        # 保存二维码
        self.qrcodePath = "images/default_qrcode.png"
        self.ui.qrcodeLabel.mousePressEvent = self.saveQrcode
        self.file = QFile()

    def saveQrcode(self, mouseEvent: QMouseEvent):
        """保存二维码"""
        if mouseEvent.button() != Qt.MouseButton.LeftButton:
            return
        ext = self.qrcodePath[-3:]  # jpg png gif
        r = QFileDialog.getSaveFileName(parent=self,
                                        caption="保存二维码",
                                        dir=f"二维码_{time.strftime('%Y%m%d%H%M%S')}.{ext}",
                                        filter=f"图片 (*.{ext})")
        imagePath = r[0]
        if not imagePath:
            return
        print("二维码保存到了" + imagePath)
        self.file.copy(self.qrcodePath, imagePath)
        QMessageBox.information(self, "提示", "二维码保存成功")

    def createQrcode(self):
        """创建二维码"""
        print("开始创建二维码")
        try:
            r = amzqr.run(
                self.ui.plainTextEdit.toPlainText(),
                version=10,
                picture=self.backgroundImagePath,
                colorized=True,
                save_name="临时二维码." + self.backgroundImagePath[-3:],
                save_dir="./temp"
            )
            self.qrcodePath = r[2]

            # 静态图片
            if self.qrcodePath[-3:] != 'gif':
                self.ui.qrcodeLabel.setPixmap(QPixmap(self.qrcodePath))
            else:
                movie = QMovie(self.qrcodePath)
                self.ui.qrcodeLabel.setMovie(movie)
                movie.start()
        except Exception as e:
            error_message = str(e)
            if error_message == 'Wrong words! Make sure the characters are supported!':
                error_message = "创建失败，目前只支持以下字符\n" + r"0-9 A-Z a-z ··,.:;+-*/\~!@#$%^&`'=<>[]()?_{}|"
            QMessageBox.critical(self, "创建失败", error_message)

    def changeBackgroundImage(self, mouseEvent: QMouseEvent):
        """修改背景图片"""
        if mouseEvent.button() != Qt.MouseButton.LeftButton:
            return
        r = QFileDialog.getOpenFileName(parent=self,
                                        caption="选择背景图片",
                                        filter="图片 (*.jpg *.png *.gif)")
        imagePath = r[0]
        if not imagePath:
            return
        print("背景图片修改为" + imagePath)
        self.backgroundImagePath = imagePath

        # 静态图片
        if self.backgroundImagePath[-3:] != 'gif':
            self.ui.backgroundImageLabel.setPixmap(QPixmap(self.backgroundImagePath))
        else:
            movie = QMovie(self.backgroundImagePath)
            self.ui.backgroundImageLabel.setMovie(movie)
            movie.start()


if __name__ == '__main__':
    # 应用
    app = QApplication()
    qrcodeGenerator = ArtisticQrcodeGenerator()
    qrcodeGenerator.show()
    # 进入消息循环
    app.exec()