AI 辅助下的 Java 与 Python 类毕设选题实战：从选题策略到代码生成

定义数据模型 (Pydantic Schema)
我们可以在 schemas/user.py 文件中，先写下注释或函数名，让 AI 补全。

# schemas/user.py
from pydantic import BaseModel, EmailStr

class UserBase(BaseModel):
username: str
email: EmailStr

class UserCreate(UserBase):
password: str # 接收前端传来的明文密码

class UserInDB(UserBase):
id: int
hashed_password: str # 存入数据库的哈希密码
is_active: bool = True

class Config:
orm_mode = True # 使其能与 SQLAlchemy 模型兼容

# AI 可能会自动为你补全一个用于 API 响应的 Schema
# class UserResponse(UserBase):
# id: int
# is_active: bool

生成数据库模型 (SQLAlchemy Model)
在 models/user.py 中，我们开始写：

# models/user.py
from sqlalchemy import Boolean, Column, Integer, String
from .database import Base

class User(Base):
__tablename__ = "users"

id = Column(Integer, primary_key=True, index=True)
username = Column(String, unique=True, index=True, nullable=False)
email = Column(String, unique=True, index=True, nullable=False)
hashed_password = Column(String, nullable=False)
is_active = Column(Boolean, default=True)

编写核心认证工具函数
这是安全的关键。在 utils/security.py 中，我们可以描述需求，让 AI 生成安全代码。

# utils/security.py
from passlib.context import CryptContext
from datetime import datetime, timedelta
from jose import JWTError, jwt
from typing import Optional

# 定义密码哈希上下文和 JWT 密钥/算法
pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
SECRET_KEY = "your-secret-key-please-change-in-production" # 务必在生产环境更改！
ALGORITHM = "HS256"
ACCESS_TOKEN_EXPIRE_MINUTES = 30

def verify_password(plain_password, hashed_password):
"""验证明文密码与哈希密码是否匹配"""
return pwd_context.verify(plain_password, hashed_password)

def get_password_hash(password):
"""生成密码的哈希值"""
return pwd_context.hash(password)

def create_access_token(data: dict, expires_delta: Optional[timedelta] = None):
"""创建 JWT 访问令牌"""
to_encode = data.copy()
if expires_delta:
expire = datetime.utcnow() + expires_delta
else:
expire = datetime.utcnow() + timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES)
to_encode.update({"exp": expire})
encoded_jwt = jwt.encode(to_encode, SECRET_KEY, algorithm=ALGORITHM)
return encoded_jwt

AI 在这里帮助我们快速、准确地引入了 passlib 和 python-jose 库的正确用法，避免了手动实现哈希和 JWT 时容易出现的漏洞。

生成 API 端点 (FastAPI Route)
最后，在 routers/auth.py 中，我们可以让 AI 补全完整的登录注册逻辑。

# routers/auth.py
from fastapi import APIRouter, Depends, HTTPException, status
from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
from sqlalchemy.orm import Session
from .. import models, schemas, utils
from ..database import get_db

router = APIRouter(prefix="/auth", tags=["authentication"])
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="auth/login")

@router.post("/register", response_model=schemas.UserInDB)
async def register(user: schemas.UserCreate, db: Session = Depends(get_db)):
# 检查用户是否已存在
db_user_by_email = db.query(models.User).filter(models.User.email == user.email).first()
db_user_by_username = db.query(models.User).filter(models.User.username == user.username).first()
if db_user_by_email or db_user_by_username:
raise HTTPException(status_code=400, detail="Email or username already registered")
# 哈希密码并创建用户
hashed_password = utils.security.get_password_hash(user.password)
db_user = models.User(
username=user.username,
email=user.email,
hashed_password=hashed_password
)
db.add(db_user)
db.commit()
db.refresh(db_user)
return db_user

@router.post("/login")
async def login(form_data: OAuth2PasswordRequestForm = Depends(), db: Session = Depends(get_db)):
# 查找用户
user = db.query(models.User).filter(models.User.username == form_data.username).first()
if not user or not utils.security.verify_password(form_data.password, user.hashed_password):
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail="Incorrect username or password",
headers={"WWW-Authenticate": "Bearer"},
)
# 创建访问令牌
access_token = utils.security.create_access_token(data={"sub": user.username})
return {"access_token": access_token, "token_type": "bearer"}