237 lines
9.2 KiB
Python
237 lines
9.2 KiB
Python
from datetime import datetime
|
||
from logging import error, info
|
||
from pathlib import Path
|
||
|
||
from PySide6.QtWidgets import QMessageBox, QWidget, QPushButton, QProgressBar, QApplication, QRadioButton, QCheckBox
|
||
|
||
from func.utils.Constants import Constants
|
||
from func.utils.CustomException import TipsTypeValueNotExistError, MsgBoxTypeValueNotExistError
|
||
from func.utils.Result import Result
|
||
|
||
|
||
class PublicFunc:
|
||
|
||
@staticmethod
|
||
def get_current_localtime() -> str:
|
||
"""
|
||
获取当前本地时间
|
||
|
||
Parameters:
|
||
|
||
Returns:
|
||
str，格式化为"%H:%M:%S"的当前本地时间
|
||
|
||
Raises:
|
||
"""
|
||
return str(datetime.now().strftime("%H:%M:%S"))
|
||
|
||
@staticmethod
|
||
def format_status_msg(msg) -> str:
|
||
"""
|
||
格式化状态栏信息
|
||
|
||
Parameters:
|
||
msg - str，需要被格式化的字符串
|
||
|
||
Returns:
|
||
str，格式化为"%H:%M:%S"的当前本地时间 + 信息
|
||
|
||
Raises:
|
||
"""
|
||
return str(datetime.now().strftime("%H:%M:%S")) + " " + msg
|
||
|
||
@staticmethod
|
||
def text_output(main_window: object, content: str, tips_type: str) -> None:
|
||
"""
|
||
更新textBrowser中的内容，同时输出日志
|
||
|
||
Parameters:
|
||
main_window - object，Qt的含有textBrowser属性的对象，这里一般指的是mainWindow，里面有唯一的textBrowser属性
|
||
content - str，需要输出的内容
|
||
tips_type - 日志输出的类型
|
||
|
||
Returns:
|
||
|
||
Raises:
|
||
TipsTypeValueNotExistError
|
||
"""
|
||
if tips_type is Constants.TIPS_TYPE_INFO:
|
||
info(f"{tips_type}: {content}")
|
||
main_window.textBrowser_info.append(
|
||
f"<font color='black'>{PublicFunc.get_current_localtime()} {tips_type}: {content}</font>")
|
||
elif tips_type is Constants.TIPS_TYPE_ERROR:
|
||
error(f"{tips_type}: {content}")
|
||
main_window.textBrowser_info.append(
|
||
f"<font color='red'>{PublicFunc.get_current_localtime()} {tips_type}: {content}</font>")
|
||
else:
|
||
raise TipsTypeValueNotExistError()
|
||
|
||
main_window.textBrowser_info.verticalScrollBar().setValue(
|
||
main_window.textBrowser_info.verticalScrollBar().maximum())
|
||
|
||
@staticmethod
|
||
def msgbox_output(main_window: object, content: str, msg_box_type: str) -> None:
|
||
"""
|
||
更新messageBox中的内容，并弹出
|
||
|
||
Parameters:
|
||
main_window - object，Qt的含有messageBox属性的对象，这里一般指的是mainWindow，里面有唯一的messageBox属性
|
||
content - str，需要输出的内容
|
||
msg_box_type - str，messageBox弹窗的类型
|
||
|
||
Returns:
|
||
|
||
Raises:
|
||
MsgBoxTypeValueNotExistError
|
||
"""
|
||
main_window.msgBox.setText(f"{msg_box_type}: {content}")
|
||
if msg_box_type is Constants.MSGBOX_TYPE_INFO:
|
||
main_window.msgBox.setIcon(QMessageBox.Information)
|
||
elif msg_box_type is Constants.MSGBOX_TYPE_WARNING:
|
||
main_window.msgBox.setIcon(QMessageBox.Warning)
|
||
elif msg_box_type is Constants.MSGBOX_TYPE_ERROR:
|
||
main_window.msgBox.setIcon(QMessageBox.Critical)
|
||
elif msg_box_type is Constants.MSGBOX_TYPE_QUESTION:
|
||
main_window.msgBox.setIcon(QMessageBox.Question)
|
||
else:
|
||
raise MsgBoxTypeValueNotExistError()
|
||
main_window.msgBox.exec()
|
||
|
||
@staticmethod
|
||
def __disableAllButton__(mainWindow, buttonState):
|
||
# 禁用所有按钮
|
||
all_widgets = mainWindow.centralWidget().findChildren(QWidget)
|
||
|
||
# 迭代所有部件，查找按钮并禁用它们
|
||
for widget in all_widgets:
|
||
if isinstance(widget, QPushButton):
|
||
if widget.objectName() in buttonState["Current"].keys():
|
||
widget.setEnabled(False)
|
||
|
||
if isinstance(widget, QRadioButton):
|
||
if widget.objectName() in buttonState["Current"].keys():
|
||
widget.setEnabled(False)
|
||
|
||
@staticmethod
|
||
def __enableAllButton__(mainWindow, buttonState):
|
||
# 启用按钮
|
||
all_widgets = mainWindow.centralWidget().findChildren(QWidget)
|
||
|
||
# 迭代所有部件，查找按钮并启用它们
|
||
for widget in all_widgets:
|
||
if isinstance(widget, QPushButton):
|
||
if widget.objectName() in buttonState["Current"].keys():
|
||
widget.setEnabled(buttonState["Current"][widget.objectName()])
|
||
|
||
if isinstance(widget, QRadioButton):
|
||
if widget.objectName() in buttonState["Current"].keys():
|
||
widget.setEnabled(buttonState["Current"][widget.objectName()])
|
||
|
||
@staticmethod
|
||
def __resetAllButton__(mainWindow, buttonState):
|
||
# 启用按钮
|
||
all_widgets = mainWindow.centralWidget().findChildren(QWidget)
|
||
|
||
# 迭代所有部件，查找按钮并启用它们
|
||
for widget in all_widgets:
|
||
if isinstance(widget, QPushButton):
|
||
if widget.objectName() in buttonState["Default"].keys():
|
||
widget.setEnabled(buttonState["Default"][widget.objectName()])
|
||
|
||
if isinstance(widget, QRadioButton):
|
||
if widget.objectName() in buttonState["Default"].keys():
|
||
widget.setEnabled(buttonState["Default"][widget.objectName()])
|
||
|
||
@staticmethod
|
||
def __styleAllButton__(mainWindow, buttonState):
|
||
# 启用按钮
|
||
all_widgets = mainWindow.centralWidget().findChildren(QWidget)
|
||
|
||
# 迭代所有部件，查找按钮并启用它们
|
||
for widget in all_widgets:
|
||
if isinstance(widget, QPushButton):
|
||
if widget.objectName() in buttonState["Default"].keys():
|
||
widget.setStyleSheet(Constants.LABELBTN_STYLE_NORMAL)
|
||
if isinstance(widget, QCheckBox):
|
||
widget.setStyleSheet(Constants.CHECKBOX_STYLE_NORMAL)
|
||
|
||
@staticmethod
|
||
def add_progressbar(mainWindow):
|
||
mainWindow.progressbar = QProgressBar()
|
||
mainWindow.progressbar.setRange(0, 100)
|
||
mainWindow.progressbar.setValue(0)
|
||
mainWindow.progressbar.setStyleSheet(Constants.PROGRESSBAR_STYLE)
|
||
mainWindow.ui.statusbar.addPermanentWidget(mainWindow.progressbar)
|
||
|
||
@staticmethod
|
||
def statusbar_show_msg(mainWindow, msg):
|
||
mainWindow.ui.statusbar.showMessage(msg)
|
||
|
||
@staticmethod
|
||
def statusbar_clear_msg(mainWindow):
|
||
mainWindow.ui.statusbar.clearMessage()
|
||
|
||
@staticmethod
|
||
def finish_operation(mainWindow, buttonState):
|
||
PublicFunc.statusbar_show_msg(mainWindow, PublicFunc.format_status_msg(Constants.OPERATION_FINISHED))
|
||
mainWindow.progressbar.setValue(100)
|
||
QApplication.processEvents()
|
||
|
||
PublicFunc.__enableAllButton__(mainWindow, buttonState)
|
||
|
||
@staticmethod
|
||
def progressbar_update(mainWindow, current: int, total: int, msg: str, progressbarState: int):
|
||
if current > total:
|
||
raise ValueError("当前进度值大于总进度值")
|
||
if progressbarState < 0 or progressbarState > 100:
|
||
raise ValueError("进度条值的范围应该在[0, 100]")
|
||
PublicFunc.statusbar_show_msg(mainWindow, PublicFunc.format_status_msg(f"({str(current)}/{str(total)}){msg}"))
|
||
mainWindow.progressbar.setValue(progressbarState)
|
||
QApplication.processEvents()
|
||
|
||
@staticmethod
|
||
def examine_file(path_str, filename, suffix):
|
||
temp_path = [p for p in Path(path_str).glob(f"{filename}*") if p.stem.startswith(filename) and p.suffix == suffix]
|
||
|
||
if len(temp_path) == 0:
|
||
return Result().failure(
|
||
info=Constants.INPUT_FAILURE + "\n" +
|
||
filename + "：" +
|
||
path_str +
|
||
Constants.FAILURE_REASON["File_Not_Exist"])
|
||
elif len(temp_path) > 1:
|
||
return Result().failure(
|
||
info=Constants.INPUT_FAILURE + "\n" +
|
||
filename + "：" +
|
||
path_str +
|
||
Constants.FAILURE_REASON["File_More_Than_One"])
|
||
else:
|
||
path = temp_path[0]
|
||
freq = path.stem.split("_")[-1]
|
||
# 验证是否为整数或是否存在
|
||
if not freq.isdigit():
|
||
return Result().failure(
|
||
info=Constants.INPUT_FAILURE + "\n" +
|
||
filename + "：" +
|
||
Constants.FAILURE_REASON["Data_Frequency_Not_In_Filename"])
|
||
if Path(path).suffix != suffix:
|
||
return Result().failure(
|
||
info=Constants.INPUT_FAILURE + "\n" +
|
||
filename + "：" +
|
||
Constants.FAILURE_REASON["Suffix_Not_Correct"])
|
||
data = {
|
||
"path": str(path),
|
||
"freq": int(freq)
|
||
}
|
||
return Result().success(data=data)
|
||
|
||
@staticmethod
|
||
def examine_artifact(artifact):
|
||
# 检查体动标签正确性，长度
|
||
if len(artifact) % 4 != 0:
|
||
return Result().failure(info=Constants.INPUT_FAILURE +
|
||
Constants.FAILURE_REASON["Artifact_Format_Not_Correct"])
|
||
for i in range(0, len(artifact), 4):
|
||
unit_data = artifact[i:i + 4]
|
||
if len(unit_data) < 4:
|
||
break |