Merge branch 'dev' of https://github.com/MaiM-with-u/MaiBot into dev
This commit is contained in:
@@ -3,7 +3,7 @@ from tkinter import ttk, messagebox, filedialog
|
|||||||
import tomli
|
import tomli
|
||||||
import tomli_w
|
import tomli_w
|
||||||
import os
|
import os
|
||||||
from typing import Any, Dict, List, Union
|
from typing import Any, Dict, List
|
||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
import re
|
import re
|
||||||
@@ -271,8 +271,11 @@ class ConfigEditor:
|
|||||||
already_in_quick = any(item.get("path") == full_config_path_key for item in quick_settings)
|
already_in_quick = any(item.get("path") == full_config_path_key for item in quick_settings)
|
||||||
icon = "★" if already_in_quick else "☆"
|
icon = "★" if already_in_quick else "☆"
|
||||||
icon_fg = "#FFD600" # 始终金色
|
icon_fg = "#FFD600" # 始终金色
|
||||||
|
|
||||||
def on_star_click():
|
def on_star_click():
|
||||||
self.toggle_quick_setting(full_config_path_key, widget_type, item_name_to_display, item_desc_to_display, already_in_quick)
|
self.toggle_quick_setting(
|
||||||
|
full_config_path_key, widget_type, item_name_to_display, item_desc_to_display, already_in_quick
|
||||||
|
)
|
||||||
# 立即刷新本分组
|
# 立即刷新本分组
|
||||||
for widget in parent.winfo_children():
|
for widget in parent.winfo_children():
|
||||||
widget.destroy()
|
widget.destroy()
|
||||||
@@ -319,7 +322,7 @@ class ConfigEditor:
|
|||||||
# 布尔值使用复选框
|
# 布尔值使用复选框
|
||||||
var = tk.BooleanVar(value=value)
|
var = tk.BooleanVar(value=value)
|
||||||
checkbox = ttk.Checkbutton(frame, variable=var, command=lambda: self.on_value_changed())
|
checkbox = ttk.Checkbutton(frame, variable=var, command=lambda: self.on_value_changed())
|
||||||
checkbox.grid(row=widget_row, column=0, columnspan=content_col_offset_for_star +1, sticky=tk.W, padx=5)
|
checkbox.grid(row=widget_row, column=0, columnspan=content_col_offset_for_star + 1, sticky=tk.W, padx=5)
|
||||||
self.widgets[tuple(path + [key])] = var
|
self.widgets[tuple(path + [key])] = var
|
||||||
widget_type = "bool"
|
widget_type = "bool"
|
||||||
|
|
||||||
@@ -335,13 +338,17 @@ class ConfigEditor:
|
|||||||
elif isinstance(value, list):
|
elif isinstance(value, list):
|
||||||
# 列表使用每行一个输入框的形式
|
# 列表使用每行一个输入框的形式
|
||||||
frame_list = ttk.Frame(frame)
|
frame_list = ttk.Frame(frame)
|
||||||
frame_list.grid(row=widget_row, column=0, columnspan=content_col_offset_for_star +1, sticky=tk.W+tk.E, padx=5)
|
frame_list.grid(
|
||||||
|
row=widget_row, column=0, columnspan=content_col_offset_for_star + 1, sticky=tk.W + tk.E, padx=5
|
||||||
|
)
|
||||||
|
|
||||||
# 创建添加和删除按钮
|
# 创建添加和删除按钮
|
||||||
button_frame = ttk.Frame(frame_list)
|
button_frame = ttk.Frame(frame_list)
|
||||||
button_frame.pack(side=tk.RIGHT, padx=5)
|
button_frame.pack(side=tk.RIGHT, padx=5)
|
||||||
|
|
||||||
add_button = ttk.Button(button_frame, text="+", width=3, command=lambda p=path + [key]: self.add_list_item(frame_list, p))
|
add_button = ttk.Button(
|
||||||
|
button_frame, text="+", width=3, command=lambda p=path + [key]: self.add_list_item(frame_list, p)
|
||||||
|
)
|
||||||
add_button.pack(side=tk.TOP, pady=2)
|
add_button.pack(side=tk.TOP, pady=2)
|
||||||
|
|
||||||
# 创建列表项框架
|
# 创建列表项框架
|
||||||
@@ -548,10 +555,10 @@ class ConfigEditor:
|
|||||||
if setting_type == "bool":
|
if setting_type == "bool":
|
||||||
value = bool(value) if value is not None else False
|
value = bool(value) if value is not None else False
|
||||||
var = tk.BooleanVar(value=value)
|
var = tk.BooleanVar(value=value)
|
||||||
checkbox = ttk.Checkbutton(frame, text="",
|
checkbox = ttk.Checkbutton(
|
||||||
variable=var,
|
frame, text="", variable=var, command=lambda p=path, v=var: self.on_quick_setting_changed(p, v)
|
||||||
command=lambda p=path, v=var: self.on_quick_setting_changed(p, v))
|
)
|
||||||
checkbox.pack(anchor=tk.W, padx=5, pady=(0,5))
|
checkbox.pack(anchor=tk.W, padx=5, pady=(0, 5))
|
||||||
|
|
||||||
elif setting_type == "text":
|
elif setting_type == "text":
|
||||||
value = str(value) if value is not None else ""
|
value = str(value) if value is not None else ""
|
||||||
@@ -569,9 +576,10 @@ class ConfigEditor:
|
|||||||
|
|
||||||
elif setting_type == "list":
|
elif setting_type == "list":
|
||||||
# 对于列表类型,创建一个按钮来打开编辑窗口
|
# 对于列表类型,创建一个按钮来打开编辑窗口
|
||||||
button = ttk.Button(frame, text="编辑列表",
|
button = ttk.Button(
|
||||||
command=lambda p=path, s=setting: self.open_list_editor(p, s))
|
frame, text="编辑列表", command=lambda p=path, s=setting: self.open_list_editor(p, s)
|
||||||
button.pack(anchor=tk.W, padx=5, pady=(0,5))
|
)
|
||||||
|
button.pack(anchor=tk.W, padx=5, pady=(0, 5))
|
||||||
|
|
||||||
def create_list_item(self, parent, value, index, entry_vars, path):
|
def create_list_item(self, parent, value, index, entry_vars, path):
|
||||||
"""创建单个列表项的输入框"""
|
"""创建单个列表项的输入框"""
|
||||||
@@ -585,8 +593,12 @@ class ConfigEditor:
|
|||||||
var.trace_add("write", lambda *args: self.on_value_changed())
|
var.trace_add("write", lambda *args: self.on_value_changed())
|
||||||
|
|
||||||
# 创建删除按钮
|
# 创建删除按钮
|
||||||
del_button = ttk.Button(item_frame, text="-", width=3,
|
del_button = ttk.Button(
|
||||||
command=lambda: self.remove_list_item(parent, item_frame, entry_vars, index, path))
|
item_frame,
|
||||||
|
text="-",
|
||||||
|
width=3,
|
||||||
|
command=lambda: self.remove_list_item(parent, item_frame, entry_vars, index, path),
|
||||||
|
)
|
||||||
del_button.pack(side=tk.RIGHT, padx=5)
|
del_button.pack(side=tk.RIGHT, padx=5)
|
||||||
|
|
||||||
# 存储变量引用
|
# 存储变量引用
|
||||||
@@ -757,13 +769,13 @@ class ConfigEditor:
|
|||||||
button_frame.pack(fill=tk.X, pady=10)
|
button_frame.pack(fill=tk.X, pady=10)
|
||||||
|
|
||||||
# 添加按钮
|
# 添加按钮
|
||||||
add_button = ttk.Button(button_frame, text="添加",
|
add_button = ttk.Button(button_frame, text="添加", command=lambda: self.add_list_item(items_frame, path))
|
||||||
command=lambda: self.add_list_item(items_frame, path))
|
|
||||||
add_button.pack(side=tk.LEFT, padx=5)
|
add_button.pack(side=tk.LEFT, padx=5)
|
||||||
|
|
||||||
# 保存按钮
|
# 保存按钮
|
||||||
save_button = ttk.Button(button_frame, text="保存",
|
save_button = ttk.Button(
|
||||||
command=lambda: self.save_list_editor(dialog, path, entry_vars))
|
button_frame, text="保存", command=lambda: self.save_list_editor(dialog, path, entry_vars)
|
||||||
|
)
|
||||||
save_button.pack(side=tk.RIGHT, padx=5)
|
save_button.pack(side=tk.RIGHT, padx=5)
|
||||||
|
|
||||||
def save_list_editor(self, dialog, path, entry_vars):
|
def save_list_editor(self, dialog, path, entry_vars):
|
||||||
@@ -810,20 +822,21 @@ class ConfigEditor:
|
|||||||
self.on_value_changed()
|
self.on_value_changed()
|
||||||
|
|
||||||
def toggle_quick_setting(self, full_path, widget_type, name, desc, already_in_quick):
|
def toggle_quick_setting(self, full_path, widget_type, name, desc, already_in_quick):
|
||||||
quick_settings = self.editor_config.setdefault("editor", {}).setdefault("quick_settings", {}).setdefault("items", [])
|
quick_settings = (
|
||||||
|
self.editor_config.setdefault("editor", {}).setdefault("quick_settings", {}).setdefault("items", [])
|
||||||
|
)
|
||||||
if already_in_quick:
|
if already_in_quick:
|
||||||
# 移除
|
# 移除
|
||||||
self.editor_config["editor"]["quick_settings"]["items"] = [item for item in quick_settings if item.get("path") != full_path]
|
self.editor_config["editor"]["quick_settings"]["items"] = [
|
||||||
|
item for item in quick_settings if item.get("path") != full_path
|
||||||
|
]
|
||||||
else:
|
else:
|
||||||
# 添加
|
# 添加
|
||||||
quick_settings.append({
|
quick_settings.append({"name": name, "description": desc, "path": full_path, "type": widget_type})
|
||||||
"name": name,
|
|
||||||
"description": desc,
|
|
||||||
"path": full_path,
|
|
||||||
"type": widget_type
|
|
||||||
})
|
|
||||||
# 保存到configexe.toml
|
# 保存到configexe.toml
|
||||||
import tomli_w, os
|
import tomli_w
|
||||||
|
import os
|
||||||
|
|
||||||
config_path = os.path.join(os.path.dirname(__file__), "configexe.toml")
|
config_path = os.path.join(os.path.dirname(__file__), "configexe.toml")
|
||||||
with open(config_path, "wb") as f:
|
with open(config_path, "wb") as f:
|
||||||
tomli_w.dump(self.editor_config, f)
|
tomli_w.dump(self.editor_config, f)
|
||||||
@@ -1087,8 +1100,9 @@ class ConfigEditor:
|
|||||||
|
|
||||||
def main():
|
def main():
|
||||||
root = tk.Tk()
|
root = tk.Tk()
|
||||||
app = ConfigEditor(root)
|
_app = ConfigEditor(root)
|
||||||
root.mainloop()
|
root.mainloop()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
|||||||
@@ -2,7 +2,11 @@ from src.common.logger_manager import get_logger
|
|||||||
from src.chat.message_receive.message import MessageRecv
|
from src.chat.message_receive.message import MessageRecv
|
||||||
from src.chat.message_receive.storage import MessageStorage
|
from src.chat.message_receive.storage import MessageStorage
|
||||||
from src.config.config import global_config
|
from src.config.config import global_config
|
||||||
|
from src.chat.message_receive.chat_stream import ChatStream
|
||||||
|
|
||||||
|
from maim_message import UserInfo
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
import re
|
||||||
|
|
||||||
logger = get_logger("pfc")
|
logger = get_logger("pfc")
|
||||||
|
|
||||||
@@ -14,7 +18,7 @@ class MessageProcessor:
|
|||||||
self.storage = MessageStorage()
|
self.storage = MessageStorage()
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _check_ban_words(text: str, chat, userinfo) -> bool:
|
def _check_ban_words(text: str, chat: ChatStream, userinfo: UserInfo) -> bool:
|
||||||
"""检查消息中是否包含过滤词"""
|
"""检查消息中是否包含过滤词"""
|
||||||
for word in global_config.message_receive.ban_words:
|
for word in global_config.message_receive.ban_words:
|
||||||
if word in text:
|
if word in text:
|
||||||
@@ -26,13 +30,12 @@ class MessageProcessor:
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _check_ban_regex(text: str, chat, userinfo) -> bool:
|
def _check_ban_regex(text: str, chat: ChatStream, userinfo: UserInfo) -> bool:
|
||||||
"""检查消息是否匹配过滤正则表达式"""
|
"""检查消息是否匹配过滤正则表达式"""
|
||||||
for pattern in global_config.chat.ban_msgs_regex:
|
for pattern in global_config.message_receive.ban_msgs_regex:
|
||||||
if pattern.search(text):
|
if re.search(pattern, text):
|
||||||
logger.info(
|
chat_name = chat.group_info.group_name if chat.group_info else "私聊"
|
||||||
f"[{chat.group_info.group_name if chat.group_info else '私聊'}]{userinfo.user_nickname}:{text}"
|
logger.info(f"[{chat_name}]{userinfo.user_nickname}:{text}")
|
||||||
)
|
|
||||||
logger.info(f"[正则表达式过滤]消息匹配到{pattern},filtered")
|
logger.info(f"[正则表达式过滤]消息匹配到{pattern},filtered")
|
||||||
return True
|
return True
|
||||||
return False
|
return False
|
||||||
|
|||||||
Reference in New Issue
Block a user