#FastAPI 接口编写（main.py）

from fastapi import FastAPI, Depends
from sqlalchemy import select
from sqlalchemy.orm import Session
from database import SessionLocal, engine, Base
from models import User

# 启动时自动建表（开发阶段方便，生产环境用 Alembic 迁移）
Base.metadata.create_all(bind=engine)

app = FastAPI()


# ---------- 依赖注入：每个请求分配一个数据库会话 ----------
def get_db():
    db = SessionLocal()       # 打开会话
    try:
        yield db              # 把会话交给路由函数使用
    finally:
        db.close()            # 请求结束后关闭会话


# ---------- 创建用户 ----------
@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)   # 刷新，拿到数据库生成的 id
    return user


# ---------- 查询所有用户 ----------
@app.get("/users/")
def list_users(db: Session = Depends(get_db)):
    return db.execute(select(User)).scalars().all()


# ---------- 查询单个用户 ----------
@app.get("/users/{user_id}")
def get_user(user_id: int, db: Session = Depends(get_db)):
    return db.execute(select(User).where(User.id == user_id)).scalars().first()

Base.metadata.create_all(bind=engine)

def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()

def list_users(db: Session = Depends(get_db)):
    #                     ↑ FastAPI 自动调用 get_db()，把 db 注入进来

from sqlalchemy import select

# 新增
user = User(name="张三", email="zhangsan@example.com")
db.add(user)        # 加入会话（还没写入数据库）
db.commit()         # 提交事务（真正写入）
db.refresh(user)    # 刷新，拿到数据库自动生成的 id

# 查询
db.execute(select(User)).scalars().all()                              # 查全部
db.execute(select(User).where(User.id == 1)).scalars().first()       # 按条件查一条
db.execute(select(User).where(User.name.contains("张"))).scalars().all()  # 模糊查询

# 修改
user.name = "李四"
db.commit()         # 直接 commit，SQLAlchemy 自动检测变更并生成 UPDATE

# 删除
db.delete(user)
db.commit()