SOITA/users/views.py

from django.http import HttpResponseRedirect
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
from django.shortcuts import render, redirect
from django.template import loader
from .forms import RegistrationForm
from django.contrib.auth.forms import AuthenticationForm
from django.contrib.auth import login as auth_login
from config.authh import SettingsBackend
from django.contrib.auth.decorators import login_required
from django.contrib.auth import logout as django_logout
from rest_framework.renderers import TemplateHTMLRenderer
from rest_framework.views import APIView
from django.contrib.auth.hashers import make_password


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


def logout(request):
    django_logout(request)
    return render(request, 'logout.html')

def register_success(request):
    return render(request, 'registerSuccess.html')

def register(request):
    context = {}
    if request.POST:
        form = RegistrationForm(request.POST)
        if form.is_valid():
            User.objects.create(
                email=form.cleaned_data["email"],
                first_name=form.cleaned_data["first_name"],
                last_name=form.cleaned_data["last_name"],
                password=form.cleaned_data["password1"],
                type="standard"
            )
            return redirect('register_success')
        else:
            context['registration_form'] = form
    else:
        form = RegistrationForm()
        context['registration_form'] = form
    return render(request, 'register.html', context)


@login_required
def login_success(request):
    return render(request, 'great.html')


def login(request):
    try:
        context = {}
        if request.POST:
            form = AuthenticationForm(request.POST)
            username = request.POST.get("username")
            password = request.POST.get("password")

            user = SettingsBackend().authenticate(request, email=username, password=password)
            if user is not None:
                auth_login(request, user)
                return redirect('home')
            context['login_form'] = form
        else:
            form = AuthenticationForm()
            context['login_form'] = form
    except:
        return render(request, 'login.html', {"messages": {"Podano błędne dane logowania lub użytkownik nie istnieje.": ""}})
    return render(request, 'login.html', context)


@login_required
def account(request):
    context = {}
    context['user'] = request.user
    return render(request, 'account.html', context)


@login_required
def changeEmail(request):
    if request.POST:
        email = request.POST.get("email")

        u = request.user
        u.email = email
        u.save()
        return redirect('account')
    return render(request, 'changeEmail.html')


@login_required
def changeName(request):
    if request.POST:
        firstName = request.POST.get("firstName")
        lastName = request.POST.get("lastName")

        u = request.user
        u.first_name = firstName
        u.last_name = lastName
        u.save()
        return redirect('account')
    return render(request, 'changeName.html')


@login_required
def changePassword(request):
    if request.POST:
        oldPassword = request.POST.get("oldPassword")
        newPassword = request.POST.get("newPassword")
        repeatPassword = request.POST.get("repeatPassword")
        if not request.user.check_password(oldPassword):
            message = {"Błędne hasło użytkownika.": ""}
            return render(request, 'changePassword.html', {"messages": message})

        if newPassword != repeatPassword:
            message = {"Nowe hasła nie są identyczne.": ""}
            return render(request, 'changePassword.html', {"messages": message})

        request.user.password = make_password(newPassword)
        request.user.save()
        return redirect('account')
    return render(request, 'changePassword.html')


def resetPassword(request):
    """
       ```json
       {
           "email": "test@test.com",
       }
       ```
    """

    if request.POST:
        email = request.POST.get("email")
        serializer_class = UserPasswordResetSerializer
        permission_classes = (permissions.AllowAny,)
        return redirect('resetPasswordConfirm')
    return render(request, 'resetPassword.html')


class ResetPassword(generics.GenericAPIView):
    renderer_classes = [TemplateHTMLRenderer]
    template_name = 'resetPassword.html'
    serializer_class = UserPasswordResetSerializer
    permission_classes = (permissions.AllowAny, )


    def get(self, request):
        return render(request, 'resetPassword.html')

    def post(self, request):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid()
        return redirect('resetPasswordConfirm')


class ResetPasswordConfirm(generics.GenericAPIView):
    renderer_classes = [TemplateHTMLRenderer]
    template_name = 'resetPasswordConfirm.html'
    serializer_class = UserPasswordResetConfirmSerializer
    permission_classes = (permissions.AllowAny, )

    def get(self, request):
        return render(request, 'resetPasswordConfirm.html')

    def post(self, request):
        serializer = UserPasswordResetConfirmSerializer(data=request.data)
        serializer.is_valid()
        return redirect('/users/login')


def resetPasswordConfirm(request):
    """
       ```json
       {
           "uid": "NYz",
           "token": "asdasdasd",
           "code": 123456,
           "newPassword": "testowe",
           "repeatNewPassword": "testowe"
       }
       ```
    """
    if request.POST:
        uid = request.POST.get("uid")
        token = request.POST.get("token")
        code = request.POST.get("code")
        newPassword = request.POST.get("newPassword")
        repeatPassword = request.POST.get("repeatNewPassword")

        serializer_class = UserPasswordResetConfirmSerializer
        permission_classes = (permissions.AllowAny,)
        token_generator = default_token_generator
        return redirect('login')
    return render(request, 'resetPasswordConfirm.html')