SOITA/users/views.py
2021-12-14 00:18:30 +01:00

62 lines
2.0 KiB
Python

from rest_framework import viewsets
from users.models import User
from users.serializers import UserSerializer
from rest_framework import views, exceptions, status, viewsets, permissions, generics
from tools.tools import PasswordResetShortcut, PasswordResetConfirmShortcut
from django.contrib.auth.tokens import default_token_generator
from .serializers import UserPasswordResetSerializer, UserPasswordResetConfirmSerializer, RegisterSerializer
from rest_framework.response import Response
class UserModelViewSet(viewsets.ModelViewSet):
queryset = User.objects.all()
serializer_class = UserSerializer
class RegisterViewSet(generics.GenericAPIView):
serializer_class = RegisterSerializer
permission_classes = (permissions.AllowAny, )
def post(self, request):
serializer = self.get_serializer(data=request.data)
if serializer.is_valid():
data = self.request.data
User.objects.create(
email=data["email"],
first_name=data["first_name"],
last_name=data["last_name"],
password=data["password"]
)
return Response({"detail": "Konto zostało pomyślnie założone. Możesz się zalogować!"}, status=status.HTTP_200_OK)
else:
return Response(data=serializer.errors, status=status.HTTP_400_BAD_REQUEST)
class PasswordReset(PasswordResetShortcut, generics.GenericAPIView):
"""
```json
{
"email": "testowymail@gmail.com"
}
```
"""
serializer_class = UserPasswordResetSerializer
permission_classes = (permissions.AllowAny, )
class UserPasswordResetConfirmView(PasswordResetConfirmShortcut, generics.GenericAPIView):
"""
```json
{
"uid": "NYz",
"token": "asdasdasd",
"code": 123456,
"newPassword": "testowe",
"repeatNewPassword": "testowe"
}
```
"""
serializer_class = UserPasswordResetConfirmSerializer
permission_classes = (permissions.AllowAny, )
token_generator = default_token_generator