Merge branch 'model_callback' into UI_2.0

This commit is contained in:
misyaguziya
2023-09-12 10:39:18 +09:00
2 changed files with 24 additions and 5 deletions

22
main.py
View File

@@ -33,7 +33,9 @@ def sendMicMessage(message):
view.printToTextbox_SentMessage(message, translation) view.printToTextbox_SentMessage(message, translation)
if config.ENABLE_LOGGER is True: if config.ENABLE_LOGGER is True:
model.logger.info(f"[SEND] {message} ({translation})") if len(translation) > 0:
translation = f" ({translation})"
model.logger.info(f"[SEND] {message}{translation}")
def startTranscriptionSendMessage(): def startTranscriptionSendMessage():
model.startMicTranscript(sendMicMessage) model.startMicTranscript(sendMicMessage)
@@ -61,7 +63,9 @@ def receiveSpeakerMessage(message):
model.notificationXSOverlay(xsoverlay_message) model.notificationXSOverlay(xsoverlay_message)
view.printToTextbox_ReceivedMessage(message, translation) view.printToTextbox_ReceivedMessage(message, translation)
if config.ENABLE_LOGGER is True: if config.ENABLE_LOGGER is True:
model.logger.info(f"[RECEIVE] {message} ({translation})") if len(translation) > 0:
translation = f" ({translation})"
model.logger.info(f"[RECEIVE] {message}{translation}")
def startTranscriptionReceiveMessage(): def startTranscriptionReceiveMessage():
model.startSpeakerTranscript(receiveSpeakerMessage) model.startSpeakerTranscript(receiveSpeakerMessage)
@@ -97,7 +101,9 @@ def sendChatMessage(message):
# update textbox message log # update textbox message log
view.printToTextbox_SentMessage(message, translation) view.printToTextbox_SentMessage(message, translation)
if config.ENABLE_LOGGER is True: if config.ENABLE_LOGGER is True:
model.logger.info(f"[SEND] {message} ({translation})") if len(translation) > 0:
translation = f" ({translation})"
model.logger.info(f"[SEND] {message}{translation}")
# delete message in entry message box # delete message in entry message box
if config.ENABLE_AUTO_CLEAR_MESSAGE_BOX is True: if config.ENABLE_AUTO_CLEAR_MESSAGE_BOX is True:
@@ -386,6 +392,15 @@ def callbackSetEnableNoticeXsoverlay(value):
print("callbackSetEnableNoticeXsoverlay", value) print("callbackSetEnableNoticeXsoverlay", value)
config.ENABLE_NOTICE_XSOVERLAY = value config.ENABLE_NOTICE_XSOVERLAY = value
def callbackSetEnableAutoExportMessageLogs(value):
print("callbackSetEnableAutoExportMessageLogs", value)
config.ENABLE_LOGGER = value
if config.ENABLE_LOGGER is True:
model.startLogger()
else:
model.stopLogger()
def callbackSetMessageFormat(value): def callbackSetMessageFormat(value):
print("callbackSetMessageFormat", value) print("callbackSetMessageFormat", value)
if len(value) > 0: if len(value) > 0:
@@ -487,6 +502,7 @@ view.register(
# Others Tab # Others Tab
"callback_set_enable_auto_clear_chatbox": callbackSetEnableAutoClearMessageBox, "callback_set_enable_auto_clear_chatbox": callbackSetEnableAutoClearMessageBox,
"callback_set_enable_notice_xsoverlay": callbackSetEnableNoticeXsoverlay, "callback_set_enable_notice_xsoverlay": callbackSetEnableNoticeXsoverlay,
"callback_set_enable_auto_export_message_logs": callbackSetEnableAutoExportMessageLogs,
"callback_set_message_format": callbackSetMessageFormat, "callback_set_message_format": callbackSetMessageFormat,
# Advanced Settings Tab # Advanced Settings Tab

View File

@@ -1,4 +1,6 @@
import sys
from os import makedirs from os import makedirs
from os import path as os_path
from datetime import datetime from datetime import datetime
from logging import getLogger, FileHandler, Formatter, INFO from logging import getLogger, FileHandler, Formatter, INFO
from time import sleep from time import sleep
@@ -82,15 +84,16 @@ class Model:
return result return result
def startLogger(self): def startLogger(self):
makedirs("./logs", exist_ok=True) makedirs(os_path.join(os_path.dirname(sys.argv[0]), "logs"), exist_ok=True)
logger = getLogger() logger = getLogger()
logger.setLevel(INFO) logger.setLevel(INFO)
file_name = f"./logs/{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.log" file_name = os_path.join(os_path.dirname(sys.argv[0]), "logs", f"{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.log")
file_handler = FileHandler(file_name, encoding="utf-8", delay=True) file_handler = FileHandler(file_name, encoding="utf-8", delay=True)
formatter = Formatter("[%(asctime)s] %(message)s") formatter = Formatter("[%(asctime)s] %(message)s")
file_handler.setFormatter(formatter) file_handler.setFormatter(formatter)
logger.addHandler(file_handler) logger.addHandler(file_handler)
self.logger = logger self.logger = logger
self.logger.disabled = False
def stopLogger(self): def stopLogger(self):
self.logger.disabled = True self.logger.disabled = True