Python自动化炒股：使用FastAPI和Docker Compose部署股票数据服务的实战案例

引言

在当今的金融市场中，自动化交易系统已经成为许多投资者和交易员的重要工具。Python以其强大的库和灵活性，成为了实现自动化炒股的首选语言。本文将带你了解如何使用FastAPI和Docker Compose来部署一个股票数据服务，为自动化炒股提供实时数据支持。

为什么选择FastAPI和Docker Compose？

• FastAPI：这是一个现代、快速（高性能）的Web框架，用于构建APIs，使用Python 3.6+基于标准Python类型提示。它非常适合构建轻量级的RESTful API。
• Docker Compose：这是一个用于定义和运行多容器Docker应用程序的工具。使用Docker Compose，我们可以通过一个YAML文件来配置应用的服务，然后使用一个命令来启动和停止所有服务。

环境准备

在开始之前，请确保你的开发环境中安装了以下软件：

• Python 3.8+
• Docker
• Docker Compose

步骤1：创建FastAPI应用

首先，我们需要创建一个FastAPI应用来提供股票数据服务。

1.1 安装FastAPI和Uvicorn

pip install fastapi uvicorn

1.2 创建FastAPI应用

创建一个名为mAIn.py的文件，并添加以下代码：

from fastapi import FastAPI
from typing import List

app = FastAPI()

@app.get("/stock/{symbol}")
async def read_stock(symbol: str):
    # 这里模拟从数据库或API获取股票数据
    stock_data = {"symbol": symbol, "price": 100, "volume": 1000}
    return stock_data

步骤2：创建Dockerfile

为了将我们的FastAPI应用容器化，我们需要创建一个Dockerfile。

2.1 创建Dockerfile

在项目根目录下创建一个名为Dockerfile的文件，并添加以下内容：

# 使用官方Python镜像
FROM python:3.8-slim

# 设置工作目录
WORKDIR /app

# 复制项目文件到工作目录
COPY . .

# 安装依赖
RUN pip install -r requirements.txt

# 运行Uvicorn服务器
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]

2.2 创建requirements.txt

在同一目录下创建一个名为requirements.txt的文件，并列出所有依赖：

fastapi
uvicorn

步骤3：使用Docker Compose部署

3.1 创建docker-compose.yml

在项目根目录下创建一个名为docker-compose.yml的文件，并添加以下内容：

version: '3.8'
services:
  web:
    build: .
    ports:
      - "8000:8000"
    volumes:
      - .:/app
    command: uvicorn main:app --host 0.0.0.0 --port 8000

3.2 启动服务

在终端中运行以下命令来启动服务：

docker-compose up --build

这将构建Docker镜像，并启动FastAPI服务。

步骤4：测试API

现在，你可以通过访问http://localhost:8000/stock/AAPL来测试你的API。

步骤5：扩展服务

为了使服务更加健壮，我们可以添加数据库支持和缓存机制。

5.1 添加数据库服务

修改docker-compose.yml文件，添加数据库服务：

version: '3.8'
services:
  web:
    build: .
    ports:
      - "8000:8000"
    volumes:
      - .:/app
    command: uvicorn main:app --host 0.0.0.0 --port 8000
    depends_on:
      - db
  db:
    image: postgres
    environment:
      POSTGRES_DB: stockdb
      POSTGRES_USER: user
      POSTGRES_PASSWORD: password
    volumes:
      - db-data:/var/lib/postgresql/data
volumes:
  db-data:

5.2 修改FastAPI应用以使用数据库

更新main.py以连接数据库并查询股票数据：

from fastapi import FastAPI
from typing import List
import psycopg2

app = FastAPI()

@app.get("/stock/{symbol}")
async def read_stock(symbol: str):
    conn = psycopg2.connect(
        dbname="