FastAPI中怎么使用ORM

在FastAPI中使用ORM，一般情况下会使用SQLAlchemy作为ORM框架。以下是一个简单的示例：

1. 首先，安装SQLAlchemy和相应数据库驱动：

pip install sqlalchemy pip install databases pip install psycopg2 

2. 创建一个数据库连接：

from databases import Database from sqlalchemy import create_engine  DATABASE_URL = "sqlite:///./test.db" database = Database(DATABASE_URL)  # 创建数据库引擎 engine = create_engine(DATABASE_URL) 

3. 创建一个基础的模型类：

from sqlalchemy import Column, Integer, String from sqlalchemy.ext.declarative import declarative_base  Base = declarative_base()  class User(Base):     __tablename__ = "users"          id = Column(Integer, primary_key=True, index=True)     name = Column(String)     email = Column(String, unique=True) 

4. 创建数据库表和连接数据库：

from sqlalchemy.orm import sessionmaker  # 创建数据库表 Base.metadata.create_all(engine)  # 创建Session SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)  # 创建一个函数来获取数据库会话 def get_db():     db = SessionLocal()     try:         yield db     finally:         db.close() 

5. 在FastAPI中使用ORM：

from fastapi import FastAPI, Depends from sqlalchemy.orm import Session from models import User from database import get_db  app = FastAPI()  # 创建用户 @app.post("/users/") def create_user(name: str, email: str, db: Session = Depends(get_db)):     user = User(name=name, email=email)     db.add(user)     db.commit()     db.refresh(user)     return user  # 获取用户 @app.get("/users/{user_id}") def get_user(user_id: int, db: Session = Depends(get_db)):     return db.query(User).filter(User.id == user_id).first() 

以上是一个简单的FastAPI与SQLAlchemy ORM结合使用的示例。你可以根据自己的需求进一步扩展和优化。