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_())