90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
import json
|
||
import uuid
|
||
from fastapi import Depends, HTTPException
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from database.creator import get_db_session
|
||
from database.user import User, AuthME
|
||
from sqlalchemy import select, desc
|
||
from pydantic import BaseModel
|
||
from loguru import logger
|
||
from typing import Optional
|
||
|
||
|
||
class AuthmeRequest(BaseModel):
|
||
token: str
|
||
|
||
|
||
class AuthmeResponse(BaseModel):
|
||
code: int
|
||
message: str
|
||
|
||
|
||
async def fetch_user_by_token(
|
||
AuthmeRequest: AuthmeRequest,
|
||
asyncsession: AsyncSession = Depends(get_db_session)
|
||
) -> User:
|
||
"""
|
||
根据令牌获取用户信息
|
||
:param AuthmeRequest: 包含token的请求对象
|
||
:param asyncsession: 数据库会话
|
||
:return: User
|
||
"""
|
||
async with asyncsession as session:
|
||
# 根据token查找AuthME记录
|
||
result = await session.execute(
|
||
select(AuthME).where(AuthME.authme_token == AuthmeRequest.token)
|
||
)
|
||
authme = result.scalars().first()
|
||
|
||
if not authme:
|
||
raise HTTPException(status_code=401, detail="无效的令牌或用户不存在")
|
||
|
||
# 根据userid获取用户信息
|
||
user_result = await session.execute(
|
||
select(User).where(User.userid == authme.userid)
|
||
)
|
||
user = user_result.scalars().first()
|
||
if not user:
|
||
raise HTTPException(status_code=401, detail="用户不存在")
|
||
|
||
logger.info(f"User {user.userid} fetched successfully using token.")
|
||
return user
|
||
|
||
|
||
async def manage_user_tokens(userid: str, new_token: str, device_id: str, session: AsyncSession) -> None:
|
||
"""
|
||
管理用户token，每个用户最多保持5个设备会话，超出时删除最旧的2个
|
||
:param userid: 用户ID
|
||
:param new_token: 新的token
|
||
:param device_id: 设备标识符
|
||
:param session: 数据库会话
|
||
"""
|
||
# 检查当前用户的token数量
|
||
result = await session.execute(
|
||
select(AuthME)
|
||
.where(AuthME.userid == userid)
|
||
.order_by(desc(AuthME.create_date))
|
||
)
|
||
existing_tokens = result.scalars().all()
|
||
|
||
# 如果超过4个token（即将添加第6个），删除最旧的2个
|
||
if len(existing_tokens) >= 5:
|
||
# 删除最旧的2个token
|
||
oldest_tokens = existing_tokens[-2:]
|
||
for token_record in oldest_tokens:
|
||
await session.delete(token_record)
|
||
|
||
# 添加新的token记录
|
||
new_authme = AuthME(
|
||
userid=userid,
|
||
authme_token=new_token,
|
||
device_id=device_id
|
||
)
|
||
session.add(new_authme)
|
||
await session.commit()
|
||
|
||
|
||
def generate_device_id() -> str:
|
||
"""生成设备标识符"""
|
||
return str(uuid.uuid4())
|