wbhuman_resources 1.58.4__py2.py3-none-any.whl

wbhuman_resources/viewsets/employee.py

@@ -0,0 +1,210 @@
+from django.db.models import F, OuterRef, Subquery
+from rest_framework.decorators import action
+from rest_framework.response import Response
+from wbcore import viewsets
+from wbcore.contrib.directory.models import Person
+from wbcore.utils.date import current_month_date_end
+
+from wbhuman_resources.filters import (
+    EmployeeBalanceFilterSet,
+    EmployeeFilterSet,
+    PositionFilterSet,
+)
+from wbhuman_resources.models import (
+    EmployeeHumanResource,
+    EmployeeWeeklyOffPeriods,
+    EmployeeYearBalance,
+    Position,
+    deactivate_profile_as_task,
+)
+from wbhuman_resources.serializers import (
+    EmployeeBalanceModelSerializer,
+    EmployeeHumanResourceRepresentationSerializer,
+    EmployeeModelSerializer,
+    EmployeeWeeklyOffPeriodsModelSerializer,
+    EmployeeYearBalanceModelSerializer,
+    EmployeeYearBalanceRepresentationSerializer,
+    PositionModelSerializer,
+    PositionRepresentationSerializer,
+)
+from wbhuman_resources.viewsets.buttons import (
+    EmployeeButtonConfig,
+    YearBalanceEmployeeHumanResourceButtonConfig,
+)
+from wbhuman_resources.viewsets.display import (
+    EmployeeBalanceDisplayConfig,
+    EmployeeDisplayConfig,
+    PositionDisplayConfig,
+    WeeklyOffPeriodEmployeeHumanResourceDisplayConfig,
+    YearBalanceEmployeeHumanResourceDisplayConfig,
+)
+from wbhuman_resources.viewsets.endpoints import (
+    EmployeeBalanceEndpointConfig,
+    WeeklyOffPeriodEmployeeHumanResourceEndpointConfig,
+    YearBalanceEmployeeHumanResourceEndpointConfig,
+)
+from wbhuman_resources.viewsets.titles import (
+    EmployeeBalanceTitleConfig,
+    EmployeeTitleConfig,
+)
+
+from .mixins import EmployeeViewMixin
+
+
+class PositionRepresentationViewSet(viewsets.RepresentationViewSet):
+    ordering_fields = ordering = ("name",)
+    search_fields = ("name",)
+    filterset_fields = {"height": ["exact"], "level": ["exact"]}
+    queryset = Position.objects.all()
+    serializer_class = PositionRepresentationSerializer
+
+
+class EmployeeHumanResourceRepresentationViewSet(viewsets.RepresentationViewSet):
+    ordering_fields = ordering = ("profile__computed_str",)
+    search_fields = ("profile__computed_str",)
+
+    queryset = EmployeeHumanResource.active_employees.select_related("profile")
+    serializer_class = EmployeeHumanResourceRepresentationSerializer
+
+
+class EmployeeYearBalanceRepresentationViewSet(viewsets.RepresentationViewSet):
+    ordering_fields = ordering = "-year"
+
+    queryset = EmployeeYearBalance.objects.all()
+    serializer_class = EmployeeYearBalanceRepresentationSerializer
+
+
+class PositionModelViewSet(viewsets.ModelViewSet):
+    queryset = Position.objects.select_related(
+        "parent",
+        "manager",
+    )
+    serializer_class = PositionModelSerializer
+
+    filterset_class = PositionFilterSet
+    ordering_fields = ordering = ["name"]
+    search_fields = ["name"]
+
+    display_config_class = PositionDisplayConfig
+
+
+class EmployeeBalanceModelViewSet(EmployeeViewMixin, viewsets.ModelViewSet):
+    queryset = EmployeeHumanResource.objects.all()
+    serializer_class = EmployeeBalanceModelSerializer
+
+    filterset_class = EmployeeBalanceFilterSet
+    ordering_fields = [
+        "profile__computed_str",
+        "contract_type",
+        "is_active",
+        "position__name",
+        "extra_days_frequency",
+        "extra_days_per_period",
+        "took_long_vacations",
+        "available_vacation_balance_previous_year",
+        "available_vacation_balance_current_year",
+        "available_vacation_balance_next_year",
+    ]
+    ordering = ["profile__computed_str"]
+    search_fields = ["profile__computed_str"]
+
+    title_config_class = EmployeeBalanceTitleConfig
+    display_config_class = EmployeeBalanceDisplayConfig
+    endpoint_config_class = EmployeeBalanceEndpointConfig
+
+    def get_queryset(self):
+        qs = EmployeeHumanResource.objects.none()
+        if self.is_administrator:
+            qs = super().get_queryset()
+        if employee := getattr(self.request.user.profile, "human_resources", None):
+            qs = employee.get_managed_employees()
+        return (
+            EmployeeHumanResource.annotated_queryset(qs, current_month_date_end())
+            .filter(balances__isnull=False)
+            .distinct()
+        ).select_related(
+            "profile",
+            "position",
+            "calendar",
+        )
+
+
+class EmployeeModelViewSet(EmployeeViewMixin, viewsets.ModelViewSet):
+    queryset = EmployeeHumanResource.objects.all()
+    serializer_class = EmployeeModelSerializer
+
+    ordering_fields = [
+        "profile__computed_str",
+        "position__name",
+        "top_position_repr",
+        "primary_telephone",
+        "primary_email",
+        "primary_address",
+    ]
+    ordering = ["profile__computed_str"]
+    search_fields = ["profile__computed_str", "primary_telephone", "primary_email", "primary_address"]
+
+    filterset_class = EmployeeFilterSet
+
+    title_config_class = EmployeeTitleConfig
+    display_config_class = EmployeeDisplayConfig
+    button_config_class = EmployeeButtonConfig
+
+    def get_queryset(self):
+        return (
+            super()
+            .get_queryset()
+            .annotate(
+                top_position_repr=F("position__parent__name"),
+                position_manager=F("position__manager"),
+                primary_email=Subquery(
+                    Person.objects.filter(id=OuterRef("profile")).annotate_all().values("primary_email")[:1]
+                ),
+                primary_address=Subquery(
+                    Person.objects.filter(id=OuterRef("profile")).annotate_all().values("primary_address")[:1]
+                ),
+                primary_telephone=Subquery(
+                    Person.objects.filter(id=OuterRef("profile")).annotate_all().values("primary_telephone")[:1]
+                ),
+            )
+            .select_related(
+                "direct_manager",
+                "position",
+                "calendar",
+                "profile",
+            )
+            .prefetch_related("balances", "position__manager")
+        )
+
+    @action(detail=True, methods=["PATCH"])
+    def deactivate(self, request, pk=None):
+        if pk and EmployeeHumanResource.is_administrator(self.request.user):
+            deactivate_profile_as_task.delay(request.user.id, pk, request.POST.get("substitute", None))
+        return Response({"send": True})
+
+
+# Employee Subs Viewsets
+
+
+class WeeklyOffPeriodEmployeeHumanResourceModelViewSet(viewsets.ModelViewSet):
+    display_config_class = WeeklyOffPeriodEmployeeHumanResourceDisplayConfig
+    endpoint_config_class = WeeklyOffPeriodEmployeeHumanResourceEndpointConfig
+
+    serializer_class = EmployeeWeeklyOffPeriodsModelSerializer
+    queryset = EmployeeWeeklyOffPeriods.objects.all()
+
+    def get_queryset(self):
+        return super().get_queryset().filter(employee=self.kwargs["employee_id"])
+
+
+class YearBalanceEmployeeHumanResourceModelViewset(viewsets.ModelViewSet):
+    display_config_class = YearBalanceEmployeeHumanResourceDisplayConfig
+    endpoint_config_class = YearBalanceEmployeeHumanResourceEndpointConfig
+    button_config_class = YearBalanceEmployeeHumanResourceButtonConfig
+
+    ordering = ["-year"]
+    serializer_class = EmployeeYearBalanceModelSerializer
+    queryset = EmployeeYearBalance.objects.all()
+
+    def get_queryset(self):
+        return super().get_queryset().filter(employee=self.kwargs["employee_id"])

wbhuman_resources/viewsets/endpoints/__init__.py

@@ -0,0 +1,42 @@
+from .absence import (
+    AbsenceRequestCrossBorderCountryEndpointConfig,
+    AbsenceRequestEndpointConfig,
+    AbsenceRequestPeriodsAbsenceRequestEndpointConfig,
+    AbsenceRequestPlannerEndpointConfig,
+    AbsenceTablePandasEndpointConfig,
+    AbsenceTypeCountEmployeeEndpointConfig,
+)
+from .calendars import (
+    DayOffDayOffCalendarEndpointConfig,
+    DefaultDailyPeriodDayOffCalendar,
+)
+from .employee import (
+    AbsenceRequestEmployeeHumanResourceEndpointConfig,
+    EmployeeBalanceEndpointConfig,
+    EmployeeEndpointConfig,
+    WeeklyOffPeriodEmployeeHumanResourceEndpointConfig,
+    YearBalanceEmployeeHumanResourceEndpointConfig,
+)
+from .kpis import (
+    EvaluationGraphEndpointConfig,
+    KPIEndpointConfig,
+    KPIEvaluationEndpointConfig,
+    KPIEvaluationPandasEndpointConfig,
+)
+from .review import (
+    ReviewAnswerReviewPandasEndpointConfig,
+    ReviewAnswerEndpointConfig,
+    ReviewAnswerReviewNoCategoryEndpointConfig,
+    ReviewAnswerReviewQuestionCategoryEndpointConfig,
+    ReviewEndpointConfig,
+    ReviewGroupEndpointConfig,
+    ReviewProgressPandasEndpointConfig,
+    ReviewProgressReviewPandasEndpointConfig,
+    ReviewQuestionCategoryEndpointConfig,
+    ReviewQuestionEndpointConfig,
+    ReviewQuestionReviewCategoryEndpointConfig,
+    ReviewQuestionReviewEndpointConfig,
+    ReviewQuestionReviewNoCategoryEndpointConfig,
+    ReviewQuestionReviewQuestionCategoryEndpointConfig,
+    ReviewReviewGroupEndpointConfig,
+)

wbhuman_resources/viewsets/endpoints/absence.py

@@ -0,0 +1,57 @@
+from rest_framework.reverse import reverse
+from wbcore.metadata.configs.endpoints import EndpointViewConfig
+
+from wbhuman_resources.models import AbsenceRequest, EmployeeHumanResource
+
+
+class AbsenceRequestEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return reverse(f"{self.view.get_model().get_endpoint_basename()}-list", request=self.request)
+
+    def get_instance_endpoint(self, **kwargs):
+        if self.instance:
+            obj = self.view.get_object()
+            if obj.status != AbsenceRequest.Status.DRAFT.name and not EmployeeHumanResource.is_administrator(
+                self.request.user
+            ):
+                return None
+        return self.get_endpoint()
+
+    def get_create_endpoint(self, **kwargs):
+        return self.get_endpoint()
+
+    def get_delete_endpoint(self, **kwargs):
+        if self.instance:
+            obj = self.view.get_object()
+            if obj.is_deletable_for_user(self.request.user):
+                return self.get_endpoint()
+        return None
+
+
+class AbsenceRequestCrossBorderCountryEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return reverse(
+            "wbhuman_resources:absencerequesttype-crossbordercountry-list",
+            args=[self.view.kwargs["absencerequesttype_id"]],
+            request=self.request,
+        )
+
+
+class AbsenceTypeCountEmployeeEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return None
+
+
+class AbsenceRequestPeriodsAbsenceRequestEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return None
+
+
+class AbsenceRequestPlannerEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return None
+
+
+class AbsenceTablePandasEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return None

wbhuman_resources/viewsets/endpoints/calendars.py

@@ -0,0 +1,18 @@
+from rest_framework.reverse import reverse
+from wbcore.metadata.configs.endpoints import EndpointViewConfig
+
+
+class DayOffDayOffCalendarEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return reverse(
+            "wbhuman_resources:calendar-dayoff-list", args=[self.view.kwargs["calendar_id"]], request=self.request
+        )
+
+
+class DefaultDailyPeriodDayOffCalendar(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return reverse(
+            "wbhuman_resources:calendar-defaultperiod-list",
+            args=[self.view.kwargs["calendar_id"]],
+            request=self.request,
+        )

wbhuman_resources/viewsets/endpoints/employee.py

@@ -0,0 +1,51 @@
+from rest_framework.reverse import reverse
+from wbcore.metadata.configs.endpoints import EndpointViewConfig
+
+
+class AbsenceRequestEmployeeHumanResourceEndpointConfig(EndpointViewConfig):
+    def get_create_endpoint(self, **kwargs):
+        return reverse("wbhuman_resources:absencerequest-list", args=[], request=self.request)
+
+    def get_endpoint(self, **kwargs):
+        return reverse(
+            "wbhuman_resources:employee-absencerequest-list",
+            args=[self.view.kwargs["employee_id"]],
+            request=self.request,
+        )
+
+
+class YearBalanceEmployeeHumanResourceEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return reverse(
+            "wbhuman_resources:employee-employeeyearbalance-list",
+            args=[self.view.kwargs["employee_id"]],
+            request=self.request,
+        )
+
+    def get_create_endpoint(self, **kwargs):
+        return None
+
+    def get_delete_endpoint(self, **kwargs):
+        return None
+
+
+class EmployeeBalanceEndpointConfig(EndpointViewConfig):
+    def get_instance_endpoint(self, **kwargs):
+        return reverse("wbhuman_resources:employeebalance-list", args=[], request=self.request)
+
+    def get_endpoint(self, **kwargs):
+        return None
+
+
+class EmployeeEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return None
+
+
+class WeeklyOffPeriodEmployeeHumanResourceEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return reverse(
+            "wbhuman_resources:employee-weeklyoffperiod-list",
+            args=[self.view.kwargs["employee_id"]],
+            request=self.request,
+        )

wbhuman_resources/viewsets/endpoints/kpis.py

@@ -0,0 +1,53 @@
+from rest_framework.reverse import reverse
+from wbcore.metadata.configs.endpoints import EndpointViewConfig
+
+from wbhuman_resources.models import KPI
+
+
+class KPIEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return super().get_endpoint()
+
+    def get_instance_endpoint(self, **kwargs):
+        if self.instance:
+            if not KPI.is_administrator(self.request.user):
+                return None
+        return super().get_instance_endpoint()
+
+    def get_create_endpoint(self, **kwargs):
+        if KPI.is_administrator(self.request.user):
+            return super().get_instance_endpoint()
+        return None
+
+    def get_delete_endpoint(self, **kwargs):
+        if KPI.is_administrator(self.request.user):
+            return super().get_delete_endpoint()
+        return None
+
+
+class KPIEvaluationEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return reverse(
+            "wbhuman_resources:kpi-evaluation-list",
+            args=[self.view.kwargs["kpi_id"]],
+            request=self.request,
+        )
+
+    def get_instance_endpoint(self, **kwargs):
+        if self.instance:
+            return None
+        return super().get_instance_endpoint()
+
+    def get_create_endpoint(self, **kwargs):
+        return None
+
+    def get_delete_endpoint(self, **kwargs):
+        return None
+
+
+class EvaluationGraphEndpointConfig(EndpointViewConfig):
+    pass
+
+
+class KPIEvaluationPandasEndpointConfig(EndpointViewConfig):
+    pass

wbhuman_resources/viewsets/endpoints/review.py

@@ -0,0 +1,191 @@
+from django.db.models import Q
+from rest_framework.reverse import reverse
+from wbcore.metadata.configs.endpoints import EndpointViewConfig
+
+from wbhuman_resources.models import Review
+
+
+class ReviewGroupEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return super().get_endpoint()
+
+
+class ReviewEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        if "pk" in self.view.kwargs:
+            obj = Review.objects.get(id=self.view.kwargs.get("pk"))
+            if (moderator := obj.moderator) and (moderator != self.request.user.profile):
+                return None
+            if obj.status in [Review.Status.REVIEW, Review.Status.EVALUATION, Review.Status.VALIDATION]:
+                return None
+        # Block click on the row to access to the view if it's not an administrator and no review is linked to this person and only has read access
+        elif (
+            self.request.user in Review.get_administrators()
+            and not self.request.user.has_perm("wbcompliance.add_review")
+            and self.request.user not in Review.get_administrators()
+            and self.view.get_queryset()
+            .filter(
+                Q(moderator=self.request.user.profile)
+                | Q(reviewer=self.request.user.profile)
+                | Q(reviewee=self.request.user.profile)
+            )
+            .count()
+            == 0
+        ):
+            return None
+        return super().get_endpoint()
+
+    def get_delete_endpoint(self, **kwargs):
+        if "pk" in self.view.kwargs:
+            obj = Review.objects.get(id=self.view.kwargs.get("pk"))
+            if obj.status in [Review.Status.REVIEW, Review.Status.EVALUATION, Review.Status.VALIDATION]:
+                return None
+            elif obj.moderator == self.request.user.profile:
+                return super().get_delete_endpoint()
+        return None
+
+
+class ReviewReviewGroupEndpointConfig(ReviewEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if review_group_id := self.view.kwargs.get("review_group_id", None):
+            if "pk" in self.view.kwargs:
+                obj = Review.objects.get(id=self.view.kwargs.get("pk"))
+                if (moderator := obj.moderator) and (moderator != self.request.user.profile):
+                    return None
+                if obj.status in [Review.Status.REVIEW, Review.Status.EVALUATION, Review.Status.VALIDATION]:
+                    return None
+            return reverse("wbhuman_resources:reviewgroup-review-list", args=[review_group_id], request=self.request)
+        return None
+
+
+class ReviewQuestionCategoryEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return super().get_endpoint()
+
+
+class ReviewQuestionEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return super().get_endpoint()
+
+
+class ReviewQuestionReviewEndpointConfig(ReviewQuestionEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if review_id := self.view.kwargs.get("review_id", None):
+            return reverse("wbhuman_resources:review-reviewquestion-list", args=[review_id], request=self.request)
+        return None
+
+    def get_create_endpoint(self, **kwargs):
+        if review_id := self.view.kwargs.get("review_id", None):
+            obj = Review.objects.get(id=review_id)
+            if obj.moderator == self.request.user.profile:
+                return self.get_endpoint()
+        return None
+
+    def get_delete_endpoint(self, **kwargs):
+        if review_id := self.view.kwargs.get("review_id", None):
+            obj = Review.objects.get(id=review_id)
+            if obj.moderator == self.request.user.profile:
+                if "pk" in self.view.kwargs:
+                    return f'{self.get_endpoint()}{self.view.kwargs["pk"]}/'
+                return super().get_delete_endpoint()
+        return None
+
+
+class ReviewQuestionReviewNoCategoryEndpointConfig(ReviewQuestionReviewEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if review_id := self.view.kwargs.get("review_id", None):
+            return reverse(
+                "wbhuman_resources:review-reviewquestionnocategory-list", args=[review_id], request=self.request
+            )
+        return None
+
+
+class ReviewQuestionReviewCategoryEndpointConfig(ReviewQuestionReviewEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if (review_id := self.view.kwargs.get("review_id", None)) and (
+            category_id := self.view.kwargs.get("category_id", None)
+        ):
+            return reverse(
+                "wbhuman_resources:review-reviewquestioncategory-list",
+                args=[review_id, category_id],
+                request=self.request,
+            )
+        return None
+
+
+class ReviewQuestionReviewQuestionCategoryEndpointConfig(ReviewQuestionEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if category_id := self.view.kwargs.get("category_id", None):
+            return reverse(
+                "wbhuman_resources:reviewquestioncategory-reviewquestion-list",
+                args=[category_id],
+                request=self.request,
+            )
+        return None
+
+
+class ReviewAnswerEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return super().get_endpoint()
+
+    def get_instance_endpoint(self, **kwargs):
+        if self.instance:
+            obj = self.view.get_object()
+            if obj.question.review.status in [
+                Review.Status.PREPARATION_OF_REVIEW,
+                Review.Status.REVIEW,
+                Review.Status.EVALUATION,
+                Review.Status.VALIDATION,
+            ]:
+                return None
+        return super().get_endpoint()
+
+    def get_create_endpoint(self, **kwargs):
+        return None
+
+    def get_delete_endpoint(self, **kwargs):
+        return None
+
+
+class ReviewAnswerReviewNoCategoryEndpointConfig(ReviewAnswerEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if review_id := self.view.kwargs.get("review_id", None):
+            return reverse(
+                "wbhuman_resources:review-reviewanswerquestionnocategory-list", args=[review_id], request=self.request
+            )
+        return None
+
+
+class ReviewAnswerReviewQuestionCategoryEndpointConfig(ReviewAnswerEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if (review_id := self.view.kwargs.get("review_id", None)) and (
+            category_id := self.view.kwargs.get("category_id", None)
+        ):
+            return reverse(
+                "wbhuman_resources:review-reviewanswerquestioncategory-list",
+                args=[review_id, category_id],
+                request=self.request,
+            )
+        return None
+
+
+class ReviewProgressReviewPandasEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return None
+
+
+class ReviewAnswerReviewPandasEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return reverse(
+            "wbhuman_resources:review-reviewquestion-list",
+            [self.view.kwargs["review_id"]],
+            request=self.request,
+        )
+
+
+class ReviewProgressPandasEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return None
+
+    def get_instance_endpoint(self, **kwargs):
+        return reverse("wbhuman_resources:review-list", args=[], request=self.request)