Flask搭建一个API服务器的步骤

下面我将详细讲解使用Flask框架搭建API服务器的步骤。

简介

Flask 是一个“微型”Python Web框架。它非常简单易学且具有高度的灵活性。本文将介绍如何使用 Flask 框架搭建一个API服务器。

步骤

1. 安装 Flask

在开始之前，请确保您的电脑上已经安装了 Python 3.x，并且安装了 Flask 框架。您可以使用以下命令安装 Flask：

pip install flask

2. 搭建基本的API服务器

创建 Python 文件

首先，我们创建一个 Python 文件，命名为 app.py。在这个 Python 文件中，我们将编写一些代码来启动我们的 Flask 应用程序，并创建一个简单的 API 端点。您可以复制下面的代码并将其粘贴到 app.py 文件中：

from flask import Flask

app = Flask(__name__)

@app.route("/")
def home():
    return "Hello, World!"

if __name__ == "__main__":
    app.run(debug=True)

运行这个文件：

python app.py

当 Flask 运行时，您会看到以下输出：

* Running on http://127.0.0.1:5000/ (Press CTRL+C to quit)

这意味着您的 Flask 应用程序正在运行，并正在监听本机的 5000 端口。

测试

现在，打开您的 Web 浏览器，并输入以下网址：

http://127.0.0.1:5000/

您将看到您的应用程序返回一个简单的消息，"Hello, World!"。

3. 创建 API 端点

现在我们已经成功搭建了一个非常基本的 API 服务器，让我们来创建一个更具实用性的 API 端点。

示例一

假设我们要创建一个非常简单的 API，用于检索 GitHub 上的用户信息。我们将使用 requests 库来获取用户信息，并使用 Flask 来创建 API 端点。

以下是代码示例：

import requests
from flask import Flask, jsonify

app = Flask(__name__)

@app.route('/users/<username>', methods=['GET'])
def get_user(username):
    url = f'https://api.github.com/users/{username}'
    response = requests.get(url)

    if response.status_code == 200:
        return jsonify(response.json())
    else:
        return jsonify({'error': 'User not found'}), 404

if __name__ == "__main__":
    app.run(debug=True)

代码说明：

/users/<username> 表示这是一个动态路由，需要传递一个用户名作为参数。
url = f'https://api.github.com/users/{username}' 构造 GitHub API 的 URL。
response = requests.get(url) 使用 requests 库发送 GET 请求，并获取响应。
if response.status_code == 200: 判断请求是否成功。
return jsonify(response.json()) 如果成功，返回 GitHub API 返回的 JSON 数据。
return jsonify({'error': 'User not found'}), 404 如果失败，返回一个 JSON 对象 (包含 error 属性)，并设置状态码为 404。

示例二

下面我们再看一下如何创建一个更复杂的 API 端点，比如一个可以添加、更新和删除用户信息的 API。

以下是代码示例：

import json
from flask import Flask, jsonify, request

app = Flask(__name__)

# 用于存储用户信息的列表
users = []

# 用于生成一个新的用户 ID
def generate_user_id():
    return len(users) + 1

# 根据 ID 获取用户信息
def get_user_by_id(user_id):
    for user in users:
        if user['id'] == user_id:
            return user
    return None

# 根据 ID 删除用户信息
def delete_user_by_id(user_id):
    user = get_user_by_id(user_id)
    if user:
        users.remove(user)
        return user
    else:
        return None

@app.route('/users', methods=['GET'])
def get_all_users():
    return jsonify(users)

@app.route('/users', methods=['POST'])
def add_new_user():
    data = request.json
    user_id = generate_user_id()
    user = {
        'id': user_id,
        'name': data['name'],
        'email': data['email']
    }
    users.append(user)
    return jsonify(user)

@app.route('/users/<int:user_id>', methods=['GET'])
def get_user(user_id):
    user = get_user_by_id(user_id)
    if user:
        return jsonify(user)
    else:
        return jsonify({'error': 'User not found'}), 404

@app.route('/users/<int:user_id>', methods=['PUT'])
def update_user(user_id):
    data = request.json
    user = get_user_by_id(user_id)
    if user:
        user.update(data)
        return jsonify(user)
    else:
        return jsonify({'error': 'User not found'}), 404

@app.route('/users/<int:user_id>', methods=['DELETE'])
def delete_user(user_id):
    user = delete_user_by_id(user_id)
    if user:
        return jsonify(user)
    else:
        return jsonify({'error': 'User not found'}), 404

if __name__ == "__main__":
    app.run(debug=True)

代码说明：

users = [] 创建一个空的用户信息列表。
generate_user_id() 用于生成一个新的用户 ID。
get_user_by_id(user_id) 根据 ID 获取用户信息。
delete_user_by_id(user_id) 根据 ID 删除用户信息。
@app.route('/users', methods=['GET']) 创建一个 GET 请求的路由，并返回用户信息列表。
@app.route('/users', methods=['POST']) 创建一个 POST 请求的路由，并添加一个新的用户信息。
@app.route('/users/<int:user_id>', methods=['GET']) 创建一个 GET 请求的路由，并根据 ID 获取用户信息。
@app.route('/users/<int:user_id>', methods=['PUT']) 创建一个 PUT 请求的路由，并根据 ID 更新用户信息。
@app.route('/users/<int:user_id>', methods=['DELETE']) 创建一个 DELETE 请求的路由，并根据 ID 删除用户信息。