78 lines
1.9 KiB
Python
78 lines
1.9 KiB
Python
|
import uuid
|
|||
|
|
from rest_framework.response import Response
|
|||
|
|
from rest_framework import status
|
|||
|
|
|
|||
|
|
|
|||
|
|
def convert_to_uuid(pk_str):
|
|||
|
|
"""
|
|||
|
|
将字符串转换为UUID格式
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
pk_str: 字符串形式的UUID
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
转换后的UUID对象或None
|
|||
|
|
|
|||
|
|
Raises:
|
|||
|
|
ValueError: 如果字符串不是有效的UUID格式
|
|||
|
|
"""
|
|||
|
|
if isinstance(pk_str, uuid.UUID):
|
|||
|
|
return pk_str
|
|||
|
|
|
|||
|
|
pk_str = pk_str.strip()
|
|||
|
|
|
|||
|
|
# 处理没有连字符的UUID
|
|||
|
|
if '-' not in pk_str and len(pk_str) == 32:
|
|||
|
|
pk_with_hyphens = f"{pk_str[0:8]}-{pk_str[8:12]}-{pk_str[12:16]}-{pk_str[16:20]}-{pk_str[20:]}"
|
|||
|
|
return uuid.UUID(pk_with_hyphens)
|
|||
|
|
else:
|
|||
|
|
return uuid.UUID(pk_str)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def format_user_response(user, include_is_active=False):
|
|||
|
|
"""
|
|||
|
|
格式化用户信息响应
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
user: 用户对象
|
|||
|
|
include_is_active: 是否包含is_active字段
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
格式化后的用户信息字典
|
|||
|
|
"""
|
|||
|
|
response = {
|
|||
|
|
"id": str(user.id),
|
|||
|
|
"username": user.name,
|
|||
|
|
"email": user.email,
|
|||
|
|
"name": user.name,
|
|||
|
|
"role": user.role,
|
|||
|
|
"department": user.department,
|
|||
|
|
"group": user.group
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if include_is_active:
|
|||
|
|
response["is_active"] = user.is_active
|
|||
|
|
|
|||
|
|
return response
|
|||
|
|
|
|||
|
|
|
|||
|
|
def validate_uuid_param(pk_str):
|
|||
|
|
"""
|
|||
|
|
验证并转换UUID参数，返回标准的错误响应或UUID对象
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
pk_str: 字符串形式的UUID
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
元组 (uuid_obj, None) 或 (None, error_response)
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
uuid_obj = convert_to_uuid(pk_str)
|
|||
|
|
return uuid_obj, None
|
|||
|
|
except ValueError:
|
|||
|
|
error_response = Response({
|
|||
|
|
"code": 400,
|
|||
|
|
"message": f"无效的用户ID格式: {pk_str}。用户ID应为有效的UUID格式。",
|
|||
|
|
"data": None
|
|||
|
|
}, status=status.HTTP_400_BAD_REQUEST)
|
|||
|
|
return None, error_response
|