Files
opsxsql/backend/app/api/v1/endpoints/auth.py
T
loganandClaude Opus 4.7 439155c92e Initial commit: SQL audit platform v0.1.0
FastAPI backend with auth, user, instance, ticket, query, dashboard APIs.
React + TypeScript frontend with mock data UI (10 pages).
SQLite + async SQLAlchemy, JWT auth, role-based access control.

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-05-22 11:11:57 +08:00

56 lines
1.7 KiB
Python

from datetime import datetime, timezone
from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession
from app.core.database import get_db
from app.core.security import verify_password, create_access_token, get_current_user
from app.models.user import User
from app.schemas.auth import LoginRequest, TokenResponse, UserBrief
router = APIRouter(prefix="/auth", tags=["认证"])
@router.post("/login", response_model=TokenResponse)
async def login(req: LoginRequest, db: AsyncSession = Depends(get_db)):
result = await db.execute(select(User).where(User.username == req.username))
user = result.scalar_one_or_none()
if not user or not verify_password(req.password, user.hashed_password):
raise HTTPException(status_code=401, detail="用户名或密码错误")
if not user.is_active:
raise HTTPException(status_code=403, detail="用户已禁用")
user.last_login = datetime.now(timezone.utc)
await db.commit()
token = create_access_token(data={"sub": str(user.id), "role": user.role})
return TokenResponse(
access_token=token,
user=UserBrief(
id=user.id,
username=user.username,
display_name=user.display_name,
role=user.role,
email=user.email,
),
)
@router.get("/me", response_model=UserBrief)
async def get_me(current_user: User = Depends(get_current_user)):
return UserBrief(
id=current_user.id,
username=current_user.username,
display_name=current_user.display_name,
role=current_user.role,
email=current_user.email,
)
@router.post("/logout")
async def logout():
return {"message": "已登出"}