feat: 新增股票数据波形图和截图功能
refactor: 重构数据库和LLM分析器逻辑 fix: 修复爬虫解析和UI显示问题 docs: 更新配置文件和注释 style: 优化代码格式和日志输出
This commit is contained in:
43
test_screenshot.py
Normal file
43
test_screenshot.py
Normal file
@@ -0,0 +1,43 @@
|
||||
"""
|
||||
测试截图功能
|
||||
"""
|
||||
import os
|
||||
from screenshot_manager import ScreenshotManager
|
||||
from loguru import logger
|
||||
|
||||
|
||||
def test_screenshot():
|
||||
"""测试截图功能"""
|
||||
logger.info("开始测试截图功能")
|
||||
|
||||
# 创建截图管理器
|
||||
screenshot_manager = ScreenshotManager()
|
||||
|
||||
# 测试截图
|
||||
screenshot_path = screenshot_manager.capture_chart_screenshot()
|
||||
|
||||
if screenshot_path:
|
||||
logger.info(f"✅ 截图成功: {screenshot_path}")
|
||||
|
||||
# 检查文件是否存在
|
||||
if os.path.exists(screenshot_path):
|
||||
file_size = os.path.getsize(screenshot_path)
|
||||
logger.info(f"截图文件大小: {file_size} 字节")
|
||||
|
||||
# 获取最新截图
|
||||
latest_screenshot = screenshot_manager.get_latest_screenshot()
|
||||
logger.info(f"最新截图: {latest_screenshot}")
|
||||
|
||||
# 清理旧截图
|
||||
screenshot_manager.cleanup_old_screenshots(keep_count=2)
|
||||
logger.info("清理旧截图完成")
|
||||
else:
|
||||
logger.error("截图文件不存在")
|
||||
else:
|
||||
logger.error("截图失败")
|
||||
|
||||
logger.info("截图功能测试完成")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
test_screenshot()
|
||||
Reference in New Issue
Block a user