当前位置：首页 > news >正文

如何编写一个API——Python代码示例及拓展

news 2025/8/3 2:27:06

下面是一个必备的API的demo，包括用户认证、数据库交互、错误处理和更复杂的异步任务处理。使用Flask来创建一个RESTful API，涉及用户注册、登录、以及获取用户信息的功能。

示例1：编写API

安装依赖

首先，你需要安装以下库来支持示例的功能：

pip install flask flask-caching flask_sqlalchemy flask_jwt_extended aiohttp

完整的Python代码示例

from flask import Flask, jsonify, request, make_response
from flask_sqlalchemy import SQLAlchemy
from flask_jwt_extended import JWTManager, create_access_token, jwt_required, get_jwt_identity
from flask_caching import Cache
from werkzeug.security import generate_password_hash, check_password_hash
import asyncio
import aiohttpapp = Flask(__name__)# 配置数据库
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///site.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)# 配置JWT
app.config['JWT_SECRET_KEY'] = 'super-secret-key'  # Change this!
jwt = JWTManager(app)# 配置缓存
app.config['CACHE_TYPE'] = 'simple'
cache = Cache(app)# 定义用户模型
class User(db.Model):id = db.Column(db.Integer, primary_key=True)username = db.Column(db.String(20), unique=True, nullable=False)password_hash = db.Column(db.String(120), nullable=False)def set_password(self, password):self.password_hash = generate_password_hash(password)def check_password(self, password):return check_password_hash(self.password_hash, password)# 创建数据库表
@app.before_first_request
def create_tables():db.create_all()# 用户注册
@app.route('/api/register', methods=['POST'])
def register():data = request.get_json()username = data.get(</