from base64 import b64decode

from PyQt5.QtCore import QBasicTimer
import win32api, win32con

import sys, os
from MyQR import myqr
import qrcode
import 二维码
from PyQt5.QtWidgets import QApplication, QMainWindow, QSplashScreen
from picture import *
import MyQR.myqr


class MainDialog(QMainWindow):
    def __init__(self, parent=None):
        super(QMainWindow, self).__init__(parent)

        self.ui = 二维码.Ui_MainWindow()
        self.ui.setupUi(self)
        self.ui.textEdit.setText("可以是任意字符")
        self.ui.lineEdit.setText("https://www.chinaacme.net/")
        self.ui.lineEdit.hide()
        self.get_pic(dlkj_jpg, "dlkj.jpg")

    def message(self):
        win32api.MessageBox(0, "生成的图片在软件运行目录下", "提醒", win32con.MB_OK)

    def run(self):
        if self.ui.textEdit.isHidden():
            self.myqr(self.ui.lineEdit.text())
        elif self.ui.lineEdit.isHidden():
            self.qrcode(self.ui.textEdit.toPlainText())
        os.system(os.path.join(self.getpath(), "顶立科技.png"))

    def qrcode(self, text=None):
        img = qrcode.make(str(text))

        with open(os.path.join(self.getpath(), "顶立科技.png"), "wb") as w:
            img.save(w)

    def myqr(self, text=None):
        if  os.path.exists(os.path.join(self.getpath(), "顶立科技.png")):
            try:

                myqr.run(words=str(text), version=9, picture="dlkj.jpg",
                         colorized=True,
                         save_name="顶立科技.png",
                         save_dir=self.getpath())
            except BaseException as e:
                win32api.MessageBox(0, str(e), "提醒", win32con.MB_OK)
                win32api.MessageBox(0, "这里不支持中文", "提醒", win32con.MB_OK)
        else:
            win32api.MessageBox(0, "配置文件已经删除，请重新启动本软件", "提醒", win32con.MB_OK)


    def getpath(self):
        return os.getcwd()

    def get_pic(self,pic_code, pic_name):
        image = open(pic_name, 'wb')
        image.write(b64decode(pic_code))
        image.close()



if __name__ == '__main__':
    myapp = QApplication(sys.argv)
    # 创建启动界面，支持png透明图片
    myDlg = MainDialog()
    myDlg.show()
    sys.exit(myapp.exec_())