# -*- coding:utf-8 -*-
import sys
import logging

from file import fileDataCreating
from common import global_var

from pathlib import Path

from PySide6.QtUiTools import QUiLoader
from PySide6.QtCore import Signal, QThread, QObject, QEventLoop, QTimer
from PySide6.QtWidgets import QApplication, QFileDialog, QMessageBox
from PySide6.QtGui import QTextCursor

# 声明全局变量
global_var._init()
global_var.set_value('FromGUI', True)


# 控制台重定向到GUI
class EmittingStr(QObject):
    # 定义信号
    textWritten = Signal(str)
    # 将控制台的内容输出到QTextBrowser

    def write(self, text):
        # 调用 emit方法 发信号时，传入参数 必须是这里指定的 参数类型
        self.textWritten.emit(str(text))
        loop = QEventLoop()
        QTimer.singleShot(100, loop.quit)
        loop.exec_()
        QApplication.processEvents()

# logging重定向到print


class PrintHandler(logging.Handler):
    def emit(self, record):
        pass


# 处理文件的线程
class FileThread(QThread):

    def __init__(self, parent=None):
        super().__init__(parent=parent)
    # 覆盖原来run去调用文件造数据脚本

    def run(self):
        fileDataCreating.main()


# GUI的主程序
class Window:

    def __init__(self):
        # 载入ui
        path = Path(__file__)
        ui_file_patch = path.parent.parent / "ui" / "ui_main.ui"
        self.ui = QUiLoader().load(str(ui_file_patch))
        # 根据UI尺寸设定值，禁止拉伸窗体
        self.ui.setFixedSize(self.ui.width(), self.ui.height())
        # 指定按钮和槽函数
        self.ui.startButton.clicked.connect(self.start)
        self.ui.stopButton.clicked.connect(self.stop)
        self.ui.conn_Button.clicked.connect(self.conntest)
        self.ui.choosedir_Button.clicked.connect(self.choosedir)
        self.ui.actionabout.triggered.connect(self.about)

        self.setup_thread()

        # QTextBrowser输出设置，限制条数为100
        self.ui.Log_Output.document().setMaximumBlockCount(100)
        self.ui.Log_Output.ensureCursorVisible()
        # 将控制台输出重定向到QTextBrowser
        sys.stdout = EmittingStr()
        sys.stdout.textWritten.connect(self.outputWritten)

        # 创建并配置logger
        self.logger = logging.getLogger()
        self.logger.setLevel(logging.INFO)
        # 创建并配置处理程序
        handler = PrintHandler()
        handler.setFormatter(logging.Formatter(
            '%(asctime)s - %(levelname)s - %(message)s'))
        self.logger.addHandler(handler)

    # 输出到GUI，信号处理函数

    def outputWritten(self, text):
        cursor = self.ui.Log_Output.textCursor()
        cursor.movePosition(QTextCursor.End)
        cursor.insertText(text)
        self.ui.Log_Output.setTextCursor(cursor)
        self.ui.Log_Output.ensureCursorVisible()

    def start(self):
        self.logger.info("点击了开始按钮")
        # 按钮置灰
        self.ui.startButton.setEnabled(False)
        # 输入框参数校验
        num = self.ui.tabWidget.currentIndex()
        if num == 0:
            if not self.check_file_input():
                self.logger.info("参数校验失败，请检查输入的值")
                self.ui.startButton.setEnabled(True)
                return
            self.logger.info("参数校验通过，开始执行")
            # 持续性生成和一次性的
            if self.ui.checkBox.isChecked():
                self.start_thread()
            else:
                fileDataCreating.new()
                self.ui.startButton.setEnabled(True)
                self.logger.info("*"*5 + "一次性任务执行完成" + "*"*5)
        else:
            self.conntest()
            self.start_run_db()

    def stop(self):
        self.logger.info("点击了停止按钮")
        num = self.ui.tabWidget.currentIndex()
        if num == 0:
            self.stop_run_file()
        else:
            self.stop_run_db()
        # 按钮恢复
        # self.startButton.isEnabled() 获取按钮的状态
        self.ui.startButton.setEnabled(True)

    def conntest(self):
        self.logger.info("点击了测试连接按钮")
        self.check_db_input()
        # TODO(MH):校验连接信息
        self.logger.info("数据库连接成功")

    def choosedir(self):
        self.logger.info("点击了选择目录按钮")
        dir_path = QFileDialog.getExistingDirectory(
            self.ui, "选择目录", "", QFileDialog.ShowDirsOnly)
        self.ui.fileOutputText.setText(dir_path)
        self.logger.info("已经选择文件夹：{}".format(dir_path))

    def setup_thread(self):
        self.thread = FileThread()
        global_var.set_value('start_flag', True)
        self.thread_running = global_var.get_value('start_flag')

    def start_thread(self):
        if self.thread_running:
            self.thread.start()
        else:
            self.setup_thread()
            self.thread.start()

    def check_common(self, elemnet):
        # 为空标红处理
        elemnet.setStyleSheet("")
        if not elemnet.text():
            elemnet.setStyleSheet("border: 1px solid red;")
            # 改变状态
            self.check_flag = False

        return elemnet.text()

    def check_file_input(self):
        self.check_flag = True
        # 获取界面输入
        file_dict = {}
        file_dict['filename'] = self.check_common(self.ui.fileNameText)
        file_dict['outputpath'] = self.ui.fileOutputText.text()
        file_dict['fileinit'] = self.check_common(self.ui.fileInitRowsText)
        # 是否持续
        file_dict['filecreating'] = self.ui.checkBox.isChecked()
        if file_dict['filecreating']:
            file_dict['fileinsert'] = self.check_common(
                self.ui.fileInsertRowsText)
            file_dict['filedel'] = self.check_common(self.ui.fileDelRowsText)
        # 将值放到全局变量中
        global_var.set_value('file_args', file_dict)

        return self.check_flag

    def check_db_input(self):
        self.logger.info("数据库参数校验通过，开始连接数据库测试。。。")

    def start_run_file(self):

        # 调用文件处理脚本
        self.logger.info("开始生成文件。。。")

    def start_run_db(self):
        pass

    def stop_run_file(self):
        # 终止线程的事件循环
        global_var.set_value('start_flag', False)
        self.thread.quit()
        # 标记线程停止
        self.thread_running = global_var.get_value('start_flag')

    def stop_run_db(self):
        pass

    def about(self):
        msg = "版本:     <a>v1.0.0</a><br>\
               项目地址: <a href='https://gitea.kiki.kim/ha/dataCreatingGUI'>https://gitea.kiki.kim/ha/dataCreatingGUI</a><br>\
               E-mail:  <a href='mailto:halliday2023@163.com'>halliday2023@163.com</a>"
        QMessageBox.about(
            self.ui,
            '关于',
            msg)