feat: 初始化黄小瓜AI助手记忆仓库
- 核心配置: IDENTITY, USER, SOUL, AGENTS, TOOLS, HEARTBEAT, MEMORY - memory/: 每日总结和临时记录 - skills/: 所有已安装技能 - notes/: 语音配置笔记
This commit is contained in:
107
skills/stock-monitor-pro/scripts/monitor_daemon.py
Normal file
107
skills/stock-monitor-pro/scripts/monitor_daemon.py
Normal file
@@ -0,0 +1,107 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Stock Monitor Daemon - 后台常驻进程
|
||||
自动运行监控,智能控制频率,支持 graceful shutdown
|
||||
"""
|
||||
|
||||
import sys
|
||||
import time
|
||||
import signal
|
||||
import logging
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
|
||||
# 设置日志
|
||||
log_dir = Path.home() / ".stock_monitor"
|
||||
log_dir.mkdir(exist_ok=True)
|
||||
logging.basicConfig(
|
||||
level=logging.INFO,
|
||||
format='%(asctime)s - %(levelname)s - %(message)s',
|
||||
handlers=[
|
||||
logging.FileHandler(log_dir / "monitor.log"),
|
||||
logging.StreamHandler(sys.stdout)
|
||||
]
|
||||
)
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# 导入监控类
|
||||
sys.path.insert(0, str(Path(__file__).parent))
|
||||
from monitor import StockAlert, WATCHLIST
|
||||
|
||||
class MonitorDaemon:
|
||||
def __init__(self):
|
||||
self.monitor = StockAlert()
|
||||
self.running = True
|
||||
self.last_run_time = 0
|
||||
|
||||
# 设置信号处理
|
||||
signal.signal(signal.SIGTERM, self.handle_shutdown)
|
||||
signal.signal(signal.SIGINT, self.handle_shutdown)
|
||||
|
||||
def handle_shutdown(self, signum, frame):
|
||||
"""优雅退出"""
|
||||
logger.info(f"收到信号 {signum},正在关闭...")
|
||||
self.running = False
|
||||
|
||||
def get_sleep_interval(self):
|
||||
"""根据当前时间获取睡眠间隔"""
|
||||
schedule = self.monitor.should_run_now()
|
||||
if not schedule.get("run"):
|
||||
# 如果当前不需要运行,计算到下次运行的时间
|
||||
now = datetime.now()
|
||||
hour = now.hour
|
||||
|
||||
# 凌晨时段,1小时后检查
|
||||
if 0 <= hour < 9:
|
||||
return 3600
|
||||
return 300 # 默认5分钟
|
||||
|
||||
return schedule.get("interval", 300)
|
||||
|
||||
def run(self):
|
||||
"""主循环"""
|
||||
logger.info("=" * 60)
|
||||
logger.info("🚀 Stock Monitor Daemon 启动")
|
||||
logger.info(f"📋 监控标的: {len(WATCHLIST)} 只")
|
||||
logger.info("=" * 60)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
# 检查是否应该执行
|
||||
schedule = self.monitor.should_run_now()
|
||||
|
||||
if schedule.get("run"):
|
||||
mode = schedule.get("mode", "normal")
|
||||
stocks_count = len(schedule.get("stocks", []))
|
||||
logger.info(f"[{mode}] 扫描 {stocks_count} 只标的...")
|
||||
|
||||
# 执行监控
|
||||
alerts = self.monitor.run_once(smart_mode=False) # 已经判断过了
|
||||
|
||||
if alerts:
|
||||
logger.info(f"⚠️ 触发 {len(alerts)} 条预警")
|
||||
# 这里会通过 message 工具发送通知
|
||||
else:
|
||||
logger.debug("✅ 无预警")
|
||||
|
||||
self.last_run_time = time.time()
|
||||
|
||||
# 计算睡眠间隔
|
||||
sleep_interval = self.get_sleep_interval()
|
||||
logger.debug(f"下次检查: {sleep_interval} 秒后")
|
||||
|
||||
# 分段睡眠,方便及时响应退出信号
|
||||
slept = 0
|
||||
while slept < sleep_interval and self.running:
|
||||
time.sleep(1)
|
||||
slept += 1
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"运行出错: {e}", exc_info=True)
|
||||
time.sleep(60) # 出错后等待1分钟重试
|
||||
|
||||
logger.info("👋 Daemon 已停止")
|
||||
|
||||
if __name__ == '__main__':
|
||||
daemon = MonitorDaemon()
|
||||
daemon.run()
|
||||
Reference in New Issue
Block a user