OpenSearcher.py

import datetime
import locale
import os
import socket
import sys
import traceback
from queue import Queue
import win32api
import win32con
import winshell
from PyQt5 import QtCore
from PyQt5.QtCore import QModelIndex, QThread, pyqtSignal, QTimer, QProcess, QSettings, Qt
from PyQt5.QtGui import QIcon
from PyQt5.QtWidgets import QApplication, QMainWindow, QDesktopWidget, QMessageBox, qApp
from sqlitedict import SqliteDict

from qcustomdialog.CustomTypeWindow import CustomTypeWindow
from qcustomdialog.FileWindow import FileWindow
from qcustomdialog.HelpWindow import HelpWindow
from qcustomdialog.IndexWindow import IndexWindow
from qcustomdialog.SettingWindow import SettingWindow
from qcustomdialog.UpdateWindow import UpdateWindow
from MainWindow import Ui_MainWindow
from qutils.searchthread import SearchThread
from qutils.traythread import TrayThread
import logging

from qutils.types import GetTextTypesDict, GetTypesDict, GetImageTypesDict
from qutils.util import create_right_menu, remove_right_menu, create_shortcut, set_auto, set_right_menu

logger = logging.getLogger(__name__)


class MainWindow(QMainWindow):
    def __init__(self, icon, title, Version, UpdateUrl, socket, CurPath, DirPath, Name_, Name, LogoPath, ExePath):
        super().__init__()
        self.icon = icon
        self.title = title
        self.socket = socket
        self.Version = Version
        self.UpdateUrl = UpdateUrl
        self.CurPath = CurPath
        self.DirPath = DirPath
        self.Name_ = Name_
        self.Name = Name
        self.LogoPath = LogoPath
        self.ExePath = ExePath

        self.ErrorTempDir = os.path.abspath(os.path.join(self.CurPath, '.errortemp'))
        self.TempDir = os.path.abspath(os.path.join(self.CurPath, '.temp'))
        self.IconDir = os.path.abspath(os.path.join(self.CurPath, "icon"))
        self.LogDir = os.path.abspath(os.path.join(self.CurPath, "logs"))

        self.HelpPath = os.path.abspath(os.path.join(self.CurPath, 'HELP.md'))
        self.SettingPath = os.path.abspath(os.path.join(self.CurPath, 'setting.db'))
        self.LnkPath = os.path.abspath(os.path.join(winshell.desktop(), self.Name_ + ".lnk"))
        self.LnkPath2 = os.path.abspath(os.path.join(winshell.common_desktop(), self.Name_ + ".lnk"))

        self.isMax = False
        self.queue = Queue()
        self.SettingsIni = QSettings(os.path.abspath(os.path.join(self.CurPath, 'settings.ini')), QSettings.IniFormat)

        self.IconDict = {}
        self.SearchInfoDict = {}
        self.Tips = '''使用提示：

1. 尽量不要选择全盘搜索，而是选择某个文件夹进行搜索，Python读取全盘目录太慢了。

2. 利用空闲时间，提前建立索引缓存，第一次搜索某目录速度很慢，但是下次搜索将会很快。

3. 在搜索时，必须退出word、excel等文档，因为搜索中强行关闭Office会影响你的文档。

4. 点击右上角“搜索就绪”或者“搜索进度”，可以查看搜索信息。

5. 更新版本时，不用重新建立索引，直接将上一版本安装目录下.temp文件夹复制到本版本的安装目录下。'''

        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        self.ui.frame.setEnabled(False)
        self.Init()
        self.InitIconDict()
        self.InitSearchInfoDict()
        self.InitTrayIcon()
        self.InitLogger()
        self.InitGeometry()
        self.InitRelayUpdate()

    '''-----------------------------------主窗体设置 and 更新UI-----------------------------------------------'''

    def Init(self):
        # 从右键启动软件
        if self.DirPath is not None:
            self.ui.comboBox_dir.setCurrentText(self.DirPath)

        # 不允许崩溃
        self.ui.splitter.setCollapsible(0, False)
        self.ui.splitter.setCollapsible(1, False)
        self.ui.splitter.setCollapsible(2, False)
        self.ui.splitter_2.setCollapsible(0, False)
        self.ui.splitter_2.setCollapsible(1, False)

        self.ui.textframe.setReadOnly(True)
        self.ui.frame_center_left.hide()
        self.ui.progressBar.stop()
        self.ui.progressBar.hide()

        self.ui.tableView.setColumnWidth(0, 300)
        self.ui.tableView.setColumnWidth(1, 75)
        self.ui.tableView.setColumnWidth(2, 700)
        self.ui.tableView.setColumnWidth(3, 150)

        self.ui.search_button.clicked.connect(self.onClickedSearch)
        self.ui.textEdit_keywords.returnPressed.connect(self.onClickedSearch)
        self.ui.scale_up_button.clicked.connect(self.onClickedZoomIn)
        self.ui.scale_down_button.clicked.connect(self.onClickedZoomOut)
        self.ui.update_button.clicked.connect(self.onClickedUpdate)
        self.ui.show_error_button.clicked.connect(self.onClickedShowErrorListView)
        self.ui.show_all_button.clicked.connect(self.onClickedShowAllListView)
        self.ui.setting_button.clicked.connect(self.onClickedShowSettingWindow)
        self.ui.help_button.clicked.connect(self.onClickedHelpWindow)
        self.ui.index_button.clicked.connect(self.onClickedIndexWindow)
        self.ui.common_dir_button.clicked.connect(self.onClickedFileWindow)
        self.ui.tableView.clicked.connect(self.onClickedTableView)
        self.ui.listView_error.clicked.connect(self.onClickedListView)
        self.ui.listView_all.clicked.connect(self.onClickedListView_)
        self.ui.top_lable.clicked.connect(self.onClickedShowSearchInfo)
        self.ui.types_button.clicked.connect(self.onClickedCustomTypeWindow)

        self.timer = QTimer(self)
        self.timer.timeout.connect(self.UpdateUi)
        self.timer.start(150)

    def UpdateUi(self):
        if not self.queue.empty():
            for i in range(self.queue.qsize()):
                pass
                content = self.queue.get()
                if content[0] == "error":
                    self.ui.listView_error.listModel.appendRow(content[1])
                elif content[0] == "all":
                    self.ui.listView_all.listModel.appendRow(content[1])
                elif content[0] == "trigger":
                    self.ui.tableView.tableModel.append_data(content[1])
                elif content[0] == "completed":
                    self.SearchComplete()
                elif content[0] == "update_top_lable":
                    self.ui.top_lable.setText(content[1])
                elif content[0] == "update_bottom_lable":
                    self.ui.bottom_lable.setText(content[1])
                elif content[0] == "update_search_info":
                    self.UpdateSearchInfo(content[1])
                elif content[0] == "loading":
                    self.ui.bottom_lable.setText(content[1])
                    self.ui.textframe.Search(Loading=True, CurPath=self.CurPath, FilePath=content[1],
                                             Keywords=self.textEdit_keywords_text, IsError=content[2])
                elif content[0] == "_show_all":
                    if not content[1]:
                        if self.ui.frame_center_left.isVisible():
                            self.ui.frame_center_left.hide()
                elif content[0] == "_common_dir":
                    self.ui.comboBox_dir.setCurrentText(content[1])

    def UpdateSearchInfo(self, info_dict=None):
        if info_dict is None:
            info_dict = {}
        try:
            self.SearchInfoDict["start_time"] = info_dict["start_time"]
        except:
            pass
        try:
            self.SearchInfoDict["all"] = info_dict["all"]
        except:
            pass
        try:
            self.SearchInfoDict["selected"] = info_dict["selected"]
        except:
            pass
        try:
            self.SearchInfoDict["haskey"] = info_dict["haskey"]
        except:
            pass
        try:
            self.SearchInfoDict["error"] = info_dict["error"]
        except:
            pass
        try:
            self.SearchInfoDict["types"] = info_dict["types"]
        except:
            pass
        try:
            self.SearchInfoDict["dirs"] = info_dict["dirs"]
        except:
            pass
        try:
            self.SearchInfoDict["limited"] = info_dict["limited"]
        except:
            pass
        try:
            self.SearchInfoDict["exclude"] = info_dict["exclude"]
        except:
            pass
        self.UpdateSearchInfo_()

    def UpdateSearchInfo_(self):
        self.SearchInfo = "\n\n".join((
            "开始时间：%s\t\t\t" % self.SearchInfoDict["start_time"],
            "搜索类型：%s\t\t\t" % self.SearchInfoDict["types"],
            "文件限制：%s\t\t\t" % self.SearchInfoDict["limited"],
            "搜索目录：\n%s\t\t\t" % self.SearchInfoDict["dirs"],
            "忽略目录：\n%s\t\t\t" % self.SearchInfoDict["exclude"],
            "共搜索文件：%d个，满足搜索条件的文件：%d个\t\t\t" % (self.SearchInfoDict["all"], self.SearchInfoDict["selected"]),
            "其中：包含关键词的文件：%d个，处理出错的文件：%d个\t\t\t" % (self.SearchInfoDict["haskey"], self.SearchInfoDict["error"]),
        ))

    def onClickedCustomTypeWindow(self):
        self.onClickedShow()
        self.ui.comboBox_type.saveState()
        self.ui.comboBox_type.Clear()
        customtypewindow = CustomTypeWindow(self)
        customtypewindow.show()

    def onClickedShowSearchInfo(self):
        QMessageBox.information(self, '搜索信息', self.SearchInfo, QMessageBox.Ok)

    def onClickedZoomIn(self):
        self.ui.textframe.zoomIn(1)  # 放大

    def onClickedZoomOut(self):
        self.ui.textframe.zoomIn(-1)

    def onClickedShowAllListView(self):
        if self.mysetting_dict['_show_all']:
            if self.ui.frame_center_left.isVisible():
                self.ui.frame_center_left.hide()
            else:
                self.ui.frame_center_left.show()
        else:
            QMessageBox.information(self, '注意', '当前未开启显示全部文件功能（请在设置中开启）', QMessageBox.Ok)

    def onClickedShowErrorListView(self):
        if self.ui.listView_error.isVisible():
            self.ui.listView_error.hide()
        else:
            self.ui.listView_error.show()

    def onClickedTableView(self, index):
        table_row = index.row()
        _index = self.ui.tableView.tableModel.index(table_row, 2, QModelIndex())
        file_path = os.path.abspath(str(_index.data()))
        self.queue.put(("loading", file_path, False))

    def onClickedListView(self, index):
        table_row = index.row()
        _item = self.ui.listView_error.listModel.item(table_row, 0)
        file_path = _item.text()
        file_path_ = file_path.split(" -> (")[-1][:-1]
        self.queue.put(("loading", file_path_, True))

    def onClickedListView_(self, index):
        table_row = index.row()
        _item = self.ui.listView_all.listModel.item(table_row, 0)
        file_path = _item.text()
        file_path_ = file_path.split(" -> (")[-1][:-1]
        self.queue.put(("loading", file_path_, False))

    def closeEvent(self, event):
        event.ignore()
        if self.isMaximized():
            self.isMax = True
        else:
            self.isMax = False
        # 执行顺序不能颠倒
        self.showMinimized()
        self.setWindowFlags(QtCore.Qt.SplashScreen | QtCore.Qt.FramelessWindowHint)

    '''-----------------------------------初始化设置 and 耗时操作线程（以及监听端口）-----------------------------------------------'''

    def InitIconDict(self):
        self.IconDict["icon_doc"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "doc.png")))
        self.IconDict["icon_docx"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "docx.png")))
        self.IconDict["icon_xls"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "xls.png")))
        self.IconDict["icon_xlsx"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "xlsx.png")))
        self.IconDict["icon_ppt"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "ppt.png")))
        self.IconDict["icon_pptx"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "pptx.png")))
        self.IconDict["icon_pdf"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "pdf.png")))
        self.IconDict["icon_txt"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "txt.png")))
        self.IconDict["icon_epub"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "epub.png")))
        self.IconDict["icon_mobi"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "mobi.png")))
        self.IconDict["icon_et"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "et.png")))
        self.IconDict["icon_wps"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "wps.png")))

        self.IconDict["icon_psd"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "psd.png")))
        self.IconDict["icon_png"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "png.png")))
        self.IconDict["icon_jpg"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "jpg.png")))
        self.IconDict["icon_jpeg"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "jpg.png")))
        self.IconDict["icon_tiff"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "tiff.png")))
        self.IconDict["icon_raw"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "raw.png")))
        self.IconDict["icon_bmp"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "bmp.png")))
        self.IconDict["icon_image"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "image.png")))

        self.IconDict["icon_md"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "md.png")))
        self.IconDict["icon_markdown"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "md.png")))
        self.IconDict["icon_conf"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "conf.png")))
        self.IconDict["icon_yaml"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "yaml.png")))
        self.IconDict["icon_json"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "json.png")))
        self.IconDict["icon_srt"] = QIcon(os.path.abspath(os.path.join(self.IconDir, "srt.png")))

    def InitSearchInfoDict(self):
        self.SearchInfoDict["start_time"] = ""
        self.SearchInfoDict["all"] = 0
        self.SearchInfoDict["selected"] = 0
        self.SearchInfoDict["haskey"] = 0
        self.SearchInfoDict["error"] = 0
        self.SearchInfoDict["types"] = ""
        self.SearchInfoDict["dirs"] = ""
        self.SearchInfoDict["limited"] = ""
        self.SearchInfoDict["exclude"] = ""
        self.UpdateSearchInfo_()

    def InitComboBox(self, mysetting_dict):
        self.ui.comboBox_type.InitComboBox(mysetting_dict)

    def InitLogger(self):
        if not os.path.isdir(self.LogDir):
            os.makedirs(self.LogDir)
        self.logger = logging.getLogger(__name__)
        logging.basicConfig(level='INFO', format='%(asctime)s: %(message)s')
        fmt = '%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s:%(message)s'
        fh = logging.FileHandler(
            os.path.abspath(os.path.join(self.LogDir, f'{datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")}.txt')),
            encoding="utf8")
        fh.setFormatter(logging.Formatter(fmt=fmt))
        fh.setLevel(logging.INFO)
        self.logger.addHandler(fh)

    def InitGeometry(self):
        if os.path.exists(os.path.join(self.CurPath, "settings.ini")):
            self.restoreGeometry(self.SettingsIni.value("geometry", self.saveGeometry()))
            self.restoreState(self.SettingsIni.value("state", self.saveState()))
            self.ui.splitter.restoreGeometry(
                self.SettingsIni.value("splitter_geometry", self.ui.splitter.saveGeometry()))
            self.ui.splitter.restoreState(self.SettingsIni.value("splitter_state", self.ui.splitter.saveState()))
            self.ui.splitter_2.restoreGeometry(
                self.SettingsIni.value("splitter_2_geometry", self.ui.splitter_2.saveGeometry()))
            self.ui.splitter_2.restoreState(self.SettingsIni.value("splitter_2_state", self.ui.splitter_2.saveState()))
        else:
            screen = QDesktopWidget().screenGeometry()
            height = screen.height() * 6 / 7
            width = screen.width() * 9 / 10
            self.setGeometry(int((screen.width() - width) / 2), int((screen.height() - height) / 2), int(width),
                             int(height))

    def InitRelayUpdate(self):
        self.relay_thread = RelayUpdateThread(self.socket, self.SettingPath, self.Name,
                                              self.Name_, self.LogoPath, self.ExePath, self.IconDir)
        self.relay_thread.DateBaseCompleted.connect(self.InitSettingDateBase)
        self.relay_thread.DirAccepted.connect(self.UpdateDir)
        self.relay_thread.ExitAccepted.connect(self.ProgramExit)
        self.relay_thread.start()

    def ProgramExit(self):
        try:
            self.backend_thread.OnDestroy()
        except:
            self.logger.info(traceback.format_exc())
        try:
            qApp.quit()
        except:
            self.logger.info(traceback.format_exc())
        sys.exit(0)

    def UpdateDir(self, dir):
        if self.isMinimized() or not self.isVisible():
            self.setWindowFlags(QtCore.Qt.Window)
            self.activateWindow()
            if self.isMax:
                self.showMaximized()
            else:
                self.showNormal()
        self.ui.comboBox_dir.setCurrentText(dir)

    def InitSettingDateBase(self, setting):
        self.mysetting_dict = setting
        self._auto_run = self.mysetting_dict['_auto_run']
        self._last_dir_flag = self.mysetting_dict['_last_dir_flag']
        self._right_menu = self.mysetting_dict['_right_menu']
        self.ui.textframe.Init(fontsize=self.mysetting_dict['_fontsize'],
                               fontfamily=self.mysetting_dict['_fontfamily'],
                               linewrap=self.mysetting_dict['_linewrap'],
                               linenumber=self.mysetting_dict['_linenumber'],
                               mysetting_dict=self.mysetting_dict)
        self.ui.comboBox_type.InitComboBox(mysetting_dict=self.mysetting_dict)
        if self.mysetting_dict["_desktop"]:
            if not os.path.exists(self.LnkPath) and not os.path.exists(self.LnkPath2):
                self.logger.info("桌面没有创建快捷方式,准备创建")
                create_shortcut(sys.argv[0], self.Name_, "一个开源的、本地的、安全的、支持全文检索的搜索器。", self.LogoPath)
        if self.mysetting_dict["_last_dir_flag"] and self.mysetting_dict["_last_dir"].strip() != "":
            self.ui.comboBox_dir.setCurrentText(self.mysetting_dict["_last_dir"])

        self.ui.textframe.setText(self.Tips)
        self.ui.frame.setEnabled(True)

    '''-----------------------------------后台图标-----------------------------------------------'''

    def InitTrayIcon(self):
        self.backend_thread = TrayThread(icon=self.icon, title=self.title)
        self.backend_thread.start()
        self.backend_thread.setting.connect(self.onClickedShowSettingWindow)
        self.backend_thread.show.connect(lambda: self.onClickedShow(True))
        self.backend_thread.help.connect(self.onClickedHelpWindow)
        self.backend_thread.exit.connect(self.onClickedExit)
        self.backend_thread.update.connect(self.onClickedUpdate)
        self.backend_thread.restart.connect(self.onClickedRestart)

    def onClickedShow(self, close=False):
        if self.isMinimized() or not self.isVisible():
            self.setWindowFlags(QtCore.Qt.Window)
            self.activateWindow()
            if self.isMax:
                self.showMaximized()
            else:
                self.showNormal()
        else:
            if close:
                if self.isMaximized():
                    self.isMax = True
                self.close()

    def onClickedRestart(self):
        self.saveSettingIni()
        self.backend_thread.OnDestroy()
        qApp.quit()
        self.p = QProcess
        s = self.p.startDetached(sys.executable, sys.argv)

    def onClickedExit(self):
        self.saveSettingIni()
        self.backend_thread.OnDestroy()
        qApp.quit()
        sys.exit(0)

    def onClickedShowSettingWindow(self):
        self.onClickedShow()
        settingwindow = SettingWindow(self)
        settingwindow.show()

    def onClickedFileWindow(self):
        self.onClickedShow()
        filewindow = FileWindow(self)
        filewindow.show()

    def onClickedHelpWindow(self):
        self.onClickedShow()
        helpwindow = HelpWindow(self)
        helpwindow.show()

    def onClickedIndexWindow(self):
        self.onClickedShow()
        indexwindow = IndexWindow(self)
        indexwindow.show()

    def onClickedUpdate(self):
        self.onClickedShow()
        updatewindow = UpdateWindow(parent=self)
        updatewindow.show()

    def saveSettingIni(self):
        self.SettingsIni.setValue("geometry", self.saveGeometry())
        self.SettingsIni.setValue("state", self.saveState())
        self.SettingsIni.setValue("splitter_geometry", self.ui.splitter.saveGeometry())
        self.SettingsIni.setValue("splitter_state", self.ui.splitter.saveState())
        self.SettingsIni.setValue("splitter_2_geometry", self.ui.splitter_2.saveGeometry())
        self.SettingsIni.setValue("splitter_2_state", self.ui.splitter_2.saveState())

    '''-----------------------------------搜索相关-----------------------------------------------'''

    def SearchStartUi(self):
        self.ui.tableView.tableModel.remove_all()
        self.ui.listView_error.listModel.clear()
        self.ui.listView_all.listModel.clear()
        self.ui.textframe.setText(self.Tips)

        self.ui.progressBar.start()
        self.ui.progressBar.show()
        self.ui.top_lable.setText("搜索中")
        self.ui.bottom_lable.setText("搜索中·····")
        self.ui.search_button.setText("停止")

    def SearchStopUi(self):
        self.ui.search_button.setText("停止中···")
        self.ui.search_button.setEnabled(False)

    def SearchComplete(self):
        self.ui.search_button.setEnabled(True)
        self.ui.progressBar.hide()
        self.ui.progressBar.stop()
        self.ui.top_lable.setText("搜索就绪")
        self.ui.bottom_lable.setText("搜索完毕!")
        self.ui.search_button.setText("搜索")
        self.logger.info("搜索完成")
        if self.mysetting_dict['_remind']:
            self.backend_thread.ShowToast("当前搜索完成")

    def isRunning(self):
        try:
            return self.search_thread.isRunning()
        except:
            self.logger.info(traceback.format_exc())
            return False

    def onClickedSearch(self):
        if self.isRunning():
            reply = QMessageBox.information(self, '注意', '是否要停止搜索？', QMessageBox.Yes | QMessageBox.Cancel)
            if reply == QMessageBox.Yes:
                self.logger.info("开始停止搜索线程")
                self.search_thread.stop()
        else:
            self.comboBox_dir_text = self.ui.comboBox_dir.text()
            self.textEdit_keywords_text = self.ui.textEdit_keywords.text()
            self.comboBox_type_text = self.ui.comboBox_type.text()
            self.comboBox_type_list = [x for x in self.comboBox_type_text.split(";") if x]
            if self.comboBox_dir_text == "":
                self.ui.bottom_lable.setText("请先选择文件夹！")
                QMessageBox.information(self, "提示", "请先选择文件夹！", QMessageBox.Ok)
                self.logger.info("文件夹为空")
            elif self.textEdit_keywords_text.strip() == "":
                self.ui.bottom_lable.setText("请先输入关键词！")
                QMessageBox.information(self, "提示", "请先输入关键词！", QMessageBox.Ok)
                self.logger.info("关键词为空")
            elif self.comboBox_type_text == "":
                self.ui.bottom_lable.setText("请先指定文件类型！")
                QMessageBox.information(self, "提示", "请先指定文件类型！", QMessageBox.Ok)
                self.logger.info("文件类型为空")
            else:
                self.logger.info("开启搜索线程")
                self.ui.comboBox_type.saveState()
                self.mysetting_dict["_last_dir"] = self.comboBox_dir_text
                if self.comboBox_dir_text == "全部目录":
                    self._comboBox_dir_list = self.ui.comboBox_dir.get_devices()
                else:
                    self._comboBox_dir_list = self.comboBox_dir_text.split("|")
                self.comboBox_dir_list = self._comboBox_dir_list

                self.queue.put(("update_search_info", {
                    "start_time": str(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
                    "types": str(self.comboBox_type_list),
                    "limited": str(self.mysetting_dict['_limit_file_size']) + "M",
                    "dirs": str("\n".join(self.comboBox_dir_list)),
                    "exclude": str("\n".join(self.mysetting_dict['_exclude_dir'])),
                    "all": 0,
                    "selected": 0,
                    "error": 0,
                    "haskey": 0,
                }))
                self.logger.info(self.SearchInfo)
                self.search_thread = SearchThread(
                    DirPaths=self.comboBox_dir_list,
                    CurPath=self.CurPath,
                    IconDict=self.IconDict,
                    types=self.comboBox_type_list,
                    logger=self.logger,
                    queue=self.queue,
                    mysetting_dict=self.mysetting_dict,
                    keywords=self.textEdit_keywords_text)
                self.search_thread.started.connect(self.SearchStartUi)
                self.search_thread.finished.connect(lambda: self.queue.put(("completed", 1)))
                self.search_thread.stoped.connect(self.SearchStopUi)
                self.search_thread.start()


class RelayUpdateThread(QThread):
    DateBaseCompleted = pyqtSignal(SqliteDict)
    DirAccepted = pyqtSignal(str)
    ExitAccepted = pyqtSignal()

    def __init__(self, socket, SettingPath, Name, Name_, LogoPath, ExePath, IconDir):
        super().__init__()
        self.socket = socket
        self.SettingPath = SettingPath
        self.Name = Name
        self.Name_ = Name_
        self.ExePath = ExePath
        self.LogoPath = LogoPath
        self.IconDir = IconDir

    def run(self):
        self.settingdb()
        while True:
            data, addr = self.socket.recvfrom(1024)
            data_ = data.decode("utf-8")
            if data_ == "exit_":
                self.ExitAccepted.emit()
            else:
                self.DirAccepted.emit(data_)

    def settingdb(self):
        if not os.path.exists(self.SettingPath):
            self.mysetting_dict = SqliteDict(self.SettingPath, autocommit=True)
            self.mysetting_dict.clear()
            self.mysetting_dict['_auto_run'] = True
            self.mysetting_dict['_last_dir_flag'] = True
            self.mysetting_dict['_last_dir'] = ""
            self.mysetting_dict['_right_menu'] = True
            self.mysetting_dict['_remind'] = True
            self.mysetting_dict['_show_all'] = False
            self.mysetting_dict['_limit_file_size'] = 100
            self.mysetting_dict['_limit_office_time'] = 5
            self.mysetting_dict['_fontsize'] = 10
            self.mysetting_dict['_fontfamily'] = "宋体"
            self.mysetting_dict['_linewrap'] = True
            self.mysetting_dict['_linenumber'] = True
            self.mysetting_dict['_common_dir'] = []
            self.mysetting_dict['_exclude_dir'] = []
            self.mysetting_dict['_desktop'] = True
            self.mysetting_dict["_types_dict"] = GetTypesDict(self.IconDir)
            self.mysetting_dict["_text_types_dict"] = GetTextTypesDict(self.IconDir)
            self.mysetting_dict["_image_types_dict"] = GetImageTypesDict(self.IconDir)
            self.mysetting_dict["_image_types_list"] = []
            self.mysetting_dict["_text_types_list"] = []

            if not set_auto(name=self.Name, exe_path=self.ExePath):
                self.mysetting_dict['_auto_run'] = False
            if not set_right_menu(name=self.Name_, content=self.Name_ + " 搜索", icon=self.LogoPath,
                                  exe_path=self.ExePath):
                self.mysetting_dict['_right_menu'] = False
        else:
            self.mysetting_dict = SqliteDict(self.SettingPath, autocommit=True)

        self.DateBaseCompleted.emit(self.mysetting_dict)


def main():
    print(sys.argv)

    Name = "OpenSearcher"
    Name_ = "Open Searcher"
    Version = "1.0.0.2"
    UpdateUrl = "https://gitee.com/gao_yongxian/OpenSearcher/raw/main/update.txt"

    CurPath = os.path.dirname(os.path.abspath(sys.argv[0]))

    os.environ['REQUESTS_CA_BUNDLE'] = os.path.abspath(os.path.join(CurPath, 'cacert.pem'))
    LogoPath = os.path.abspath(os.path.join(CurPath, "icon/logo.ico"))
    if sys.argv[0][-3:] == ".py":
        ExePath = " ".join([sys.executable, sys.argv[0]])
    else:
        ExePath = sys.argv[0]

    # 判断是否是设置右键菜单
    if len(sys.argv) > 1:
        if sys.argv[-1] == "set_menu":
            try:
                # icon 设置成 exe文件也可以，Windows系统会自动读取ico
                create_right_menu(name=Name, content=Name_ + " 搜索", icon=LogoPath, command=ExePath + " \"%L\"",
                                  type_="DIRECTORY")
                create_right_menu(name=Name, content=Name_ + " 搜索", icon=LogoPath, command=ExePath + " \"%W\"",
                                  type_="DIRECTORY_BACKGROUND")
            except Exception:
                win32api.MessageBox(0, traceback.format_exc(), Name_, win32con.MB_OK)
            return
        elif sys.argv[-1] == "del_menu":
            try:
                remove_right_menu(Name, type_="DIRECTORY")
                remove_right_menu(Name, type_="DIRECTORY_BACKGROUND")
            except Exception:
                win32api.MessageBox(0, traceback.format_exc(), Name_, win32con.MB_OK)
            return
        elif sys.argv[-1][-4:] == ".exe":
            print("当前重启")
            DirPath = None
        else:
            DirPath = sys.argv[-1]
    else:
        DirPath = None

    # 判断是否已经启动
    try:
        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        host = socket.gethostname()
        s.bind((host, 60111))
    except Exception:
        # 如果已经启动：对当前运行的程序1.传递路径。2.停止运行
        if len(sys.argv) > 1:
            s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
            host = socket.gethostname()
            s.connect((host, 60111))
            data = sys.argv[-1]
            s.sendto(data.encode('utf-8'), (host, 60111))
        else:
            result = win32api.MessageBox(0, "提示：Open Searcher程序后台运行中\n是否停止当前正在运行的Open Searcher？", Name_,
                                         win32con.MB_YESNO)
            if result == 6:
                s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
                host = socket.gethostname()
                s.connect((host, 60111))
                data = "exit_"
                s.sendto(data.encode('utf-8'), (host, 60111))
        return

    # 高DPI
    QApplication.setHighDpiScaleFactorRoundingPolicy(Qt.HighDpiScaleFactorRoundingPolicy.PassThrough)
    QApplication.setAttribute(Qt.AA_EnableHighDpiScaling)
    QApplication.setAttribute(Qt.AA_UseHighDpiPixmaps)

    app = QApplication(sys.argv)
    window = MainWindow(icon=LogoPath,
                        title=Name_ + " " + Version,
                        socket=s,
                        Version=Version,
                        UpdateUrl=UpdateUrl,
                        CurPath=CurPath,
                        DirPath=DirPath,
                        Name_=Name_,
                        LogoPath=LogoPath,
                        Name=Name,
                        ExePath=ExePath)
    window.setWindowTitle(Name_ + " " + Version + " 开源软件")
    window.setWindowIcon(QIcon(LogoPath))
    window.show()
    sys.exit(app.exec())


if __name__ == '__main__':
    # windows7，python3使用time.strftime()函数报ValueError: embedded null byte
    locale.setlocale(locale.LC_ALL, 'en')
    locale.setlocale(locale.LC_CTYPE, 'chinese')
    main()