wbcrm 2.2.1__py2.py3-none-any.whl

wbcrm/filters/activities.py

@@ -0,0 +1,181 @@
+from datetime import date, timedelta
+
+import django_filters
+from django.contrib.auth import get_user_model
+from django.db.models import Exists, OuterRef, Q
+from django.db.models.query import QuerySet
+from django.utils.translation import gettext_lazy
+from django.utils.translation import gettext_lazy as _
+from dynamic_preferences.registries import global_preferences_registry
+from psycopg.types.range import TimestamptzRange
+from wbcore import filters as wb_filters
+from wbcore.contrib.agenda.filters import CalendarItemPeriodBaseFilterSet
+from wbcore.contrib.agenda.models import CalendarItem
+from wbcore.contrib.directory.models.entries import Company, Person
+from wbcore.filters.lookups import ALL_TEXT_LOOKUPS
+from wbcrm.models import Activity, ActivityParticipant, ActivityType, Group
+
+
+def get_employee_filter_params(request, view):
+    if employer_id := global_preferences_registry.manager()["directory__main_company"]:
+        return {"employers": employer_id}
+    return {}
+
+
+class ActivityBaseFilterSet(CalendarItemPeriodBaseFilterSet):
+    type = wb_filters.ModelMultipleChoiceFilter(
+        label=gettext_lazy("Types"),
+        queryset=ActivityType.objects.all(),
+        endpoint=ActivityType.get_representation_endpoint(),
+        value_key=ActivityType.get_representation_value_key(),
+        label_key=ActivityType.get_representation_label_key(),
+    )
+
+    visibility = wb_filters.MultipleChoiceFilter(
+        choices=CalendarItem.Visibility.choices,
+        label=gettext_lazy("Visibility"),
+    )
+
+    participants = wb_filters.ModelMultipleChoiceFilter(
+        label=gettext_lazy("Participants"),
+        queryset=Person.objects.all(),
+        endpoint=Person.get_representation_endpoint(),
+        value_key=Person.get_representation_value_key(),
+        label_key=Person.get_representation_label_key(),
+    )
+
+    companies = wb_filters.ModelMultipleChoiceFilter(
+        label=gettext_lazy("Companies"),
+        queryset=Company.objects.all(),
+        endpoint=Company.get_representation_endpoint(),
+        value_key=Company.get_representation_value_key(),
+        label_key=Company.get_representation_label_key(),
+    )
+
+    groups = wb_filters.ModelMultipleChoiceFilter(
+        label=gettext_lazy("Groups"),
+        queryset=Group.objects.all(),
+        endpoint=Group.get_representation_endpoint(),
+        value_key=Group.get_representation_value_key(),
+        label_key=Group.get_representation_label_key(),
+    )
+
+
+class ActivityFilter(ActivityBaseFilterSet):
+    clients_of = wb_filters.ModelChoiceFilter(
+        label=gettext_lazy("Clients of"),
+        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_clients_of",
+        filter_params=get_employee_filter_params,
+    )
+
+    latest_reviewer = wb_filters.ModelMultipleChoiceFilter(
+        label=gettext_lazy("Latest Reviewers"),
+        queryset=Person.objects.all(),
+        endpoint=Person.get_representation_endpoint(),
+        value_key=Person.get_representation_value_key(),
+        label_key=Person.get_representation_label_key(),
+    )
+
+    importance = wb_filters.MultipleChoiceFilter(
+        label=gettext_lazy("Importance"), choices=Activity.Importance.choices, widget=django_filters.widgets.CSVWidget
+    )
+
+    created = wb_filters.DateTimeRangeFilter(
+        method=wb_filters.DateRangeFilter.base_date_range_filter_method, label=gettext_lazy("Created")
+    )
+    edited = wb_filters.DateTimeRangeFilter(
+        method=wb_filters.DateRangeFilter.base_date_range_filter_method, label=gettext_lazy("Edited")
+    )
+
+    only_recent = wb_filters.BooleanFilter(label=gettext_lazy("Only Recent"), method="boolean_only_recent")
+
+    def boolean_only_recent(self, queryset: QuerySet[Activity], name, value: bool | None) -> QuerySet[Activity]:
+        if value:
+            today = date.today()
+            next_week = today + timedelta(days=7)
+            last_month = today - timedelta(days=30)
+            return queryset.filter(period__overlap=TimestamptzRange(last_month, next_week))
+        return queryset
+
+    def filter_clients_of(self, queryset, name, value):
+        if value:
+            user = self.request.user
+            return Activity.objects.filter(
+                (Q(participants__relationship_managers=value) | Q(companies__relationship_managers=value))
+                & (
+                    Q(visibility=CalendarItem.Visibility.PUBLIC)
+                    | (
+                        Q(visibility=CalendarItem.Visibility.CONFIDENTIAL)  # TODO move that to a queryset method
+                        & Exists(
+                            get_user_model().objects.filter(
+                                Q(id=user.id)
+                                & (
+                                    Q(groups__permissions__codename="administrate_confidential_items")
+                                    | Q(user_permissions__codename="administrate_confidential_items")
+                                )
+                            )
+                        )
+                    )
+                    | Q(
+                        Exists(
+                            ActivityParticipant.objects.filter(activity_id=OuterRef("pk"), participant=user.profile)
+                        )
+                    )
+                    | Q(assigned_to=user.profile)
+                )
+            )
+        return queryset
+
+    class Meta:
+        model = Activity
+        fields = {
+            "status": ["exact"],
+            "repeat_choice": ["exact"],
+            "title": ALL_TEXT_LOOKUPS,
+            "result": ALL_TEXT_LOOKUPS,
+            "description": ALL_TEXT_LOOKUPS,
+        }
+
+
+class ActivityChartFilter(ActivityBaseFilterSet):
+    status = wb_filters.MultipleChoiceFilter(
+        label=gettext_lazy("Status"), choices=Activity.Status.choices, widget=django_filters.widgets.CSVWidget
+    )
+
+    class Meta:
+        model = Activity
+        fields = {}
+
+
+class ActivityTypeFilter(wb_filters.FilterSet):
+    score = wb_filters.MultipleChoiceFilter(
+        label=_("Multipliers"), choices=ActivityType.Score.choices[:3], widget=django_filters.widgets.CSVWidget
+    )
+
+    class Meta:
+        model = ActivityType
+        fields = {
+            "title": ["exact", "icontains"],
+            "color": ["exact", "icontains"],
+            "default": ["exact"],
+        }
+
+
+class ActivityParticipantFilter(wb_filters.FilterSet):
+    is_occupied_filter = wb_filters.BooleanFilter(
+        label=gettext_lazy("Is occupied by different activity"), method="boolean_is_occupied"
+    )
+
+    def boolean_is_occupied(self, queryset, name, value):
+        if value is None:
+            return queryset
+        else:
+            return queryset.filter(is_occupied=value)
+
+    class Meta:
+        model = ActivityParticipant
+        fields = {}

wbcrm/filters/groups.py

@@ -0,0 +1,20 @@
+from django.utils.translation import gettext_lazy as _
+from wbcore import filters as wb_filters
+from wbcore.contrib.directory.models import Entry
+from wbcrm.models import Group
+
+
+class GroupFilter(wb_filters.FilterSet):
+    members = wb_filters.ModelMultipleChoiceFilter(
+        label=_("Members"),
+        queryset=Entry.objects.all(),
+        endpoint=Entry.get_representation_endpoint(),
+        value_key=Entry.get_representation_value_key(),
+        label_key=Entry.get_representation_label_key(),
+    )
+
+    class Meta:
+        model = Group
+        fields = {
+            "title": ["exact", "icontains"],
+        }

wbcrm/filters/products.py

@@ -0,0 +1,37 @@
+from django.utils.translation import gettext_lazy as _
+from wbcore import filters as wb_filters
+from wbcore.contrib.directory.models import Company
+from wbcrm.models import Product, ProductCompanyRelationship
+
+
+class ProductFilterSet(wb_filters.FilterSet):
+    prospects = wb_filters.ModelMultipleChoiceFilter(
+        label=_("Prospects"),
+        queryset=Company.objects.all(),
+        endpoint=Company.get_representation_endpoint(),
+        value_key=Company.get_representation_value_key(),
+        label_key=Company.get_representation_label_key(),
+    )
+
+    class Meta:
+        model = Product
+        fields = {
+            "title": ["exact", "icontains"],
+            "is_competitor": ["exact"],
+        }
+
+
+class ProductCompanyFilterSet(wb_filters.FilterSet):
+    competitor_product = wb_filters.BooleanFilter(label=_("Is Competitor"), method="filter_competitor_product")
+
+    def filter_competitor_product(self, queryset, label, value):
+        if value is None:
+            return queryset
+        else:
+            return queryset.filter(product__is_competitor=value)
+
+    class Meta:
+        model = ProductCompanyRelationship
+        fields = {
+            "product": ["exact"],
+        }

wbcrm/filters/signals.py

@@ -0,0 +1,94 @@
+from datetime import timedelta
+
+from django.db.models.query import QuerySet
+from django.dispatch import receiver
+from django.utils import timezone
+from django.utils.translation import gettext_lazy as _
+from psycopg.types.range import TimestamptzRange
+from wbcore import filters as wb_filters
+from wbcore.contrib.directory.filters.entries import (
+    CompanyFilter,
+    EntryFilter,
+    PersonFilter,
+    UserIsManagerFilter,
+)
+from wbcore.contrib.directory.models import Entry
+from wbcore.signals.filters import add_filters
+from wbcrm.models import Account, AccountRole, Product
+
+
+def choice_noactivity(queryset: QuerySet[Entry], name, value: str | None) -> QuerySet[Entry]:
+    """Returns only those entries who didn't participate in any activities during the selected timeframe"""
+
+    if value is None:
+        return queryset
+    else:
+        value = int(value)
+        if value:
+            end = timezone.now()
+            start = end - timedelta(days=value)
+            no_activity_timeframe = TimestamptzRange(start, end)
+            return queryset.exclude(
+                calendar_entities__period__overlap=no_activity_timeframe,
+                calendar_entities__item_type="wbcrm.Activity",
+            )
+        # A value of 0 corresponds to "All Time" in the filter so we need exclude any activities the entries participated in
+        return queryset.exclude(calendar_entities__item_type="wbcrm.Activity")
+
+
+@receiver(add_filters, sender=EntryFilter)
+@receiver(add_filters, sender=PersonFilter)
+@receiver(add_filters, sender=CompanyFilter)
+@receiver(add_filters, sender=UserIsManagerFilter)
+def add_account_filter(sender, request=None, *args, **kwargs):
+    def _filter_with_account(queryset: QuerySet[Entry], name, value: str | None) -> QuerySet[Entry]:
+        accounts = Account.objects.filter_for_user(request.user)
+        if value:
+            return queryset.filter(accounts__in=accounts).distinct()
+        return queryset
+
+    def _filter_with_account_role(queryset: QuerySet[Entry], name, value: str | None) -> QuerySet[Entry]:
+        roles = AccountRole.objects.filter_for_user(request.user)
+        if value:
+            return queryset.filter(account_roles__in=roles).distinct()
+        return queryset
+
+    def filter_interested_products(queryset, name, value):
+        if value:
+            return queryset.filter(interested_products__in=value)
+        return queryset
+
+    interested_products = wb_filters.ModelMultipleChoiceFilter(
+        label="Interested Products",
+        help_text="Filter by products (ours and comptetitors) that the customer is interested in",
+        queryset=Product.objects.all(),
+        endpoint=Product.get_representation_endpoint(),
+        value_key=Product.get_representation_value_key(),
+        label_key=Product.get_representation_label_key(),
+        method=filter_interested_products,
+    )
+
+    return {
+        "without_activity": wb_filters.ChoiceFilter(
+            label=_("No Activity"),
+            help_text="Filter for entries without any activities",
+            field_name="without_activity",
+            choices=[
+                (31, _("Last Month")),
+                (92, _("Last 3 Months")),
+                (182, _("Last 6 Months")),
+                (365, _("Last 12 Months")),
+                (0, _("All Time")),
+            ],
+            method=choice_noactivity,
+        ),
+        "with_account": wb_filters.BooleanFilter(
+            label="With Account", help_text="Filter for entries without an account", method=_filter_with_account
+        ),
+        "with_account_role": wb_filters.BooleanFilter(
+            label="With Account Role",
+            help_text="Filter for entries without an account role",
+            method=_filter_with_account_role,
+        ),
+        "interested_products": interested_products,
+    }