mirror of
https://github.com/Zippland/Bubbles.git
synced 2026-01-19 01:21:15 +08:00
110 lines
3.9 KiB
Python
110 lines
3.9 KiB
Python
#! /usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
import signal
|
||
import logging
|
||
import sys # 导入 sys 模块
|
||
import os
|
||
from argparse import ArgumentParser
|
||
|
||
# 确保日志目录存在
|
||
log_dir = "logs"
|
||
if not os.path.exists(log_dir):
|
||
os.makedirs(log_dir)
|
||
|
||
# 配置 logging
|
||
log_format = '%(asctime)s - %(levelname)s - %(name)s - %(message)s'
|
||
logging.basicConfig(
|
||
level=logging.WARNING, # 提高默认日志级别为 WARNING，只显示警告和错误信息
|
||
format=log_format,
|
||
handlers=[
|
||
# logging.FileHandler(os.path.join(log_dir, "app.log"), encoding='utf-8'), # 将所有日志写入文件
|
||
# logging.StreamHandler(sys.stdout) # 同时输出到控制台
|
||
]
|
||
)
|
||
|
||
# 为特定模块设置更具体的日志级别
|
||
logging.getLogger("requests").setLevel(logging.ERROR) # 提高为 ERROR
|
||
logging.getLogger("urllib3").setLevel(logging.ERROR) # 提高为 ERROR
|
||
logging.getLogger("httpx").setLevel(logging.ERROR) # 提高为 ERROR
|
||
|
||
# 常见的自定义模块日志设置，按需修改
|
||
logging.getLogger("Weather").setLevel(logging.WARNING)
|
||
logging.getLogger("ai_providers").setLevel(logging.WARNING)
|
||
logging.getLogger("commands").setLevel(logging.WARNING)
|
||
# 临时调试：为AI路由器设置更详细的日志级别
|
||
logging.getLogger("commands.ai_router").setLevel(logging.INFO)
|
||
|
||
from configuration import Config
|
||
from constants import ChatType
|
||
from robot import Robot, __version__
|
||
from wcferry import Wcf
|
||
|
||
def main(chat_type: int):
|
||
config = Config()
|
||
wcf = Wcf(debug=False) # 将 debug 设置为 False 减少 wcf 的调试输出
|
||
|
||
# 定义全局变量robot，使其在handler中可访问
|
||
global robot
|
||
robot = Robot(config, wcf, chat_type)
|
||
|
||
def handler(sig, frame):
|
||
# 先清理机器人资源（包括关闭数据库连接）
|
||
if 'robot' in globals() and robot:
|
||
robot.LOG.info("程序退出，开始清理资源...")
|
||
robot.cleanup()
|
||
|
||
# 再清理wcf环境
|
||
wcf.cleanup() # 退出前清理环境
|
||
exit(0)
|
||
|
||
signal.signal(signal.SIGINT, handler)
|
||
|
||
robot.LOG.info(f"WeChatRobot【{__version__}】成功启动···")
|
||
|
||
# # 机器人启动发送测试消息
|
||
# robot.sendTextMsg("机器人启动成功！", "filehelper")
|
||
|
||
# 接收消息
|
||
# robot.enableRecvMsg() # 可能会丢消息？
|
||
robot.enableReceivingMsg() # 加队列
|
||
|
||
# 每天 7 点发送天气预报
|
||
robot.onEveryTime("07:00", robot.weatherReport)
|
||
|
||
# 每天 7:30 发送新闻
|
||
robot.onEveryTime("07:30", robot.newsReport)
|
||
|
||
|
||
# 让机器人一直跑
|
||
robot.keepRunningAndBlockProcess()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
parser = ArgumentParser()
|
||
parser.add_argument('-c', type=int, default=0,
|
||
help=f'选择默认模型参数序号: {ChatType.help_hint()}（可通过配置文件为不同群指定模型）')
|
||
parser.add_argument('-d', '--debug', action='store_true',
|
||
help='启用调试模式，输出更详细的日志信息')
|
||
parser.add_argument('-q', '--quiet', action='store_true',
|
||
help='安静模式，只输出错误信息')
|
||
parser.add_argument('-v', '--verbose', action='store_true',
|
||
help='详细输出模式，显示所有信息日志')
|
||
args = parser.parse_args()
|
||
|
||
# 处理日志级别参数
|
||
if args.debug:
|
||
# 调试模式优先级最高
|
||
logging.getLogger().setLevel(logging.DEBUG)
|
||
print("已启用调试模式，将显示所有详细日志信息")
|
||
elif args.quiet:
|
||
# 安静模式，控制台只显示错误
|
||
logging.getLogger().setLevel(logging.ERROR)
|
||
print("已启用安静模式，控制台只显示错误信息")
|
||
elif args.verbose:
|
||
# 详细模式，显示所有 INFO 级别日志
|
||
logging.getLogger().setLevel(logging.INFO)
|
||
print("已启用详细模式，将显示所有信息日志")
|
||
|
||
main(args.c)
|