78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
#! /usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
import signal
|
||
import threading
|
||
from argparse import ArgumentParser
|
||
|
||
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=True)
|
||
|
||
def handler(sig, frame):
|
||
# 在退出前先关闭插件系统
|
||
robot.plugin_manager.shutdown_plugins()
|
||
wcf.cleanup() # 退出前清理环境
|
||
exit(0)
|
||
|
||
signal.signal(signal.SIGINT, handler)
|
||
|
||
robot = Robot(config, wcf, chat_type)
|
||
robot.LOG.info(f"WeChatRobot【{__version__}】成功启动···")
|
||
|
||
# 机器人启动发送测试消息
|
||
robot.send_text_msg("启动成功！", "filehelper")
|
||
|
||
# 接收消息
|
||
robot.enableReceivingMsg() # 加队列
|
||
# 每天 8:30 发送新闻
|
||
robot.onEveryTime("08:30", robot.news_baidu_report_auto)
|
||
|
||
# epic
|
||
robot.onEveryTime("10:30", robot.send_epic_free_games)
|
||
|
||
# message report 1:数据自动从redis 转到sqllite
|
||
robot.onEveryTime("00:30", robot.message_count_to_db)
|
||
# 从db中提取并发送给相关群
|
||
robot.onEveryTime("09:30", robot.generate_and_send_ranking)
|
||
|
||
# sehuatang
|
||
robot.onEveryTime("15:30", robot.generate_sehuatang_pdf)
|
||
|
||
# 游戏的定时任务每小时执行
|
||
robot.onEveryTime("17:58", robot.game_auto_tasks)
|
||
|
||
# 秀人网每天自动下载帖子
|
||
robot.onEveryTime("01:30", robot.xiu_ren_download_task)
|
||
|
||
# 秀人网每天自动发pdf
|
||
robot.onEveryTime("17:30", robot.xiu_ren_pdf_send)
|
||
|
||
# 启动Dashboard服务器
|
||
dashboard_server = None
|
||
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}")
|
||
|
||
# 让机器人一直跑
|
||
robot.keep_running_and_block_process()
|
||
|
||
if __name__ == "__main__":
|
||
parser = ArgumentParser()
|
||
parser.add_argument('-c', type=int, default=0, help=f'选择模型参数序号: {ChatType.help_hint()}')
|
||
args = parser.parse_args().c
|
||
main(args)
|