wbhuman_resources 1.58.4__py2.py3-none-any.whl

wbhuman_resources/filters/employee.py

@@ -0,0 +1,81 @@
+from django.db.models import Q
+from django.utils.translation import gettext_lazy as _
+from wbcore import filters as wb_filters
+
+from wbhuman_resources.models import EmployeeHumanResource, Position
+
+
+class PositionFilterSet(wb_filters.FilterSet):
+    class Meta:
+        model = Position
+        fields = {"name": ["exact", "icontains"], "parent": ["exact"], "level": ["exact"], "manager": ["exact"]}
+
+
+class BaseEmployeeFilterSet(wb_filters.FilterSet):
+    contract_type = wb_filters.ChoiceFilter(
+        label=_("Contract"),
+        choices=EmployeeHumanResource.ContractType.choices,
+        initial=EmployeeHumanResource.ContractType.INTERNAL,
+    )
+    is_active = wb_filters.BooleanFilter(label=_("Is Active"), initial=True)
+    position = wb_filters.ModelChoiceFilter(
+        label=_("Position"),
+        queryset=Position.objects.all(),
+        endpoint=Position.get_representation_endpoint(),
+        value_key=Position.get_representation_value_key(),
+        label_key=Position.get_representation_label_key(),
+        method="filter_position",
+        filter_params={"height": 0},
+    )
+
+    def filter_position(self, queryset, name, value):
+        if value:
+            return queryset.filter(position__in=value.get_descendants(include_self=True)).distinct()
+        return queryset
+
+
+class EmployeeBalanceFilterSet(BaseEmployeeFilterSet):
+    took_long_vacations = wb_filters.BooleanFilter(field_name="took_long_vacations", lookup_expr="exact")
+
+    extra_days_per_period__gte = wb_filters.NumberFilter(
+        field_name="extra_days_per_period", method="filter_str_number_gte", lookup_expr="gte"
+    )
+
+    def filter_str_number_lte(self, queryset, name, value):
+        if value:
+            numerical_field_name = name.replace("_repr", "")
+            return queryset.filter(**{f"{numerical_field_name}__lte": value})
+        return queryset
+
+    def filter_str_number_gte(self, queryset, name, value):
+        if value:
+            numerical_field_name = name.replace("_repr", "")
+            return queryset.filter(**{f"{numerical_field_name}__gte": value})
+        return queryset
+
+    class Meta:
+        model = EmployeeHumanResource
+        fields = {"profile": ["exact"], "position": ["exact"], "extra_days_frequency": ["exact"]}
+
+
+class EmployeeFilterSet(BaseEmployeeFilterSet):
+    top_position_repr = wb_filters.ModelChoiceFilter(
+        label=_("Position"),
+        queryset=Position.objects.all(),
+        endpoint=Position.get_representation_endpoint(),
+        value_key=Position.get_representation_value_key(),
+        label_key=Position.get_representation_label_key(),
+        filter_params={"height": 1},
+        method="filter_top_position",
+    )
+
+    def filter_top_position(self, queryset, name, value):
+        if value:
+            return queryset.filter(
+                Q(position__parent__isnull=False) & Q(position__parent__in=value.get_descendants(include_self=True))
+            ).distinct()
+        return queryset
+
+    class Meta:
+        model = EmployeeHumanResource
+        fields = {"calendar": ["exact"], "contract_type": ["exact"], "enrollment_at": ["gte", "lte"]}

wbhuman_resources/filters/kpi.py

@@ -0,0 +1,35 @@
+from django.utils.translation import gettext_lazy as _
+from wbcore import filters as wb_filters
+
+from wbhuman_resources.models import KPI, Evaluation
+
+
+class KPIFilterSet(wb_filters.FilterSet):
+    class Meta:
+        model = KPI
+        fields = {
+            "name": ["exact", "icontains"],
+            "evaluated_persons": ["exact"],
+        }
+
+
+class KPIEvaluationFilterSet(wb_filters.FilterSet):
+    class Meta:
+        model = Evaluation
+        fields = {
+            "person": ["exact"],
+            "evaluation_date": ["lte", "gte"],
+        }
+
+
+class KPIEvaluationPandasFilter(wb_filters.FilterSet):
+    kpi_name = wb_filters.CharFilter(label=_("KPI"), lookup_expr="icontains")
+    goal = wb_filters.CharFilter(label=_("Goal"), lookup_expr="icontains")
+
+    class Meta:
+        model = Evaluation
+        fields = {
+            "person": ["exact"],
+            "kpi": ["exact"],
+            "evaluation_date": ["lte", "gte"],
+        }

wbhuman_resources/filters/review.py

@@ -0,0 +1,134 @@
+from django.db.models import Q
+from django.utils.translation import gettext_lazy as _
+from wbcore import filters as wb_filters
+from wbcore.contrib.directory.models import Person
+
+from wbhuman_resources.models import (
+    Review,
+    ReviewAnswer,
+    ReviewGroup,
+    ReviewQuestion,
+    ReviewQuestionCategory,
+)
+
+
+class ReviewGroupFilter(wb_filters.FilterSet):
+    class Meta:
+        model = ReviewGroup
+        fields = {
+            "name": ["exact", "icontains"],
+        }
+
+
+def get_filter_params(request, view):
+    return {"related_to": request.user.profile}
+
+
+def filter_default_related_to(field, request, view):
+    return request.user.profile.id if request.user.profile else None
+
+
+class ReviewTemplateFilter(wb_filters.FilterSet):
+    related_to = wb_filters.ModelChoiceFilter(
+        label=_("Related to"),
+        queryset=Person.objects.all(),
+        endpoint=Person.get_representation_endpoint(),
+        value_key=Person.get_representation_value_key(),
+        label_key=Person.get_representation_label_key(),
+        method="filter_related_to",
+        initial=filter_default_related_to,
+    )
+
+    def filter_related_to(self, queryset, name, value):
+        if value:
+            queryset = queryset.filter(Q(reviewee=value) | Q(reviewer=value) | Q(moderator=value))
+        return queryset
+
+    class Meta:
+        model = Review
+        fields = {
+            "review": ["lte", "gte"],
+            "status": ["exact"],
+            "moderator": ["exact"],
+            "review_group": ["exact"],
+            "changed": ["lte", "gte"],
+        }
+
+
+class ReviewFilter(ReviewTemplateFilter):
+    is_template = wb_filters.BooleanFilter(initial=False, label=_("Is Template"))
+
+    class Meta:
+        model = Review
+        fields = {
+            "from_date": ["lte", "gte"],
+            "to_date": ["lte", "gte"],
+            "review_deadline": ["lte", "gte"],
+            "review": ["lte", "gte"],
+            "status": ["exact"],
+            "reviewee": ["exact"],
+            "reviewer": ["exact"],
+            "moderator": ["exact"],
+            "review_group": ["exact"],
+            "is_template": ["exact"],
+            "year": ["exact"],
+            "type": ["exact"],
+            "changed": ["lte", "gte"],
+        }
+
+
+class ReviewQuestionCategoryFilter(wb_filters.FilterSet):
+    class Meta:
+        model = ReviewQuestionCategory
+        fields = {
+            "name": ["exact", "icontains"],
+            "order": ["exact"],
+            "weight": ["exact"],
+        }
+
+
+class ReviewQuestionFilter(wb_filters.FilterSet):
+    class Meta:
+        model = ReviewQuestion
+        fields = {
+            "review": ["exact"],
+            "category": ["exact"],
+            "mandatory": ["exact"],
+            "answer_type": ["exact"],
+            "order": ["exact"],
+            "weight": ["exact"],
+        }
+
+
+class ReviewAnswerFilter(wb_filters.FilterSet):
+    question_name = wb_filters.CharFilter(label=_("Question"), lookup_expr="icontains")
+
+    class Meta:
+        model = ReviewAnswer
+        fields = {"answer_text": ["exact", "icontains"], "answered_by": ["exact"], "question": ["exact"]}
+        hidden_fields = ["question"]
+
+
+class ReviewProgressReviewFilter(wb_filters.FilterSet):
+    class Meta:
+        model = ReviewAnswer
+        fields = {}
+
+
+class RatingReviewAnswerReviewFilter(wb_filters.FilterSet):
+    deviation = wb_filters.ChoiceFilter(
+        label=_("Deviation"),
+        choices=[("EQUAL", _("Equal")), ("LESS", _("Less")), ("GREAT", _("Greater"))],
+        method="fake_filter",
+        clearable=False,
+        required=False,
+    )
+    category_question_name = wb_filters.CharFilter(label=_("Category"), lookup_expr="icontains")
+    question_name = wb_filters.CharFilter(label=_("Question"), lookup_expr="icontains")
+    answer_type = wb_filters.ChoiceFilter(
+        label="Answer Type", choices=ReviewQuestion.ANSWERTYPE.choices, required=False, hidden=True
+    )
+
+    class Meta:
+        model = ReviewAnswer
+        fields = {}

wbhuman_resources/filters/signals.py

@@ -0,0 +1,27 @@
+from django.utils.translation import gettext_lazy as _
+from wbcore import filters as wb_filters
+
+
+def add_position_filter(sender, request=None, *args, **kwargs):
+    from wbcore.contrib.directory.models import Entry
+
+    from wbhuman_resources.models import Position
+
+    def filter_position(queryset, name, value):
+        if value:
+            ids = value.get_employees().values_list("profile")
+            entries = Entry.objects.filter(id__in=ids)
+            return queryset.filter(participants__in=entries)
+        return queryset
+
+    return {
+        "position": wb_filters.ModelChoiceFilter(
+            label=_("Participant's position"),
+            field_name="position",
+            queryset=Position.objects.all(),
+            endpoint=Position.get_representation_endpoint(),
+            value_key=Position.get_representation_value_key(),
+            label_key=Position.get_representation_label_key(),
+            method=filter_position,
+        )
+    }