81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
from rest_framework import serializers, status, views
|
||
from rest_framework.response import Response
|
||
from django.contrib.auth import authenticate, get_user_model
|
||
|
||
User = get_user_model()
|
||
|
||
|
||
class LoginSerializer(serializers.Serializer):
|
||
username = serializers.CharField()
|
||
password = serializers.CharField()
|
||
|
||
|
||
class LoginView(views.APIView):
|
||
def post(self, request):
|
||
serializer = LoginSerializer(data=request.data)
|
||
if not serializer.is_valid():
|
||
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
||
|
||
username = serializer.validated_data['username']
|
||
password = serializer.validated_data['password']
|
||
|
||
user = authenticate(username=username, password=password)
|
||
if not user:
|
||
return Response(
|
||
{'detail': '用户名或密码错误'},
|
||
status=status.HTTP_401_UNAUTHORIZED
|
||
)
|
||
|
||
# 简单 Token（生产环境应该用 JWT）
|
||
import uuid
|
||
token = uuid.uuid4().hex
|
||
|
||
# 实际项目中应该存储 token 到数据库/缓存
|
||
# 这里简化处理
|
||
|
||
return Response({
|
||
'token': token,
|
||
'user': {
|
||
'id': user.id,
|
||
'username': user.username,
|
||
'email': user.email
|
||
}
|
||
})
|
||
|
||
|
||
class RegisterSerializer(serializers.Serializer):
|
||
username = serializers.CharField()
|
||
email = serializers.EmailField()
|
||
password = serializers.CharField()
|
||
|
||
|
||
class RegisterView(views.APIView):
|
||
def post(self, request):
|
||
serializer = RegisterSerializer(data=request.data)
|
||
if not serializer.is_valid():
|
||
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
||
|
||
try:
|
||
user = User.objects.create_user(
|
||
username=serializer.validated_data['username'],
|
||
email=serializer.validated_data['email'],
|
||
password=serializer.validated_data['password']
|
||
)
|
||
|
||
import uuid
|
||
token = uuid.uuid4().hex
|
||
|
||
return Response({
|
||
'token': token,
|
||
'user': {
|
||
'id': user.id,
|
||
'username': user.username,
|
||
'email': user.email
|
||
}
|
||
}, status=status.HTTP_201_CREATED)
|
||
except Exception as e:
|
||
return Response(
|
||
{'detail': str(e)},
|
||
status=status.HTTP_400_BAD_REQUEST
|
||
)
|