use uv & make lint happy
This commit is contained in:
MingxuanGame
242
main.py
242
main.py
@@ -1,18 +1,32 @@
|
||||
from fastapi import FastAPI, Depends, HTTPException, Form
|
||||
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
|
||||
from fastapi.responses import JSONResponse
|
||||
from sqlalchemy.orm import Session
|
||||
from typing import Optional
|
||||
from datetime import datetime, timedelta
|
||||
from __future__ import annotations
|
||||
|
||||
from datetime import datetime, timedelta
|
||||
from typing import Optional
|
||||
|
||||
from app.auth import (
|
||||
authenticate_user,
|
||||
create_access_token,
|
||||
generate_refresh_token,
|
||||
get_token_by_access_token,
|
||||
get_token_by_refresh_token,
|
||||
store_token,
|
||||
)
|
||||
from app.config import settings
|
||||
from app.dependencies import get_db, get_redis, engine
|
||||
from app.models import TokenRequest, TokenResponse, User as ApiUser
|
||||
from app.database import Base, User as DBUser
|
||||
from app.auth import authenticate_user, create_access_token, generate_refresh_token, store_token
|
||||
from app.auth import get_token_by_access_token, get_token_by_refresh_token, verify_token
|
||||
from app.database import (
|
||||
User as DBUser,
|
||||
)
|
||||
from app.dependencies import get_db
|
||||
from app.models import (
|
||||
TokenResponse,
|
||||
User as ApiUser,
|
||||
)
|
||||
from app.utils import convert_db_user_to_api_user
|
||||
|
||||
from fastapi import Depends, FastAPI, Form, HTTPException
|
||||
from fastapi.responses import JSONResponse
|
||||
from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
# 注意: 表结构现在通过 migrations 管理，不再自动创建
|
||||
# 如需创建表，请运行: python quick_sync.py
|
||||
|
||||
@@ -30,110 +44,114 @@ async def oauth_token(
|
||||
username: Optional[str] = Form(None),
|
||||
password: Optional[str] = Form(None),
|
||||
refresh_token: Optional[str] = Form(None),
|
||||
db: Session = Depends(get_db)
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""OAuth 令牌端点"""
|
||||
# 验证客户端凭据
|
||||
if client_id != settings.OSU_CLIENT_ID or client_secret != settings.OSU_CLIENT_SECRET:
|
||||
if (
|
||||
client_id != settings.OSU_CLIENT_ID
|
||||
or client_secret != settings.OSU_CLIENT_SECRET
|
||||
):
|
||||
raise HTTPException(status_code=401, detail="Invalid client credentials")
|
||||
|
||||
|
||||
if grant_type == "password":
|
||||
# 密码授权流程
|
||||
if not username or not password:
|
||||
raise HTTPException(status_code=400, detail="Username and password required")
|
||||
|
||||
raise HTTPException(
|
||||
status_code=400, detail="Username and password required"
|
||||
)
|
||||
|
||||
# 验证用户
|
||||
user = authenticate_user(db, username, password)
|
||||
if not user:
|
||||
raise HTTPException(status_code=401, detail="Invalid username or password")
|
||||
|
||||
|
||||
# 生成令牌
|
||||
access_token_expires = timedelta(minutes=settings.ACCESS_TOKEN_EXPIRE_MINUTES)
|
||||
access_token = create_access_token(
|
||||
data={"sub": str(user.id)}, expires_delta=access_token_expires
|
||||
)
|
||||
refresh_token_str = generate_refresh_token()
|
||||
|
||||
|
||||
# 存储令牌
|
||||
store_token(
|
||||
db,
|
||||
user.id,
|
||||
access_token,
|
||||
refresh_token_str,
|
||||
settings.ACCESS_TOKEN_EXPIRE_MINUTES * 60
|
||||
db,
|
||||
user.id,
|
||||
access_token,
|
||||
refresh_token_str,
|
||||
settings.ACCESS_TOKEN_EXPIRE_MINUTES * 60,
|
||||
)
|
||||
|
||||
|
||||
return TokenResponse(
|
||||
access_token=access_token,
|
||||
token_type="Bearer",
|
||||
expires_in=settings.ACCESS_TOKEN_EXPIRE_MINUTES * 60,
|
||||
refresh_token=refresh_token_str,
|
||||
scope=scope
|
||||
scope=scope,
|
||||
)
|
||||
|
||||
|
||||
elif grant_type == "refresh_token":
|
||||
# 刷新令牌流程
|
||||
if not refresh_token:
|
||||
raise HTTPException(status_code=400, detail="Refresh token required")
|
||||
|
||||
|
||||
# 验证刷新令牌
|
||||
token_record = get_token_by_refresh_token(db, refresh_token)
|
||||
if not token_record:
|
||||
raise HTTPException(status_code=401, detail="Invalid refresh token")
|
||||
|
||||
|
||||
# 生成新的访问令牌
|
||||
access_token_expires = timedelta(minutes=settings.ACCESS_TOKEN_EXPIRE_MINUTES)
|
||||
access_token = create_access_token(
|
||||
data={"sub": str(token_record.user_id)}, expires_delta=access_token_expires
|
||||
)
|
||||
new_refresh_token = generate_refresh_token()
|
||||
|
||||
|
||||
# 更新令牌
|
||||
store_token(
|
||||
db,
|
||||
token_record.user_id,
|
||||
access_token,
|
||||
new_refresh_token,
|
||||
settings.ACCESS_TOKEN_EXPIRE_MINUTES * 60
|
||||
settings.ACCESS_TOKEN_EXPIRE_MINUTES * 60,
|
||||
)
|
||||
|
||||
|
||||
return TokenResponse(
|
||||
access_token=access_token,
|
||||
token_type="Bearer",
|
||||
expires_in=settings.ACCESS_TOKEN_EXPIRE_MINUTES * 60,
|
||||
refresh_token=new_refresh_token,
|
||||
scope=scope
|
||||
scope=scope,
|
||||
)
|
||||
|
||||
|
||||
else:
|
||||
raise HTTPException(status_code=400, detail="Unsupported grant type")
|
||||
|
||||
|
||||
async def get_current_user(
|
||||
credentials: HTTPAuthorizationCredentials = Depends(security),
|
||||
db: Session = Depends(get_db)
|
||||
db: Session = Depends(get_db),
|
||||
) -> DBUser:
|
||||
"""获取当前认证用户"""
|
||||
token = credentials.credentials
|
||||
|
||||
|
||||
# 验证令牌
|
||||
token_record = get_token_by_access_token(db, token)
|
||||
if not token_record:
|
||||
raise HTTPException(status_code=401, detail="Invalid or expired token")
|
||||
|
||||
|
||||
# 获取用户
|
||||
user = db.query(DBUser).filter(DBUser.id == token_record.user_id).first()
|
||||
if not user:
|
||||
raise HTTPException(status_code=404, detail="User not found")
|
||||
|
||||
|
||||
return user
|
||||
|
||||
|
||||
@app.get("/api/v2/me", response_model=ApiUser)
|
||||
@app.get("/api/v2/me/", response_model=ApiUser)
|
||||
async def get_user_info_default(
|
||||
current_user: DBUser = Depends(get_current_user),
|
||||
db: Session = Depends(get_db)
|
||||
current_user: DBUser = Depends(get_current_user), db: Session = Depends(get_db)
|
||||
):
|
||||
"""获取当前用户信息（默认使用osu模式）"""
|
||||
# 默认使用osu模式
|
||||
@@ -145,15 +163,17 @@ async def get_user_info_default(
|
||||
async def get_user_info(
|
||||
ruleset: str = "osu",
|
||||
current_user: DBUser = Depends(get_current_user),
|
||||
db: Session = Depends(get_db)
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
"""获取当前用户信息"""
|
||||
|
||||
|
||||
# 验证游戏模式
|
||||
valid_rulesets = ["osu", "taiko", "fruits", "mania"]
|
||||
if ruleset not in valid_rulesets:
|
||||
raise HTTPException(status_code=400, detail=f"Invalid ruleset. Must be one of: {valid_rulesets}")
|
||||
|
||||
raise HTTPException(
|
||||
status_code=400, detail=f"Invalid ruleset. Must be one of: {valid_rulesets}"
|
||||
)
|
||||
|
||||
# 转换用户数据
|
||||
api_user = convert_db_user_to_api_user(current_user, ruleset, db)
|
||||
return api_user
|
||||
@@ -171,26 +191,24 @@ async def health_check():
|
||||
return {"status": "ok", "timestamp": datetime.utcnow().isoformat()}
|
||||
|
||||
|
||||
|
||||
@app.get("/api/v2/friends")
|
||||
async def get_friends():
|
||||
return JSONResponse(content=[
|
||||
{
|
||||
"id": 123456,
|
||||
"username": "BestFriend",
|
||||
"is_online": True,
|
||||
"is_supporter": False,
|
||||
"country": {"code": "US", "name": "United States"}
|
||||
}
|
||||
])
|
||||
return JSONResponse(
|
||||
content=[
|
||||
{
|
||||
"id": 123456,
|
||||
"username": "BestFriend",
|
||||
"is_online": True,
|
||||
"is_supporter": False,
|
||||
"country": {"code": "US", "name": "United States"},
|
||||
}
|
||||
]
|
||||
)
|
||||
|
||||
|
||||
@app.get("/api/v2/notifications")
|
||||
async def get_notifications():
|
||||
return JSONResponse(content={
|
||||
"notifications": [],
|
||||
"unread_count": 0
|
||||
})
|
||||
return JSONResponse(content={"notifications": [], "unread_count": 0})
|
||||
|
||||
|
||||
@app.post("/api/v2/chat/ack")
|
||||
@@ -200,80 +218,80 @@ async def chat_ack():
|
||||
|
||||
@app.get("/api/v2/users/{user_id}/{mode}")
|
||||
async def get_user_mode(user_id: int, mode: str):
|
||||
return JSONResponse(content={
|
||||
"id": user_id,
|
||||
"username": "测试测试测",
|
||||
"statistics": {
|
||||
"level": {"current": 97, "progress": 96},
|
||||
"pp": 114514,
|
||||
"global_rank": 666,
|
||||
"country_rank": 1,
|
||||
"hit_accuracy": 100
|
||||
},
|
||||
"country": {"code": "JP", "name": "Japan"}
|
||||
})
|
||||
return JSONResponse(
|
||||
content={
|
||||
"id": user_id,
|
||||
"username": "测试测试测",
|
||||
"statistics": {
|
||||
"level": {"current": 97, "progress": 96},
|
||||
"pp": 114514,
|
||||
"global_rank": 666,
|
||||
"country_rank": 1,
|
||||
"hit_accuracy": 100,
|
||||
},
|
||||
"country": {"code": "JP", "name": "Japan"},
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
@app.get("/api/v2/me")
|
||||
async def get_me():
|
||||
return JSONResponse(content={
|
||||
"id": 15651670,
|
||||
"username": "Googujiang",
|
||||
"is_online": True,
|
||||
"country": {"code": "JP", "name": "Japan"},
|
||||
"statistics": {
|
||||
"level": {"current": 97, "progress": 96},
|
||||
"pp": 2826.26,
|
||||
"global_rank": 298026,
|
||||
"country_rank": 11220,
|
||||
"hit_accuracy": 95.7168
|
||||
return JSONResponse(
|
||||
content={
|
||||
"id": 15651670,
|
||||
"username": "Googujiang",
|
||||
"is_online": True,
|
||||
"country": {"code": "JP", "name": "Japan"},
|
||||
"statistics": {
|
||||
"level": {"current": 97, "progress": 96},
|
||||
"pp": 2826.26,
|
||||
"global_rank": 298026,
|
||||
"country_rank": 11220,
|
||||
"hit_accuracy": 95.7168,
|
||||
},
|
||||
}
|
||||
})
|
||||
)
|
||||
|
||||
|
||||
@app.post("/signalr/metadata/negotiate")
|
||||
async def metadata_negotiate(negotiateVersion: int = 1):
|
||||
return JSONResponse(content={
|
||||
"connectionId": "abc123",
|
||||
"availableTransports": [
|
||||
{
|
||||
"transport": "WebSockets",
|
||||
"transferFormats": ["Text", "Binary"]
|
||||
}
|
||||
]
|
||||
})
|
||||
return JSONResponse(
|
||||
content={
|
||||
"connectionId": "abc123",
|
||||
"availableTransports": [
|
||||
{"transport": "WebSockets", "transferFormats": ["Text", "Binary"]}
|
||||
],
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
@app.post("/signalr/spectator/negotiate")
|
||||
async def spectator_negotiate(negotiateVersion: int = 1):
|
||||
return JSONResponse(content={
|
||||
"connectionId": "spec456",
|
||||
"availableTransports": [
|
||||
{
|
||||
"transport": "WebSockets",
|
||||
"transferFormats": ["Text", "Binary"]
|
||||
}
|
||||
]
|
||||
})
|
||||
return JSONResponse(
|
||||
content={
|
||||
"connectionId": "spec456",
|
||||
"availableTransports": [
|
||||
{"transport": "WebSockets", "transferFormats": ["Text", "Binary"]}
|
||||
],
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
@app.post("/signalr/multiplayer/negotiate")
|
||||
async def multiplayer_negotiate(negotiateVersion: int = 1):
|
||||
return JSONResponse(content={
|
||||
"connectionId": "multi789",
|
||||
"availableTransports": [
|
||||
{
|
||||
"transport": "WebSockets",
|
||||
"transferFormats": ["Text", "Binary"]
|
||||
}
|
||||
]
|
||||
})
|
||||
return JSONResponse(
|
||||
content={
|
||||
"connectionId": "multi789",
|
||||
"availableTransports": [
|
||||
{"transport": "WebSockets", "transferFormats": ["Text", "Binary"]}
|
||||
],
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import uvicorn
|
||||
|
||||
uvicorn.run(
|
||||
"main:app",
|
||||
host=settings.HOST,
|
||||
port=settings.PORT,
|
||||
reload=settings.DEBUG
|
||||
"main:app", host=settings.HOST, port=settings.PORT, reload=settings.DEBUG
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user