jobportal/users/views.py

from django.contrib.auth.decorators import login_required
from django.contrib.auth.views import LoginView, LogoutView
from django.contrib.messages.views import SuccessMessageMixin
from django.http import HttpResponseRedirect
from django.shortcuts import redirect
from django.urls import reverse
from django.utils.decorators import method_decorator
from django.views.generic import CreateView, UpdateView, DetailView, ListView

from jobs.models import Category, Job
from users.forms import AccountRegisterForm, UserUpdateForm, InviteEmployeeForm
from users.models import Profile, Account, Invite


class UserRegisterView(SuccessMessageMixin, CreateView):
    template_name = 'users/user-register.html'
    form_class = AccountRegisterForm
    success_url = '/'
    success_message = "Your user account has been created."

    def form_valid(self, form):
        user = form.save(commit=False)
        user_type = form.cleaned_data['user_types']
        if user_type == 'is_employee':
            user.is_employee = True
        elif user_type == 'is_employer':
            user.is_employer = True

        user.save()

        return redirect(self.success_url)


class UserLoginView(LoginView):
    template_name = 'users/login.html'


class UserLogoutView(LogoutView):
    template_name = 'users/login.html'


@method_decorator(login_required(login_url='/users/login'), name='dispatch')
class UserUpdateView(SuccessMessageMixin, UpdateView):
    model = Profile
    success_message = "You updated your profile!"
    template_name = 'users/update.html'
    form_class = UserUpdateForm

    def form_valid(self, form):
        form.instance.user = self.request.user
        return super(UserUpdateView, self).form_valid(form)

    def get(self, request, *args, **kwargs):
        self.object = self.get_object()
        if self.object.user != request.user:
            return HttpResponseRedirect('/')
        return super(UserUpdateView, self).get(request, *args, **kwargs)

    def get_success_url(self):
        return reverse('users: update_profile', kwargs={'pk': self.object.pk})


class EmployeeProfileView(CreateView):
    template_name = 'users/employee-profile.html'
    model = Account
    form_class = InviteEmployeeForm

    def get_context_data(self, **kwargs):
        context = super(EmployeeProfileView, self).get_context_data(**kwargs)
        context['account'] = Account.objects.get(pk=self.kwargs['employee_id'])
        context['profile'] = Profile.objects.get(user_id=self.kwargs['employee_id'])
        context['job'] = Job.objects.get(id=self.kwargs['job_id'])
        context['categories'] = Category.objects.all()
        return context

    def form_valid(self, form):
        instance = form.save(commit=False)
        instance.user = Account.objects.get(pk=self.kwargs['employee_id'])
        instance.job = Job.objects.get(pk=self.kwargs['job_id'])
        instance.save()
        return super(EmployeeProfileView, self).form_valid(form)

    def get_success_url(self):
        return reverse('users:employer_jobs')


@method_decorator(login_required(login_url='/users/login'), name='dispatch')
class EmployerPostedJobsView(ListView):
    template_name = 'users/employer-posted-jobs.html'
    context_object_name = 'employer_jobs'
    model = Job
    paginate_by = 3

    def get_queryset(self):
        return Job.objects.filter(employer=self.request.user).order_by('-id')


@method_decorator(login_required(login_url='/users/login'), name='dispatch')
class EmployeeMessagesView(ListView):
    model = Job
    template_name = 'users/employee-messages.html'
    paginate_by = 5
    context_object_name = 'jobs'

    def get_queryset(self):
        return Job.objects.filter(invites__isnull=False, invites__user_id=self.request.user).order_by('-invites')


class EmployeeDisplayMessages(DetailView):
    model = Invite
    template_name = 'users/employee-display-messages.html'
    context_object_name = 'invite'

    def get_queryset(self):
        invite = self.model.objects.filter(id=self.kwargs['pk'])
        invite.update(unread=False)
        return invite

    def get(self, request, *args, **kwargs):
        self.object = self.get_object()
        if self.object.user != request.user:
            return HttpResponseRedirect('/')
        return super(EmployeeDisplayMessages, self).get(request, *args, **kwargs)


@method_decorator(login_required(login_url='/users/login'), name='dispatch')
class AddWishListView(UpdateView):
    template_name = 'jobs/index.html'
    model = Profile

    def get(self, request, *args, **kwargs):
        if self.request.user.is_employee:
            job = Job.objects.get(id=self.kwargs['pk'])
            profile = Profile.objects.get(user=request.user)
            profile.wish_list.add(job)
            return redirect('jobs:home')
        else:
            return redirect('jobs:home')


@method_decorator(login_required(login_url='/users/login'), name='dispatch')
class RemoveFromWishListView(UpdateView):
    template_name = 'jobs/index.html'
    model = Profile

    def get(self, request, *args, **kwargs):
        if self.request.user.is_employee:
            job = Job.objects.get(id=self.kwargs['pk'])
            profile = Profile.objects.get(user=request.user)
            profile.wish_list.remove(job)
            return redirect('jobs:home')
        else:
            return redirect('jobs:home')


class MyWishList(ListView):
    template_name = 'users/my-wish-list.html'
    context_object_name = 'jobs'
    model = Job
    paginate_by = 5

    def get_queryset(self):
        return Job.objects.filter(wish_list__user_id=self.request.user.id)

    def get_context_data(self, **kwargs):
        context = super(MyWishList, self).get_context_data(**kwargs)
        if self.request.user.is_authenticated:
            context['wish_list'] = Job.objects.filter(wish_list__user_id=self.request.user.id).values_list('id',
                                                                                                           flat=True)

        return context