欢迎访问宙启技术站
智能推送

使用Python创建和管理API接口

发布时间:2023-12-16 06:06:43

在Python中,可以使用多种库和框架来创建和管理API接口。以下是使用Flask和FastAPI库的示例代码,演示了如何创建和管理API接口。

使用Flask库创建和管理API接口的代码示例:

from flask import Flask, jsonify, request

app = Flask(__name__)

@app.route('/api/user', methods=['GET'])
def get_users():
    users = [
        {'id': 1, 'name': 'John'},
        {'id': 2, 'name': 'Alice'},
        {'id': 3, 'name': 'Bob'}
    ]
    return jsonify(users)

@app.route('/api/user', methods=['POST'])
def add_user():
    user = request.get_json()
    # Add user to the database
    return jsonify({'message': 'User added successfully'})

@app.route('/api/user/<int:user_id>', methods=['GET'])
def get_user(user_id):
    user = {'id': user_id, 'name': 'John'}
    # Fetch user from the database
    return jsonify(user)

if __name__ == '__main__':
    app.run()

在上面的示例中,使用Flask库创建了3个不同的API端点。get_users函数处理GET请求以获取用户列表。add_user函数处理POST请求以添加新用户。get_user函数处理带有用户ID参数的GET请求以获取特定用户。

使用FastAPI库创建和管理API接口的代码示例:

from fastapi import FastAPI
from pydantic import BaseModel

app = FastAPI()

class User(BaseModel):
    id: int
    name: str

users = [
    User(id=1, name='John'),
    User(id=2, name='Alice'),
    User(id=3, name='Bob')
]

@app.get('/api/user')
def get_users():
    return users

@app.post('/api/user')
def add_user(user: User):
    # Add user to the database
    return {'message': 'User added successfully'}

@app.get('/api/user/{user_id}')
def get_user(user_id: int):
    for user in users:
        if user.id == user_id:
            return user
    return {'message': 'User not found'}

if __name__ == '__main__':
    import uvicorn
    uvicorn.run(app)

上面的示例中,使用FastAPI库创建了3个不同的API端点。get_users函数处理GET请求以获取用户列表。add_user函数处理POST请求以添加新用户。get_user函数处理带有用户ID参数的GET请求以获取特定用户。

无论使用Flask还是FastAPI,都可以通过运行Python脚本来启动API服务器。Flask的服务器使用app.run()方法启动,而FastAPI的服务器使用uvicorn.run(app)方法启动。

以上是使用Python创建和管理API接口的示例代码。这些示例代码可以帮助您开始使用Python编写自己的API接口,并根据需要进行扩展和定制化。