Bugfix add auto assigning manager if he create project

managers/forms.py

@@ -1,17 +1,38 @@
+from typing import Any
+from typing import Dict
+from typing import List
+from typing import Optional
+
 from bootstrap_datepicker_plus import DatePickerInput
 from django import forms
+from django.forms import SelectMultiple
 
 from common.constants import CORRECT_DATE_FORMAT
 from managers.models import Project
+from users.models import CustomUser
+
+
+class ManagerSelectMultiple(SelectMultiple):
+    def optgroups(self, name: str, value: List, attrs: Optional[Dict] = None) -> List:
+        self.choices.queryset = CustomUser.objects.exclude(user_type=CustomUser.UserType.EMPLOYEE.name).exclude(
+            pk=self.user_pk if hasattr(self, "user_pk") else None
+        )
+        return super().optgroups(name, value, attrs)
 
 
 class ProjectAdminForm(forms.ModelForm):
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+        if hasattr(self, "user_pk"):
+            ManagerSelectMultiple.user_pk = self.user_pk
+
     class Meta:
         model = Project
         fields = "__all__"
         widgets = {
             "start_date": DatePickerInput(options={"format": CORRECT_DATE_FORMAT}),
             "stop_date": DatePickerInput(options={"format": CORRECT_DATE_FORMAT}),
+            "managers": ManagerSelectMultiple(),
         }
 
 

managers/views.py

@@ -94,6 +94,7 @@ class ProjectCreateView(CreateView):
 
     def get_context_data(self, **kwargs: Any) -> dict:
         logger.info(f"User with id: {self.request.user.pk} is in project create view")
+        ProjectAdminForm.user_pk = self.request.user.pk
         context_data = super().get_context_data(**kwargs)
         context_data["back_url"] = self.get_success_url()
         return context_data
@@ -102,6 +103,12 @@ def get_success_url(self) -> str:  # pylint: disable=no-self-use
         return reverse("custom-projects-list")
 
     def form_valid(self, form: ProjectAdminForm) -> HttpRequest:
+        if self.request.user not in form.cleaned_data["managers"]:
+            assert not self.request.user.is_employee
+            managers_pk_list = [manager.pk for manager in form.cleaned_data["managers"]]
+            managers_pk_list.append(self.request.user.pk)
+            managers_queryset = CustomUser.objects.filter(pk__in=managers_pk_list)
+            form.cleaned_data["managers"] = managers_queryset
         project = form.save()
         logger.info(f"New project with id: {project.pk} has been created")
         return super(ModelFormMixin, self).form_valid(form)  # pylint: disable=bad-super-call