build: 添加 Flask 服务构建脚本

- 新增 build-flask-and-push.sh：一键构建和推送 Flask API 镜像 - 支持自定义镜像名称和标签 - 自动检查基础镜像、构建、打标签、推送 - 提供运行示例和测试命令
feat(docker): 添加 Flask API 服务容器化配置
2026-05-07 22:15:17 +08:00 · 2026-05-07 22:15:10 +08:00
4 changed files with 240 additions and 0 deletions
--- a/Dockerfile.flask
+++ b/Dockerfile.flask
@@ -0,0 +1,38 @@
+# Flask API 服务 Dockerfile
+# =========================
+# 用于构建 Universal Data Fetcher API 服务的 Docker 镜像
+
+FROM index-base:latest
+
+# 设置工作目录
+WORKDIR /app
+
+# 复制依赖文件
+COPY requirements.txt .
+
+# 安装依赖
+RUN uv pip install --system -r requirements.txt
+
+# 仅复制除 data 目录外的应用代码
+COPY . .
+
+# 创建日志目录
+RUN mkdir -p /app/logs
+
+# 设置时区为上海
+ENV TZ=Asia/Shanghai
+
+# 暴露 Flask 服务端口
+EXPOSE 5000
+
+# 设置环境变量默认值
+ENV FLASK_APP=core/datasource/flask_server.py
+ENV FLASK_ENV=production
+ENV PYTHONUNBUFFERED=1
+
+# 健康检查
+HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
+    CMD python -c "import requests; requests.get('http://localhost:5000/health')" || exit 1
+
+# 启动 Flask 服务
+CMD ["python", "core/datasource/flask_server.py", "--host", "0.0.0.0", "--port", "5000"]
--- a/build-flask-and-push.sh
+++ b/build-flask-and-push.sh
@@ -0,0 +1,118 @@
+#!/bin/bash
+# Flask API 服务构建和推送脚本
+# =============================
+
+set -e  # 遇到错误立即退出
+
+# 配置变量
+IMAGE_NAME="${1:-etf-data-fetcher}"
+REGISTRY="192.168.0.115:5000"
+TAG="${2:-latest}"
+FULL_IMAGE_NAME="${REGISTRY}/${IMAGE_NAME}:${TAG}"
+DOCKERFILE="Dockerfile.flask"
+
+echo "========================================="
+echo "Flask API 服务 - 构建和推送脚本"
+echo "========================================="
+echo "镜像名称: ${IMAGE_NAME}"
+echo "标签: ${TAG}"
+echo "Dockerfile: ${DOCKERFILE}"
+echo ""
+
+# 检查 Dockerfile 是否存在
+if [ ! -f "${DOCKERFILE}" ]; then
+    echo "❌ Dockerfile ${DOCKERFILE} 不存在"
+    exit 1
+fi
+
+# 检查并构建基础镜像（如果不存在）
+echo "0. 检查基础镜像..."
+if ! docker images --format "{{.Repository}}:{{.Tag}}" | grep -q "index-base:latest"; then
+    echo "   基础镜像不存在，开始构建..."
+    if [ -f "Dockerfile_base" ]; then
+        docker build --platform linux/arm64 -f Dockerfile_base -t index-base:latest .
+        echo "   ✅ 基础镜像构建成功"
+    else
+        echo "   ⚠️ 未找到 Dockerfile_base，跳过基础镜像构建"
+        exit 1
+    fi
+else
+    echo "   ✅ 基础镜像已存在"
+fi
+
+# 构建镜像
+echo ""
+echo "1. 构建 Flask API 镜像..."
+echo "   镜像名称: ${IMAGE_NAME}"
+echo "   平台: linux/arm64"
+docker build --platform linux/arm64 -f ${DOCKERFILE} -t ${IMAGE_NAME} .
+
+if [ $? -eq 0 ]; then
+    echo "   ✅ 镜像构建成功"
+else
+    echo "   ❌ 镜像构建失败"
+    exit 1
+fi
+
+# 打标签
+echo ""
+echo "2. 为镜像打标签..."
+echo "   标签: ${FULL_IMAGE_NAME}"
+docker tag ${IMAGE_NAME} ${FULL_IMAGE_NAME}
+
+if [ $? -eq 0 ]; then
+    echo "   ✅ 标签添加成功"
+else
+    echo "   ❌ 标签添加失败"
+    exit 1
+fi
+
+# 推送到私有仓库
+echo ""
+echo "3. 推送镜像到私有仓库..."
+echo "   仓库地址: ${REGISTRY}"
+docker push ${FULL_IMAGE_NAME}
+
+if [ $? -eq 0 ]; then
+    echo "   ✅ 镜像推送成功"
+else
+    echo "   ❌ 镜像推送失败"
+    exit 1
+fi
+
+# 清理本地镜像（可选）
+echo ""
+echo "4. 清理临时镜像..."
+docker rmi ${IMAGE_NAME} || true
+
+# 显示最终结果
+echo ""
+echo "========================================="
+echo "🎉 构建和推送完成！"
+echo "========================================="
+echo "镜像地址: ${FULL_IMAGE_NAME}"
+echo ""
+echo "可以使用以下命令运行容器："
+echo ""
+echo "# 基础运行（仅A股数据）"
+echo "docker run -d --name ${IMAGE_NAME}-container \\"
+echo "  -p 5000:5000 \\"
+echo "  -v \$(pwd)/.env:/app/.env:ro \\"
+echo "  ${FULL_IMAGE_NAME}"
+echo ""
+echo "# 启用SSH隧道（支持港美股）"
+echo "docker run -d --name ${IMAGE_NAME}-container \\"
+echo "  -p 5000:5000 \\"
+echo "  -v \$(pwd)/.env:/app/.env:ro \\"
+echo "  -v \$(pwd)/hk_ecs.pem:/app/hk_ecs.pem:ro \\"
+echo "  -e SSH_ENABLED=true \\"
+echo "  -e SSH_HOST=8.218.167.69 \\"
+echo "  -e SSH_USERNAME=root \\"
+echo "  -e SSH_KEY_PATH=hk_ecs.pem \\"
+echo "  ${FULL_IMAGE_NAME}"
+echo ""
+echo "# 测试API"
+echo "curl http://localhost:5000/health"
+echo "curl 'http://localhost:5000/api/v1/ohlcv?code=000300.SH&start=2024-01-01&end=2024-03-31'"
+echo ""
+echo "========================================="
--- a/docker-compose.flask.yml
+++ b/docker-compose.flask.yml
@@ -0,0 +1,80 @@
+# Flask API 服务 Docker Compose 配置
+# ===================================
+
+version: '3.8'
+
+services:
+  etf-data-fetcher:
+    build:
+      context: .
+      dockerfile: Dockerfile.flask
+    image: 192.168.0.115:5000/etf-data-fetcher:latest
+    container_name: etf-data-fetcher
+    
+    ports:
+      - "5000:5000"
+    
+    environment:
+      # Flask 配置
+      - FLASK_ENV=production
+      - PYTHONUNBUFFERED=1
+      
+      # SSH 隧道配置（可选，启用后可获取港美股数据）
+      - SSH_ENABLED=${SSH_ENABLED:-false}
+      - SSH_HOST=${SSH_HOST:-8.218.167.69}
+      - SSH_PORT=${SSH_PORT:-22}
+      - SSH_USERNAME=${SSH_USERNAME:-root}
+      - SSH_KEY_PATH=${SSH_KEY_PATH:-hk_ecs.pem}
+      - SSH_LOCAL_PORT=${SSH_LOCAL_PORT:-1080}
+    
+    volumes:
+      # 环境变量配置（必需：TUSHARE_TOKEN）
+      - ./.env:/app/.env:ro
+      
+      # SSH 私钥（可选，获取港美股数据需要）
+      - ./hk_ecs.pem:/app/hk_ecs.pem:ro
+      
+      # 数据目录（可选，用于持久化缓存）
+      - ./data:/app/data
+      
+      # 日志目录
+      - ./logs:/app/logs
+    
+    # 健康检查
+    healthcheck:
+      test: ["CMD", "python", "-c", "import requests; requests.get('http://localhost:5000/health')"]
+      interval: 30s
+      timeout: 10s
+      retries: 3
+      start_period: 10s
+    
+    # 重启策略
+    restart: unless-stopped
+    
+    # 资源限制
+    deploy:
+      resources:
+        limits:
+          cpus: '2.0'
+          memory: 2G
+        reservations:
+          cpus: '0.5'
+          memory: 512M
+
+  # 可选：Nginx 反向代理
+  nginx:
+    image: nginx:alpine
+    container_name: etf-data-fetcher-nginx
+    ports:
+      - "80:80"
+    volumes:
+      - ./nginx.conf:/etc/nginx/nginx.conf:ro
+    depends_on:
+      - etf-data-fetcher
+    restart: unless-stopped
+    profiles:
+      - with-nginx
+
+networks:
+  default:
+    name: etf-network
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,6 +15,10 @@ requests[socks]>=2.28.0
 urllib3>=1.26.0
 pysocks>=1.7.0

+# ==================== Web服务 ====================
+flask>=3.0.0
+flask-cors>=4.0.0
+
 # ==================== 进度条 ====================
 tqdm>=4.65.0