payments

common/views/employee_views.py

@@ -1,6 +1,11 @@
-from django.views.generic import TemplateView
+from django.contrib.auth.mixins import LoginRequiredMixin, UserPassesTestMixin
+from django.urls import reverse_lazy
+from django.views.generic import TemplateView, ListView, CreateView, UpdateView
 
 from booking.models import Booking
+from common.forms.company import CompanyCreateForm, CompanyUpdateForm
+from common.forms.line import LineCreateForm, LineUpdateForm
+from common.models import CompanyModel, LinesModel
 from containers.models import Container
 from preinfo.models import Preinfo
 
@@ -20,4 +25,98 @@ class EmployeeDashboardView(TemplateView):
         context['containers'] = containers
         context['preinfos'] = preinfos
         context['bookings'] = bookings
-        return context
+        return context
+
+
+class EmployeeCompanyListView(LoginRequiredMixin, UserPassesTestMixin, ListView):
+    model = CompanyModel
+    template_name = 'common/../../templates/employee/company-list.html'
+    context_object_name = 'objects'
+    paginate_by = 3
+    base_template = 'employee-base.html'
+
+    def test_func(self):
+        return True # self.request.user.has_employee_perm('can_view_preinfo') or self.request.user.user_type == 'CA'
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context['base_template'] = self.base_template
+        return context
+
+
+
+class EmployeeCompanyCreateView(LoginRequiredMixin, UserPassesTestMixin, CreateView):
+    model = CompanyModel
+    template_name = 'common/../../templates/employee/company-create.html'
+    form_class = CompanyCreateForm
+    success_url = reverse_lazy('employee_company')
+
+    def test_func(self):
+        return True # self.request.user.has_company_perm('can_create_preinfo') or self.request.user.user_type == 'CA'
+
+    # def form_valid(self, form):
+    #     form.instance.created_by = self.request.user
+    #     form.instance.updated_by = self.request.user
+    #     return super().form_valid(form)
+
+
+
+class EmployeeCompanyUpdateView(LoginRequiredMixin, UserPassesTestMixin, UpdateView):
+    model = CompanyModel
+    template_name = 'common/../../templates/employee/company-update.html'
+    form_class = CompanyUpdateForm
+    success_url = reverse_lazy('employee_company')
+
+    def test_func(self):
+        return True # self.request.user.ha.s_company_perm('can_edit_preinfo') or self.request.user.user_type == 'CA'
+
+    # def form_valid(self, form):
+    #     form.instance.updated_by = self.request.user
+    #     return super().form_valid(form)
+
+
+class EmployeeLineListView(LoginRequiredMixin, UserPassesTestMixin, ListView):
+    model = LinesModel
+    template_name = 'employee/line-list.html'
+    context_object_name = 'objects'
+    paginate_by = 3
+    base_template = 'employee-base.html'
+
+    def test_func(self):
+        return True # self.request.user.has_employee_perm('can_view_preinfo') or self.request.user.user_type == 'CA'
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context['base_template'] = self.base_template
+        return context
+
+
+
+class EmployeeLineCreateView(LoginRequiredMixin, UserPassesTestMixin, CreateView):
+    model = LinesModel
+    template_name = 'employee/company-create.html'
+    form_class = LineCreateForm
+    success_url = reverse_lazy('employee_line')
+
+    def test_func(self):
+        return True # self.request.user.has_company_perm('can_create_preinfo') or self.request.user.user_type == 'CA'
+
+    # def form_valid(self, form):
+    #     form.instance.created_by = self.request.user
+    #     form.instance.updated_by = self.request.user
+    #     return super().form_valid(form)
+
+
+
+class EmployeeLineUpdateView(LoginRequiredMixin, UserPassesTestMixin, UpdateView):
+    model = LinesModel
+    template_name = 'employee/company-update.html'
+    form_class = LineUpdateForm
+    success_url = reverse_lazy('employee_line')
+
+    def test_func(self):
+        return True # self.request.user.ha.s_company_perm('can_edit_preinfo') or self.request.user.user_type == 'CA'
+
+    # def form_valid(self, form):
+    #     form.instance.updated_by = self.request.user
+    #     return super().form_valid(form)