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)