homework 11 from lection 14

course_django/settings.py

@@ -37,10 +37,11 @@
     'django.contrib.sessions',
     'django.contrib.messages',
     'django.contrib.staticfiles',
-    'hr',
+
     'general',
     'chat',
     'examples',
+    'hr.apps.HrConfig',
 ]
 
 MIDDLEWARE = [

hr/models.py

@@ -5,6 +5,7 @@
 from django.core.cache import cache, caches
 # from django.utils.translation import gettext as _
 from django.utils.translation import gettext_lazy as _
+from django.db.models.signals import pre_save
 
 
 class Company(models.Model):

hr/signals.py

@@ -1,6 +1,6 @@
 from django.db.models.signals import pre_save
 from django.dispatch import receiver
-from hr.models import Position
+from hr.models import Position, Department
 import logging
 from hr.constants import MINIMUM_SALARY
 
@@ -14,3 +14,8 @@ def ensure_minimum_wage(sender, instance, **kwargs):
         logger.info(
             f"Заробітна плата для позиції '{instance.title}' була збільшена до мінімального порогу {MINIMUM_SALARY}.",
         )
+@receiver(pre_save, sender=Department)
+def capitalize_department_name(sender, instance, **kwargs):
+    if instance.name:
+        instance.name = instance.name.capitalize()
+        logger.info(f'Department name: {instance.name}')

hr/views/generic_views.py

@@ -65,6 +65,15 @@ class EmployeeUpdateView(UserIsAdminMixin, UpdateView):
     template_name = 'employee_form.html'
     success_url = reverse_lazy('hr:employee_list')
 
+    def form_valid(self, form):
+        response = super().form_valid(form)
+        messages.success(self.request, 'Employee updated successfully.')
+        return response
+
+    def form_invalid(self, form):
+        messages.error(self.request, 'Error! Employee not updated!')
+        return super().form_invalid(form)
+
 
 class EmployeeDeleteView(UserIsAdminMixin, DeleteView):
     model = Employee