47 lines
2.5 KiB
Bash
Executable File
47 lines
2.5 KiB
Bash
Executable File
#!/bin/bash
|
||
# v2 数据下载包装（SSH 模式：rsync NAS↔本地 + 跑 v1 daily_all_update + raw 增量）
|
||
# 不依赖 SMB 挂载，用 sanguo-nas SSH key 免密。
|
||
# 用法：./run_daily_update.sh [--skip-daily|--skip-15min]
|
||
# env:
|
||
# STOCK_LIMIT=N 限制股票数（验证用，默认 0=全市场）
|
||
# SKIP_PULL=1 跳过 rsync 拉（验证用；生产必须拉，v1 才能增量）
|
||
# SKIP_RAW=1 跳过 raw 增量（默认跑，C-S3 实走需当日 raw）
|
||
set -e
|
||
NAS=sanguo-nas
|
||
LOCAL=${STOCK_MOUNT:-/tmp/stock_dl}
|
||
export STOCK_MOUNT=$LOCAL
|
||
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||
|
||
mkdir -p "$LOCAL/logs/daily_update/progress" \
|
||
"$LOCAL/A股数据/日线数据/daily" \
|
||
"$LOCAL/A股数据/日线数据/raw" \
|
||
"$LOCAL/minute_kline/15min" \
|
||
"$LOCAL/A股数据/stock_info" \
|
||
"$LOCAL/sanguo_vnpy/data"
|
||
|
||
if [ "${SKIP_PULL:-0}" != "1" ]; then
|
||
echo "=== 1. rsync 拉 NAS 现有 → 本地（增量，首次慢后续快）==="
|
||
rsync -az "$NAS:/volume1/stock/A股数据/stock_info/" "$LOCAL/A股数据/stock_info/" || echo " stock_info pull warn"
|
||
rsync -az "$NAS:/volume1/stock/A股数据/日线数据/daily/" "$LOCAL/A股数据/日线数据/daily/" || echo " daily pull warn"
|
||
rsync -az "$NAS:/volume1/stock/A股数据/日线数据/raw/" "$LOCAL/A股数据/日线数据/raw/" || echo " raw pull warn"
|
||
rsync -az "$NAS:/volume1/stock/minute_kline/15min/" "$LOCAL/minute_kline/15min/" || echo " 15min pull warn"
|
||
rsync -az "$NAS:/volume1/stock/sanguo_vnpy/data/" "$LOCAL/sanguo_vnpy/data/" || echo " vnpy data pull warn"
|
||
fi
|
||
|
||
echo "=== 2. 跑 v1 增量更新（STOCK_MOUNT=$LOCAL STOCK_LIMIT=${STOCK_LIMIT:-0}）==="
|
||
cd "$SCRIPT_DIR"
|
||
python3 daily_all_update.py "$@"
|
||
|
||
if [ "${SKIP_RAW:-0}" != "1" ]; then
|
||
echo "=== 2.5 raw 增量（最近 5 天覆盖 raw_dir，C-S3 实走需当日 raw）==="
|
||
RAW_START=$(python3 -c "import datetime;print((datetime.date.today()-datetime.timedelta(days=5)).isoformat())")
|
||
SLEEP=0.5 python3 raw_redownload.py --all --start "$RAW_START" || echo " raw 增量 warn"
|
||
fi
|
||
|
||
echo "=== 3. rsync 推 本地 → NAS ==="
|
||
rsync -az "$LOCAL/A股数据/日线数据/daily/" "$NAS:/volume1/stock/A股数据/日线数据/daily/"
|
||
rsync -az "$LOCAL/A股数据/日线数据/raw/" "$NAS:/volume1/stock/A股数据/日线数据/raw/"
|
||
rsync -az "$LOCAL/minute_kline/15min/" "$NAS:/volume1/stock/minute_kline/15min/"
|
||
rsync -az "$LOCAL/sanguo_vnpy/data/" "$NAS:/volume1/stock/sanguo_vnpy/data/"
|
||
echo "=== 完成 $(date) ==="
|