157 lines
4.9 KiB
Python
157 lines
4.9 KiB
Python
#! /usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
import asyncio
|
||
import threading
|
||
|
||
from admin.GlancesMonitor import GlancesMonitor
|
||
from utils.decorator.async_job import async_job
|
||
from configuration import Config
|
||
from plugins.xiuren_image.images_cache import ImageCacheManager
|
||
from robot import Robot
|
||
|
||
from loguru import logger
|
||
|
||
from utils.sehuatang.sehuatang_bot import SehuatangCrawler
|
||
|
||
# INFO 日志（包含 INFO、DEBUG，但不包含 WARNING、ERROR）
|
||
logger.add(
|
||
f"logs/wx_info.log",
|
||
level="INFO",
|
||
filter=lambda record: record["level"].name in ["INFO", "DEBUG"],
|
||
rotation="10 MB",
|
||
retention="7 days",
|
||
encoding="utf-8"
|
||
)
|
||
|
||
# ERROR 日志（仅 ERROR 及以上）
|
||
logger.add(
|
||
f"logs/wx_error.log",
|
||
level="ERROR",
|
||
rotation="10 MB",
|
||
retention="7 days",
|
||
encoding="utf-8"
|
||
)
|
||
# ERROR 日志（仅 ERROR 及以上）
|
||
logger.add(
|
||
f"logs/wx_debug.log",
|
||
level="DEBUG",
|
||
rotation="10 MB",
|
||
retention="7 days",
|
||
encoding="utf-8"
|
||
)
|
||
|
||
|
||
def main():
|
||
config = Config()
|
||
|
||
# 创建机器人实例
|
||
robot = Robot(config)
|
||
robot.LOG.info(f"ABOT服务 正在启动...")
|
||
|
||
# 初始化并启动wechat_ipad客户端
|
||
if robot.init_wechat_ipad():
|
||
robot.LOG.info("wechat_ipad客户端启动成功")
|
||
else:
|
||
robot.LOG.error("wechat_ipad客户端启动失败")
|
||
# 注册定时任务
|
||
jobs(robot)
|
||
# 启动Dashboard服务器
|
||
try:
|
||
# 创建Dashboard服务器实例，共享robot对象
|
||
from admin.dashboard.server import DashboardServer
|
||
dashboard_server = DashboardServer(robot_instance=robot)
|
||
|
||
# 在单独的线程中启动Dashboard服务器
|
||
dashboard_thread = threading.Thread(target=dashboard_server.run, daemon=True)
|
||
dashboard_thread.start()
|
||
robot.LOG.info(f"Dashboard服务器已在 http://{dashboard_server.host}:{dashboard_server.port} 启动")
|
||
except Exception as e:
|
||
robot.LOG.error(f"Dashboard服务器启动失败: {e}")
|
||
try:
|
||
robot.LOG.debug(f"开始启动GlancesMonitor")
|
||
# 初始化 Glances 监控
|
||
monitor = GlancesMonitor(
|
||
email_sender=robot.email_sender,
|
||
host=config.glances.get("host"),
|
||
port=config.glances.get("port"),
|
||
cpu_threshold=80.0,
|
||
load_threshold=16, # 自动设为 CPU 核心数 * 2
|
||
io_threshold=100.0,
|
||
disk_usage_threshold=60.0,
|
||
handle_threshold=20000,
|
||
recipient=config.email.get("alert_recipient")
|
||
)
|
||
monitor.run()
|
||
except Exception as e:
|
||
robot.LOG.error(f"GlancesMonitor服务器启动失败: {e}")
|
||
|
||
robot.LOG.info(f"=" * 30)
|
||
asyncio.run(async_job.run_all())
|
||
# 让机器人一直跑
|
||
robot.keep_running_and_block_process()
|
||
|
||
|
||
def jobs(robot: Robot):
|
||
# ✅ 每天 8:30 发送百度新闻
|
||
@async_job.at_times(["08:30"])
|
||
async def news_baidu_report_auto_job():
|
||
await robot.news_baidu_report_auto()
|
||
|
||
# ✅ 每天 10:30 推送 Epic 免费游戏
|
||
@async_job.every_weekday_time(weekday=4, time_str="10:00") # 0=周一，4=周五
|
||
async def epic_job():
|
||
await robot.send_epic_free_games()
|
||
|
||
# ✅ 每天 02:30 从 redis 写入 sqlite
|
||
@async_job.at_times(["02:30"])
|
||
async def msg_count_to_db_job():
|
||
await robot.message_count_to_db()
|
||
|
||
# ✅ 每天 09:30 从 sqlite 读取并发送群排行
|
||
@async_job.at_times(["09:30"])
|
||
async def msg_ranking_job():
|
||
await robot.generate_and_send_ranking()
|
||
|
||
# ✅ 每天 15:30 发涩图 PDF
|
||
@async_job.at_times(["15:30"])
|
||
async def sehuatang_pdf_job():
|
||
await robot.generate_sehuatang_pdf()
|
||
|
||
# ✅ 每天 01:30 下载秀人网帖子
|
||
@async_job.at_times(["01:30"])
|
||
async def xiuren_download_job():
|
||
await robot.xiu_ren_download_task()
|
||
|
||
# ✅ 每天 01:30 下载秀人网帖子
|
||
@async_job.at_times(["2:30"])
|
||
async def shenshiR15_download_job():
|
||
await robot.shen_shi_download_task()
|
||
|
||
# ✅ 每天 17:30 发秀人 PDF（如果启用）
|
||
# @async_job.at_times(["17:30"])
|
||
# async def xiuren_pdf_send_job():
|
||
# await robot.xiu_ren_pdf_send()
|
||
|
||
# ✅ 每 3 小时登录验证
|
||
@async_job.at_times(["14:43"])
|
||
async def login_check_job():
|
||
await robot.login_twice_auto_auth()
|
||
|
||
@async_job.at_times(["05:00"])
|
||
async def update_image_cache_job():
|
||
logger.info("开始执行图片缓存更新任务")
|
||
manager = ImageCacheManager("/mnt/nfs_share") # 替换为你的图片目录
|
||
await manager.update_image_cache()
|
||
logger.info("图片缓存更新完成")
|
||
|
||
# ✅ 每2分钟处理一次待下载的图片消息（串行处理，避免数据库锁竞争）
|
||
@async_job.every_minutes(5)
|
||
async def process_pending_images_job():
|
||
if hasattr(robot, 'message_storage') and robot.message_storage:
|
||
await robot.message_storage.process_pending_images(minutes_ago=10, batch_size=20)
|
||
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|