使用FastAPI和sqlalchemy操作数据库

本示例代码展示了如何使用FastAPI和sqlalchemy实现基本的数据库操作，包括数据的增、删、改、查等操作。

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from fastapi import FastAPI

app = FastAPI()

SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db"
engine = create_engine(SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False})
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
Base = declarative_base()

class User(Base):
    __tablename__ = "users"

    id = Column(Integer, primary_key=True, index=True)
    name = Column(String, index=True)
    age = Column(Integer)

Base.metadata.create_all(bind=engine)

@app.post("/users/") # 新增用户
async def create_user(name: str, age: int):
    db = SessionLocal()
    user = User(name=name, age=age)
    db.add(user)
    db.commit()
    db.refresh(user)
    return user

@app.get("/users/{user_id}/") # 获取指定id的用户
async def read_user(user_id: int):
    db = SessionLocal()
    user = db.query(User).filter(User.id == user_id).first()
    return user

@app.put("/users/{user_id}/") # 更新指定id的用户信息
async def update_user(user_id: int, name: str, age: int):
    db = SessionLocal()
    user = db.query(User).filter(User.id == user_id).first()
    user.name = name
    user.age = age
    db.commit()
    db.refresh(user)
    return user

@app.delete("/users/{user_id}/") # 删除指定id的用户
async def delete_user(user_id: int):
    db = SessionLocal()
    user = db.query(User).filter(User.id == user_id).first()
    db.delete(user)
    db.commit()
    return {"msg": "deleted"}

免责申明：文章和图片全部来源于公开网络，如有侵权，请通知删除 server@dude6.com