from django.apps import apps
from django.urls import reverse
from django.utils.html import format_html, format_html_join
from django.utils.translation import gettext_lazy as _
from horilla_views.cbv_methods import render_template
class DummyModel:
"""A dummy fallback class that behaves like a model placeholder."""
def __init__(self, *args, **kwargs):
pass
def __getattr__(self, name):
# if you try to access any attribute, just return another dummy
return DummyModel()
def __call__(self, *args, **kwargs):
# acts callable (like a model manager/queryset)
return DummyModel()
def get_horilla_model_class(app_label, model):
try:
return apps.get_model(app_label, model)
except LookupError:
return DummyModel
Recruitment = get_horilla_model_class(app_label="recruitment", model="recruitment")
AttendanceLateComeEarlyOut = get_horilla_model_class(
app_label="attendance", model="attendancelatecomeearlyOut"
)
Attendance = get_horilla_model_class(app_label="attendance", model="attendance")
LeaveRequest = get_horilla_model_class(app_label="leave", model="leaverequest")
Meetings = get_horilla_model_class(app_label="pms", model="Meetings")
ResignationLetter = get_horilla_model_class(
app_label="offboarding", model="ResignationLetter"
)
AssetAssignment = get_horilla_model_class(app_label="asset", model="AssetAssignment")
TimeSheet = get_horilla_model_class(app_label="project", model="TimeSheet")
def tot_hires(self):
"""
This method for get custom column for Total hires.
"""
col = f"""
{self.total_hires()} Hired of {self.candidate.all().count()} Candidates
"""
return col
def managers_detail(self):
"""
manager in detail view
"""
employees = self.recruitment_managers.all()
employee_names_string = ""
if employees:
employee_names_string = ",
".join(
[str(employee) for employee in employees]
)
return employee_names_string
def open_job_detail(self):
"""
open jobs in detail view
"""
jobs = self.open_positions.all()
jobs_names_string = ""
if jobs:
jobs_names_string = ",
".join([str(job) for job in jobs])
return jobs_names_string
def penalities_column(self):
"""
Returns an HTML snippet showing penalty status with Tailwind styling.
"""
penalties_count = self.get_penalties_count()
if penalties_count:
url = reverse("view-penalties") + f"?late_early_id={self.id}"
return format_html(
''
"Penalties :{}
",
url,
penalties_count,
)
else:
return format_html(
''
"No Penalties
"
)
def rejected_action(self):
if self.reject_reason:
if self.status == "cancelled":
label = _("Reason for Cancellation")
else:
label = _("Reason for Rejection")
return format_html(
"""
""",
label,
self.reject_reason,
)
return ""
def attachment_action(self):
if self.attachment:
return format_html(
"""
{}
""",
self.attachment.url,
_("View attachment"),
)
return ""
def attendance_detail_activity_col(self):
activity_count = self.activities().get("count", 0)
if activity_count == 0:
return ""
label = _("Activities:")
view_label = _("View Activities")
count_label = _("Activity") if activity_count == 1 else _("Activities")
url = reverse("get-attendance-activities", args=[self.id])
col = format_html(
"""
""",
label,
url,
view_label,
activity_count,
count_label,
)
return col
def leave_clash_col(self):
count = self.leave_clashes_count
label = _("View Clashes")
url = reverse("view-clashes", args=[self.id])
col = format_html(
"""
""",
url,
label,
count,
)
return col
def mom_detail_col(self):
return render_template(
path="cbv/meetings/mom_detail_col.html",
context={"instance": self},
)
def detail_description_col(self):
return render_template(
path="cbv/exit_process/detail_page_description.html",
context={"instance": self},
)
def assign_condition_img(self):
images = self.assign_images.all()
if not images:
return ""
label = _("Assign Condition Images")
links_html = format_html_join(
"",
"""
""",
((doc.image.url, doc.image.url) for doc in images),
)
col = format_html(
"""
""",
label,
links_html,
)
return col
def return_condition_img(self):
images = self.return_images.all()
if not images:
return ""
label = _("Return Condition Images")
links_html = format_html_join(
"",
"""
""",
((doc.image.url, doc.image.url) for doc in images),
)
col = format_html(
"""
""",
label,
links_html,
)
return col
def detail_view_subtitle(self):
"""
for subtitle in detail view
"""
col = format_html(
"""
Date
{}
Time Spent
{}
Project
{}
Task
{}
""",
self.date,
self.time_spent,
self.project_id,
self.task_id,
)
return col
def detail_view_title(self):
col = format_html(
"""
""",
self.employee_id.get_avatar(),
self.employee_id.get_full_name(),
)
return col
Recruitment.managers_detail = managers_detail
Recruitment.open_job_detail = open_job_detail
Recruitment.tot_hires = tot_hires
AttendanceLateComeEarlyOut.penalities_column = penalities_column
Attendance.attendance_detail_activity_col = attendance_detail_activity_col
LeaveRequest.rejected_action = rejected_action
LeaveRequest.attachment_action = attachment_action
LeaveRequest.penality_col = penalities_column
LeaveRequest.leave_clash_col = leave_clash_col
Meetings.mom_detail_col = mom_detail_col
ResignationLetter.detail_description_col = detail_description_col
AssetAssignment.assign_condition_img = assign_condition_img
AssetAssignment.return_condition_img = return_condition_img
TimeSheet.detail_view_subtitle = detail_view_subtitle
TimeSheet.detail_view_title = detail_view_title
from base.cbv.dashboard.dashboard import DashboardWorkTypeRequest, ShiftRequestToApprove
_shift_request_to_approve_init_orig = ShiftRequestToApprove.__init__
def _shift_request_to_approve_init(self, **kwargs):
_shift_request_to_approve_init_orig(self, **kwargs)
self.header_attrs = {}
_work_type_request_to_approve_init_orig = DashboardWorkTypeRequest.__init__
def _work_type_request_to_approve_init(self, **kwargs):
_work_type_request_to_approve_init_orig(self, **kwargs)
self.header_attrs = {}
ShiftRequestToApprove.__init__ = _shift_request_to_approve_init
DashboardWorkTypeRequest.__init__ = _work_type_request_to_approve_init
if apps.is_installed("pms"):
from pms.cbv.meetings import MeetingsDetailedView
_meeting_detailed_init_orig = MeetingsDetailedView.__init__
def _meeting_detailed_init(self, **kwargs):
_meeting_detailed_init_orig(self, **kwargs)
self.body = [
(_("Date"), "date"),
(_("Question Template"), "question_template"),
(_("Employees"), "employ_detail_col"),
(_("Managers"), "manager_detail_col"),
(_("Answerable employees"), "answerable_col"),
(_("Minutes of Meeting"), "mom_detail_col", True),
]
MeetingsDetailedView.__init__ = _meeting_detailed_init
if apps.is_installed("offboarding"):
from offboarding.cbv.resignation import ResignationLetterDetailView
_resignation_detailed_init_orig = ResignationLetterDetailView.__init__
def _resignation_detailed_init(self, **kwargs):
_resignation_detailed_init_orig(self, **kwargs)
self.body = [
(_("Planned To Leave"), "planned_to_leave_on"),
(_("Status"), "get_status"),
(_("Description"), "detail_description_col", True),
]
self.cols = {
"detail_description_col": 12,
}
ResignationLetterDetailView.__init__ = _resignation_detailed_init
if apps.is_installed("project"):
from project.cbv.timesheet import (
TimeSheetCardView,
TimeSheetDetailView,
TimeSheetList,
TimeSheetNavView,
)
_timesheet_nav_init_orig = TimeSheetNavView.__init__
def _timesheet_nav_init(self, **kwargs):
_timesheet_nav_init_orig(self, **kwargs)
url = f"{reverse('personal-time-sheet-view',kwargs={'emp_id': self.request.user.employee_get.id})}"
self.view_types = [
{
"type": "list",
"icon": "list-outline",
"url": reverse("time-sheet-list"),
},
{
"type": "card",
"icon": "grid-outline",
"url": reverse("time-sheet-card"),
},
{
"type": "graph",
"icon": "bar-chart",
"url": url,
},
]
_timesheet_list_init_orig = TimeSheetList.__init__
def _timesheet_list_init(self, **kwargs):
_timesheet_list_init_orig(self, **kwargs)
self.row_attrs = """
hx-get='{detail_view}?instance_ids={ordered_ids}'
hx-target="#objectDetailsModalTarget"
data-target="#objectDetailsModal"
data-toggle="oh-modal-toggle"
"""
_timesheet_card_init_orig = TimeSheetCardView.__init__
def _timesheet_card_init(self, **kwargs):
_timesheet_card_init_orig(self, **kwargs)
self.card_attrs = """
hx-get='{detail_view}?instance_ids={ordered_ids}'
hx-target="#objectDetailsModalTarget"
data-target="#objectDetailsModal"
data-toggle="oh-modal-toggle"
"""
self.details = {
"title": "{detail_view_title}",
"subtitle": "{detail_view_subtitle}",
}
_timesheet_detail_init_orig = TimeSheetDetailView.__init__
def _timesheet_detail_init(self, **kwargs):
_timesheet_detail_init_orig(self, **kwargs)
self.cols = {
"description": 12,
}
TimeSheetNavView.__init__ = _timesheet_nav_init
TimeSheetList.__init__ = _timesheet_list_init
TimeSheetCardView.__init__ = _timesheet_card_init
TimeSheetDetailView.__init__ = _timesheet_detail_init
if apps.is_installed("attendance"):
from attendance.cbv.dashboard import (
DashboardaAttendanceOT,
DashboardAttendanceToValidate,
)
_overtime_attendance_init_orig = DashboardaAttendanceOT.__init__
def _overtime_attendance_init(self, **kwargs):
_overtime_attendance_init_orig(self, **kwargs)
self.header_attrs = {}
_validate_attendance_init_orig = DashboardAttendanceToValidate.__init__
def _validate_attendance_init(self, **kwargs):
_validate_attendance_init_orig(self, **kwargs)
self.header_attrs = {}
DashboardaAttendanceOT.__init__ = _overtime_attendance_init
DashboardAttendanceToValidate.__init__ = _validate_attendance_init
if apps.is_installed("leave"):
from django.contrib.auth.decorators import login_required
from django.http import HttpResponse
from django.shortcuts import get_object_or_404
from django.template.loader import render_to_string
from employee.models import Employee
from horilla.decorators import hx_request_required, manager_can_enter
from leave import views as leave_views
from leave.cbv.dashboard import LeaveRequestsToApprove
from leave.cbv.leave_requests import LeaveRequestFormView
from leave.cbv.my_leave_request import MyLeaveRequestForm
from leave.forms import (
LeaveRequestCreationForm,
LeaveRequestUpdationForm,
UserLeaveRequestCreationForm,
)
from leave.models import LeaveType
_leave_request_to_approve_init_orig = LeaveRequestsToApprove.__init__
def _leave_request_to_approve_init(self, **kwargs):
_leave_request_to_approve_init_orig(self, **kwargs)
self.header_attrs = {}
LeaveRequestsToApprove.__init__ = _leave_request_to_approve_init
_my_leave_request_get_context_data_orig = MyLeaveRequestForm.get_context_data
def _ensure_leave_type_field(field):
if not field:
return
_merge_widget_css(field.widget, {"oh-select", "oh-select-2", "w-100"})
_set_field_placeholder(field)
queryset = getattr(field, "queryset", None)
if not _has_queryset_records(queryset):
fallback_queryset = LeaveType.objects.all()
if _has_queryset_records(fallback_queryset):
field.queryset = fallback_queryset
def _ensure_employee_field(field):
if not field:
return
_merge_widget_css(field.widget, {"oh-select", "oh-select-2", "w-100"})
_set_field_placeholder(field)
def _my_leave_request_get_context_data(self, **kwargs):
context = _my_leave_request_get_context_data_orig(self, **kwargs)
leave_field = self.form.fields.get("leave_type_id")
_ensure_leave_type_field(leave_field)
return context
MyLeaveRequestForm.get_context_data = _my_leave_request_get_context_data
_leave_request_form_view_get_context_data_orig = (
LeaveRequestFormView.get_context_data
)
def _leave_request_form_view_get_context_data(self, **kwargs):
context = _leave_request_form_view_get_context_data_orig(self, **kwargs)
leave_field = self.form.fields.get("leave_type_id")
employee_field = self.form.fields.get("employee_id")
_ensure_leave_type_field(leave_field)
_ensure_employee_field(employee_field)
return context
LeaveRequestFormView.get_context_data = _leave_request_form_view_get_context_data
_leave_request_creation_form_init_orig = LeaveRequestCreationForm.__init__
def _leave_request_creation_form_init(self, *args, **kwargs):
_leave_request_creation_form_init_orig(self, *args, **kwargs)
leave_field = self.fields.get("leave_type_id")
employee_field = self.fields.get("employee_id")
_ensure_leave_type_field(leave_field)
_ensure_employee_field(employee_field)
LeaveRequestCreationForm.__init__ = _leave_request_creation_form_init
_leave_request_updation_form_init_orig = LeaveRequestUpdationForm.__init__
def _leave_request_updation_form_init(self, *args, **kwargs):
_leave_request_updation_form_init_orig(self, *args, **kwargs)
leave_field = self.fields.get("leave_type_id")
employee_field = self.fields.get("employee_id")
_ensure_leave_type_field(leave_field)
_ensure_employee_field(employee_field)
LeaveRequestUpdationForm.__init__ = _leave_request_updation_form_init
_user_leave_request_creation_form_init_orig = (
UserLeaveRequestCreationForm.__init__
)
def _user_leave_request_creation_form_init(self, *args, **kwargs):
_user_leave_request_creation_form_init_orig(self, *args, **kwargs)
leave_field = self.fields.get("leave_type_id")
_ensure_leave_type_field(leave_field)
UserLeaveRequestCreationForm.__init__ = _user_leave_request_creation_form_init
@login_required
@hx_request_required
@manager_can_enter("leave.add_leaverequest")
def _patched_get_employee_leave_types(request):
employee_id = request.GET.get("employee_id")
form_name = request.GET.get("form")
if form_name == "LeaveRequestUpdationForm":
form = LeaveRequestUpdationForm()
else:
form = LeaveRequestCreationForm()
if employee_id:
employee = get_object_or_404(Employee, id=employee_id)
assigned_leave_types = LeaveType.objects.filter(
id__in=employee.available_leave.values_list("leave_type_id", flat=True)
)
form.fields["leave_type_id"].queryset = assigned_leave_types
else:
form.fields["leave_type_id"].queryset = LeaveType.objects.none()
_ensure_leave_type_field(form.fields.get("leave_type_id"))
leave_type_field_html = render_to_string(
"leave/leave_request/leave_type_field.html",
{
"form": form,
"field_name": "leave_type_id",
"field": form.fields["leave_type_id"],
},
)
return HttpResponse(leave_type_field_html)
leave_views.get_employee_leave_types = _patched_get_employee_leave_types
if apps.is_installed("payroll"):
from payroll.forms.component_forms import GeneratePayslipForm, PayslipForm
from payroll.models.models import Contract
def _merge_widget_css(widget, required_classes):
existing_classes = set(
filter(None, widget.attrs.get("class", "").split())
)
widget.attrs["class"] = " ".join(
sorted(existing_classes.union(required_classes))
)
def _has_queryset_records(queryset):
if queryset is None:
return False
try:
return queryset.exists()
except Exception:
return False
def _get_active_contracts():
contracts = Contract.objects.filter(contract_status="active").select_related(
"employee_id"
)
if _has_queryset_records(contracts):
return list(contracts)
entire = getattr(Contract.objects, "entire", None)
if callable(entire):
contracts = (
entire()
.filter(contract_status="active")
.select_related("employee_id")
)
if _has_queryset_records(contracts):
return list(contracts)
return []
def _set_field_placeholder(field):
placeholder = field.widget.attrs.get("data-placeholder")
if not placeholder:
label = field.label or _("Select option")
field.widget.attrs["data-placeholder"] = str(label)
def _refresh_payslip_employee_field(field):
if field is None:
return
raw_choices = list(field.choices)
has_real_choice = any(
choice_value not in ("", None)
for choice_value, _ in raw_choices
)
if has_real_choice:
return
queryset = getattr(field, "queryset", None)
if queryset is None:
return
fallback_queryset = None
contracts = _get_active_contracts()
if contracts:
employee_ids = []
seen = set()
for contract in contracts:
employee = getattr(contract, "employee_id", None)
if not employee or not getattr(employee, "is_active", False):
continue
if employee.pk in seen:
continue
seen.add(employee.pk)
employee_ids.append(employee.pk)
if employee_ids:
manager = getattr(queryset.model, "objects", None)
if manager is not None:
entire = getattr(manager, "entire", None)
if callable(entire):
fallback_queryset = entire().filter(pk__in=employee_ids)
else:
fallback_queryset = manager.filter(pk__in=employee_ids)
if fallback_queryset is None:
fallback_queryset = queryset.model._default_manager.filter(
pk__in=employee_ids
)
if fallback_queryset is None:
fallback_queryset = queryset
fallback_queryset = fallback_queryset.distinct()
if not _has_queryset_records(fallback_queryset):
return
field.queryset = fallback_queryset
choices = []
empty_label = getattr(field, "empty_label", None)
if empty_label is not None:
choices.append(("", empty_label))
for employee in fallback_queryset:
choices.append((employee.pk, field.label_from_instance(employee)))
field.choices = choices
_payslip_form_init_orig = PayslipForm.__init__
def _payslip_form_init(self, *args, **kwargs):
_payslip_form_init_orig(self, *args, **kwargs)
employee_field = self.fields.get("employee_id")
if not employee_field:
return
_merge_widget_css(
employee_field.widget, {"oh-select", "oh-select-2", "w-100"}
)
_set_field_placeholder(employee_field)
_refresh_payslip_employee_field(employee_field)
PayslipForm.__init__ = _payslip_form_init
_generate_payslip_form_init_orig = GeneratePayslipForm.__init__
def _generate_payslip_form_init(self, *args, **kwargs):
_generate_payslip_form_init_orig(self, *args, **kwargs)
employee_field = self.fields.get("employee_id")
if not employee_field:
return
_merge_widget_css(
employee_field.widget, {"oh-select", "oh-select-2", "w-100"}
)
_set_field_placeholder(employee_field)
queryset = getattr(employee_field, "queryset", None)
if queryset is None:
return
if _has_queryset_records(queryset):
return
manager = getattr(queryset.model, "objects", None)
fallback_queryset = None
if manager is not None:
entire = getattr(manager, "entire", None)
if callable(entire):
fallback_queryset = entire().filter(
is_active=True,
contract_set__isnull=False,
contract_set__contract_status="active",
)
else:
fallback_queryset = manager.filter(
is_active=True,
contract_set__isnull=False,
contract_set__contract_status="active",
)
if fallback_queryset is None:
fallback_queryset = (
queryset.model._default_manager.filter(
is_active=True,
contract_set__isnull=False,
contract_set__contract_status="active",
)
)
if fallback_queryset is None:
employee_manager = getattr(Employee, "objects", None)
if employee_manager is not None:
entire = getattr(employee_manager, "entire", None)
if callable(entire):
fallback_queryset = entire().filter(is_active=True)
else:
fallback_queryset = employee_manager.filter(is_active=True)
fallback_queryset = fallback_queryset.distinct()
if not _has_queryset_records(fallback_queryset):
employee_manager = getattr(Employee, "objects", None)
if employee_manager is not None:
entire = getattr(employee_manager, "entire", None)
if callable(entire):
fallback_queryset = entire().filter(is_active=True)
else:
fallback_queryset = employee_manager.filter(is_active=True)
if fallback_queryset is None:
employee_field.queryset = Employee.objects.none()
else:
employee_field.queryset = fallback_queryset.distinct()
if hasattr(employee_field, "_choices"):
employee_field._choices = None
GeneratePayslipForm.__init__ = _generate_payslip_form_init