[ADD] HORILLA API: Add Horilla API into master code base

horilla_api/api_methods/base/methods.py

@@ -0,0 +1,66 @@
+from django.http import QueryDict
+from rest_framework.pagination import PageNumberPagination
+from employee.models import EmployeeWorkInformation
+from collections import Counter
+from django.db.models import Q
+
+
+def get_filter_url(current_url, request):
+    url_parts = current_url.split('?')
+    base_url = request.path
+    query_params = QueryDict(url_parts[1], mutable=True)
+    query_params.pop('groupby_field', None)
+    return base_url + '?' + query_params.urlencode()
+
+
+def groupby_queryset(request, url, field_name, queryset):
+    queryset_with_counts = queryset.values(
+        field_name)
+
+    counts = Counter(item[field_name] for item in queryset_with_counts)
+
+    result_list = []
+    for i in counts:
+        result_list.append({field_name: i, 'count': counts[i]})
+
+    counts_and_objects = []
+    url = get_filter_url(url, request)
+    for item in result_list:
+        count = item['count']
+        related_fields = field_name.split("__")
+        if item[field_name]:
+            related_obj = queryset.filter(
+                **{field_name: item[field_name]}).first()
+            for field in related_fields:
+                related_obj = getattr(related_obj, field)
+            counts_and_objects.append(
+                {'count': count,
+                 'name': str(related_obj),
+                 "filter_url": f"{url}&{field_name}={item[field_name]}"})
+    pagination = PageNumberPagination()
+    page = pagination.paginate_queryset(counts_and_objects, request)
+    return pagination.get_paginated_response(page)
+
+
+def permission_based_queryset(user, perm, queryset, user_obj=None):
+    if user.has_perm(perm):
+        return queryset
+
+    employee = user.employee_get
+    is_manager = EmployeeWorkInformation.objects.filter(
+        reporting_manager_id=employee
+    ).exists()
+    if is_manager:
+        if user_obj:
+            return queryset.filter(
+                Q(employee_id=employee) |
+                Q(employee_id__employee_work_info__reporting_manager_id=employee)
+            )
+        manager_filter = Q(employee_id=employee)
+        subordinates_filter = Q(
+            employee_id__employee_work_info__reporting_manager_id=employee)
+        merged_filter = manager_filter | subordinates_filter
+        merged_queryset = queryset.filter(merged_filter)
+        return merged_queryset
+
+    return queryset.filter(employee_id=employee)

horilla_api/api_methods/employee/methods.py

@@ -0,0 +1,31 @@
+from django.http import QueryDict
+from employee.models import Employee
+from rest_framework.pagination import PageNumberPagination
+from base.models import *
+from employee.models import *
+
+
+def get_next_badge_id():
+    """
+    This method is used to generate badge id
+    """
+    try:
+        highest_badge_id = Employee.objects.filter(
+            badge_id__isnull=False).order_by('-badge_id').first().badge_id
+    except AttributeError:
+        highest_badge_id = None
+
+    # Increment the badge_id if it exists, otherwise start from '1'
+    if highest_badge_id:
+        if '#' in highest_badge_id:
+            prefix, number = highest_badge_id.split(
+                '#')  # Split prefix and number
+            # Increment the number
+            new_number = str(int(number) + 1).zfill(len(number))
+            new_badge_id = f"{prefix}#{new_number}"
+        else:
+            # Add number to existing prefix
+            new_badge_id = f"{highest_badge_id}#001"
+    else:
+        new_badge_id = "EMP#001"  # Default start badge ID if no employees exist
+    return new_badge_id