新版本重构了ai模型提供商逻辑,新增MCUNC,阿里云百炼ai提供商,支持语言可配置,优化日志输出格式
This commit is contained in:
@@ -1,14 +1,29 @@
|
||||
import logging
|
||||
from model.logger import setup_logger
|
||||
import toml
|
||||
import sqlite3
|
||||
|
||||
|
||||
|
||||
logger = setup_logger()
|
||||
class Clear:
|
||||
def __init__(self, user_id,group_id):
|
||||
self.user_id = user_id
|
||||
self.group_id = group_id
|
||||
|
||||
def main(self):
|
||||
with open("config.toml", "r", encoding="utf-8"):
|
||||
config = toml.load("config.toml")
|
||||
clean_all_success_message = config.get("clean_all_success_message")
|
||||
clean_one_success_message = config.get("clean_one_success_message")
|
||||
clean_one_notfound_message = config.get("clean_one_notfound_message")
|
||||
clean_fail_message = config.get("clean_fail_message")
|
||||
clean_nopermissoin_message = config.get("clean_nopermissoin_message")
|
||||
|
||||
if "{group_id}" in clean_one_success_message:
|
||||
clean_one_success_message.replace("{group_id}", self.group_id)
|
||||
if "{group_id}" in clean_one_notfound_message:
|
||||
clean_one_notfound_message.replace("{group_id}", self.group_id)
|
||||
|
||||
if self.is_root():
|
||||
if self.group_id == "all":
|
||||
try:
|
||||
@@ -17,13 +32,13 @@ class Clear:
|
||||
# 清空表中所有数据
|
||||
cursor.execute("DELETE FROM groups;") # 假设表名为 uuid_table,请根据实际表名修改
|
||||
conn.commit()
|
||||
logging.info("✅ 数据库表已成功清空")
|
||||
return "✅ 已清空所有群组数据"
|
||||
logger.info("已清空所有群组数据")
|
||||
return clean_all_success_message
|
||||
except sqlite3.Error as e:
|
||||
if 'conn' in locals():
|
||||
conn.close()
|
||||
logging.error(f"❌ 数据库操作失败: {e}")
|
||||
return "❌ 数据库操作失败"
|
||||
logger.error(f"数据库操作失败: {e}")
|
||||
return clean_fail_message
|
||||
else:
|
||||
try:
|
||||
with sqlite3.connect("uuid.db") as conn:
|
||||
@@ -32,19 +47,19 @@ class Clear:
|
||||
cursor.execute("DELETE FROM groups WHERE group_openid = ?", (self.group_id,))
|
||||
conn.commit()
|
||||
if cursor.rowcount > 0:
|
||||
logging.info(f"✅ 已成功删除 group_id = {self.group_id} 的数据")
|
||||
return f"✅ 已成功删除 group_id = {self.group_id} 的数据"
|
||||
logger.info(f"✅ 已成功删除 group_id = {self.group_id} 的数据")
|
||||
return clean_one_success_message
|
||||
else:
|
||||
logging.info(f"⚠️ 没有找到 group_id = {self.group_id} 的数据")
|
||||
return f"⚠️ 没有找到 group_id = {self.group_id} 的数据"
|
||||
logger.info(f"⚠️ 没有找到 group_id = {self.group_id} 的数据")
|
||||
return clean_one_notfound_message
|
||||
except sqlite3.Error as e:
|
||||
if 'conn' in locals():
|
||||
conn.close()
|
||||
logging.error(f"❌ 数据库操作失败: {e}")
|
||||
return "❌ 数据库操作失败"
|
||||
logger.error(f"❌ 数据库操作失败: {e}")
|
||||
return clean_fail_message
|
||||
|
||||
else:
|
||||
return "你不是管理员哦喵~"
|
||||
return clean_nopermissoin_message
|
||||
|
||||
def is_root(self):
|
||||
with open("./config.toml", "r", encoding="utf-8") as f:
|
||||
|
||||
Reference in New Issue
Block a user