使用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接口,并根据需要进行扩展和定制化。
