#!/bin/bash
# A股超短期选股系统 - 每日盘前规划/盘中确认脚本
# 用法: bash scripts/daily_screen.sh pre|confirm
# 建议cron:
#   20 9 * * 1-5  cd /root/nginx/gupiao && bash scripts/daily_screen.sh pre      >> logs/screen_pre.log 2>&1
#   0 10 * * 1-5  cd /root/nginx/gupiao && bash scripts/daily_screen.sh confirm   >> logs/screen_confirm.log 2>&1

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_DIR="$(dirname "$SCRIPT_DIR")"
LOG_DIR="$PROJECT_DIR/logs"
mkdir -p "$LOG_DIR"

MODE="${1:-auto}"
case "$MODE" in
  pre)     MODE_ARG="--mode pre";;
  confirm) MODE_ARG="--mode confirm";;
  auto)    MODE_ARG="";;
  *)  echo "用法: $0 [pre|confirm|auto]"; exit 1;;
esac

echo "========================================"
echo "  选股开始 ($MODE): $(date '+%Y-%m-%d %H:%M:%S')"
echo "========================================"

cd "$PROJECT_DIR"

python3 cli.py screen --top 10 $MODE_ARG

echo ""
echo "选股完成: $(date '+%Y-%m-%d %H:%M:%S')"
echo "========================================"