【Python实时盯盘与预警 #16】天天手动python runpy盯盘systemd一键守护开机自启日志自动轮转

pip install mairui

/opt/stock-monitor/
├── monitor_daemon.py      # 守护脚本（本文）
├── logs/                   # 日志目录（自动创建）
│   ├── monitor.log         # 当前日志
│   ├── monitor.log.1       # 历史日志（自动轮转）
│   └── ...
└── stock-monitor.service   # systemd unit（置 /etc/systemd/system/）

# -*- coding: utf-8 -*-
"""盯盘常驻守护进程 — 系统级守护、阈值告警、日志自动轮转。"""
import sys, os, time, logging, argparse, signal
from datetime import datetime
from logging.handlers import RotatingFileHandler

# ── 证书 ──
from mairui import Client

# ── 配置 ──
WATCH_STOCKS = ["000001", "000002", "600519", "300750", "000858"]
ALERT_THRESHOLD_PC = 3.0      # 涨跌幅 ±3% 告警
ALERT_THRESHOLD_CJE = 1e9     # 成交额 10 亿关注
LOG_FILE = os.path.join(os.path.dirname(__file__), "logs", "monitor.log")
LOG_MAX_BYTES = 10 * 1024 * 1024   # 10 MB 自动轮转
LOG_BACKUP_COUNT = 7                # 保留 7 个备份

running = True


def graceful_shutdown(signum, frame):
    global running
    logging.info(f"收到信号 {signum}，正在优雅退出...")
    running = False


def setup_logging(use_stdout=False):
    fmt = logging.Formatter(
        "%(asctime)s [%(levelname)s] %(message)s",
        datefmt="%Y-%m-%d %H:%M:%S",
    )
    root = logging.getLogger()
    root.setLevel(logging.INFO)
    if use_stdout:
        root.addHandler(logging.StreamHandler(sys.stdout))
    else:
        os.makedirs(os.path.dirname(LOG_FILE), exist_ok=True)
        root.addHandler(RotatingFileHandler(
            LOG_FILE, maxBytes=LOG_MAX_BYTES,
            backupCount=LOG_BACKUP_COUNT, encoding="utf-8",
        ))
    return root


def check_market(api):
    """一轮盯盘检查"""
    logging.info(f"── 轮次开始 {datetime.now().strftime('%H:%M:%S')} ──")
    try:
        stocks = api.stock_ssjy_more(stock_codes=",".join(WATCH_STOCKS))
        alerts = 0
        for s in stocks:
            dm, pc, cje, p = s.get("dm"), s.get("pc", 0), s.get("cje", 0), s.get("p", 0)
            if abs(pc) >= ALERT_THRESHOLD_PC:
                logging.warning(f"[ALERT] {dm} 涨跌幅={pc:+.2f}% 超过{ALERT_THRESHOLD_PC}%！现价={p}")
                alerts += 1
            if cje >= ALERT_THRESHOLD_CJE:
                logging.info(f"[NOTE] {dm} 成交额={cje/1e8:.2f}亿 超过{ALERT_THRESHOLD_CJE/1e8:.0f}亿")
        logging.info(f"  本轮完成：{len(stocks)} 只，{alerts} 条告警")
    except Exception as e:
        logging.error(f"行情拉取失败：{e}")


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--dry-run", action="store_true", help="测试模式")
    parser.add_argument("--interval", type=int, default=60, help="轮询间隔秒")
    parser.add_argument("--stdout", action="store_true", help="systemd 模式")
    args = parser.parse_args()

    setup_logging(use_stdout=args.stdout)
    interval = 5 if args.dry_run else args.interval
    max_rounds = 3 if args.dry_run else float("inf")
    logging.info(f"盯盘守护进程启动 interval={interval}s")

    signal.signal(signal.SIGTERM, graceful_shutdown)
    signal.signal(signal.SIGINT, graceful_shutdown)

    api = Client(licence="<你的证书>")  # 演示证书，请替换成你自己的正式证书
    logging.info(f"证书已加载")

    round_count = 0
    while running and round_count < max_rounds:
        round_count += 1
        check_market(api)
        if round_count < max_rounds and running:
            time.sleep(interval)

    logging.info(f"守护进程退出，共执行 {round_count} 轮")


if __name__ == "__main__":
    main()

python monitor_daemon.py --dry-run

2026-08-09 09:59:19 [INFO] 盯盘守护进程启动 interval=5s dry_run=True
2026-08-09 09:59:19 [INFO] 证书已加载
2026-08-09 09:59:19 [INFO] ── 轮次开始 09:59:19 ──
2026-08-09 09:59:19 [INFO] [NOTE] 600519 成交额=32.67亿 超过10亿
2026-08-09 09:59:19 [INFO] [NOTE] 300750 成交额=98.78亿 超过10亿
2026-08-09 09:59:19 [INFO] [NOTE] 000858 成交额=26.92亿 超过10亿
2026-08-09 09:59:19 [INFO]   本轮完成：5 只，0 条告警
...（3 轮后退出）

python monitor_daemon.py --dry-run --stdout

[Unit]
Description=Stock Market Monitor Daemon (麦蕊实时盯盘)
After=network-online.target
Wants=network-online.target

[Service]
Type=simple
User=nobody
Group=nogroup
WorkingDirectory=/opt/stock-monitor
ExecStart=/usr/bin/python3 /opt/stock-monitor/monitor_daemon.py --stdout --interval 60
Restart=always
RestartSec=30
StandardOutput=journal
StandardError=journal
SyslogIdentifier=stock-monitor

# 安全加固
NoNewPrivileges=yes
PrivateTmp=yes
ProtectSystem=strict
ProtectHome=yes
ReadWritePaths=/opt/stock-monitor/logs
ReadOnlyPaths=/opt/stock-monitor

# 证书通过环境变量注入（绝不写死在 unit 文件里！）
Environment=MAIRUI_LICENCE=<你的证书>  # 演示证书，请替换成你自己的正式证书
Environment=PYTHONUNBUFFERED=1

[Install]
WantedBy=multi-user.target

# 1. 复制文件到部署目录
sudo mkdir -p /opt/stock-monitor/logs
sudo cp monitor_daemon.py /opt/stock-monitor/
sudo cp cert_loader.py /opt/stock-monitor/  # 证书加载器（环境变量读取）
sudo chown -R nobody:nogroup /opt/stock-monitor/logs

# 2. 安装 systemd unit
sudo cp stock-monitor.service /etc/systemd/system/
sudo systemctl daemon-reload

# 3. 开机自启 + 立即启动
sudo systemctl enable --now stock-monitor

# 4. 查看运行状态
sudo systemctl status stock-monitor

# 5. 实时查看日志
sudo journalctl -u stock-monitor -f

# 重启服务（更新脚本后）
sudo systemctl restart stock-monitor

# 查看最近 50 条日志
sudo journalctl -u stock-monitor -n 50 --no-pager

# 查看今天的日志
sudo journalctl -u stock-monitor --since today --no-pager

# 停止服务
sudo systemctl stop stock-monitor

# /etc/logrotate.d/stock-monitor
/opt/stock-monitor/logs/monitor.log {
    daily                    # 每天轮转
    rotate 30                # 保留最近 30 天
    missingok                # 日志不存在不报错
    notifempty               # 空日志不轮转
    compress                 # 压缩旧日志 (.gz)
    delaycompress            # 延迟一天再压缩（保证可读）
    copytruncate             # 复制后截断（不中断进程）
    dateext                  # 使用日期后缀
    dateformat -%Y%m%d
    maxsize 50M              # 超过 50M 强制轮转
    create 0644 nobody nogroup
}

sudo cp logrotate-stock-monitor.conf /etc/logrotate.d/stock-monitor
# 手动触发一次验证
sudo logrotate -f /etc/logrotate.d/stock-monitor

Environment=MAIRUI_LICENCE=<你的证书>  # 演示证书，请替换成你自己的正式证书