108 lines
3.5 KiB
Python
108 lines
3.5 KiB
Python
#! /usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
import logging
|
||
import threading
|
||
import time
|
||
from argparse import ArgumentParser
|
||
from gewechat_client import GewechatClient
|
||
import urllib.parse
|
||
from configuration import Config
|
||
from constants import ChatType
|
||
from gewechat.api.start_server import start_fastapi_server
|
||
from robot import Robot
|
||
|
||
# 配置日志
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
def main(chat_type: int):
|
||
config = Config()
|
||
base_url = config.BASE_URL
|
||
token = config.GEWECHAT_TOKEN
|
||
app_id = config.APP_ID
|
||
callback_url = config.CALLBACK_URL
|
||
send_msg_wxid = "filehelper" # 要发送消息的好友昵称
|
||
|
||
parsed_url = urllib.parse.urlparse(callback_url)
|
||
host = parsed_url.hostname or "0.0.0.0"
|
||
port = parsed_url.port or 8999
|
||
|
||
start_fastapi_server(host, port)
|
||
|
||
# 创建 GewechatClient 实例
|
||
client = GewechatClient(base_url, token)
|
||
|
||
# 登录, 自动创建二维码，扫码后自动登录
|
||
app_id, error_msg = client.login(app_id=app_id)
|
||
|
||
if error_msg:
|
||
logger.error("登录失败")
|
||
return
|
||
# 休眠等待server启动，防止回调设置失败
|
||
time.sleep(5)
|
||
resp = client.set_callback(token, callback_url)
|
||
print(f"set_callback:{resp}")
|
||
|
||
# 如果启动时，配置文件中的app_id为空，那么将app_id写入配置文件
|
||
if not config.APP_ID:
|
||
# 更新配置文件中的APP_ID
|
||
config.update_config('gewechat', 'app_id', app_id)
|
||
logger.info(f"已将新的APP_ID: {app_id} 写入配置文件")
|
||
# 同时更新当前配置对象中的APP_ID
|
||
config.APP_ID = app_id
|
||
|
||
# 创建机器人实例
|
||
robot = Robot(config, app_id, client, chat_type)
|
||
robot.LOG.info(f"WeChatRobot gewechat 成功启动···")
|
||
|
||
# # 注册Robot实例到callback模块
|
||
from gewechat.api.callback import register_robot
|
||
register_robot(app_id, robot)
|
||
|
||
# 机器人启动发送测试消息
|
||
client.post_text(app_id, send_msg_wxid, "client 启动成功！")
|
||
#
|
||
# # 每天 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("02: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("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)
|