MaiBot/test_output_manager.py

48 lines
1.3 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters!

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

import time
import sys
from src.common.output_manager import OutputManager
def main():
# 初始化输出管理器
output_manager = OutputManager()
# 启动输出捕获
output_manager.start_capture()
print("输出管理器已启动WebSocket服务器运行在 ws://localhost:8765")
print("您可以使用任何WebSocket客户端连接到此地址来接收输出")
# 发送一些测试消息
print("这是一条标准输出消息")
sys.stderr.write("这是一条错误输出消息\n")
# 发送自定义消息
output_manager.send_custom_message(
"status", {"progress": 50, "status": "processing"}, "task_manager"
)
# 每隔1秒发送一条消息
try:
count = 0
while True:
count += 1
print(f"测试消息 #{count}")
if count % 5 == 0:
sys.stderr.write(f"错误消息 #{count}\n")
# 每10条消息发送一次自定义消息
if count % 10 == 0:
output_manager.send_custom_message(
"progress", {"count": count, "percentage": count % 100}, "counter"
)
time.sleep(1)
except KeyboardInterrupt:
# 停止输出捕获
output_manager.stop_capture()
print("程序已退出")
if __name__ == "__main__":
main()