Files
ihrm/leave/forms.py

1390 lines
48 KiB
Python

"""
This module provides Horilla ModelForms for creating and managing leave-related data,
including leave type, leave request, leave allocation request, holidays and company leaves.
"""
import re
import uuid
from datetime import date, datetime
from typing import Any
from django import forms
from django.core.exceptions import ValidationError
from django.db.models import Q
from django.forms import ModelForm
from django.forms.widgets import TextInput
from django.template.loader import render_to_string
from django.utils.translation import gettext_lazy as _
from base import thread_local_middleware
from base.methods import filtersubordinatesemployeemodel, reload_queryset
from base.models import Department
from employee.filters import EmployeeFilter
from employee.forms import MultipleFileField
from employee.models import Employee
from horilla_widgets.forms import HorillaForm, HorillaModelForm
from horilla_widgets.widgets.horilla_multi_select_field import HorillaMultiSelectField
from horilla_widgets.widgets.select_widgets import HorillaMultiSelectWidget
from .methods import (
calculate_requested_days,
company_leave_dates_list,
get_leave_day_attendance,
holiday_dates_list,
leave_requested_dates,
)
from .models import (
AvailableLeave,
CompanyLeave,
CompensatoryLeaveRequest,
CompensatoryLeaverequestComment,
Holiday,
LeaveAllocationRequest,
LeaveallocationrequestComment,
LeaveRequest,
LeaverequestComment,
LeaverequestFile,
LeaveType,
RestrictLeave,
)
CHOICES = [("yes", _("Yes")), ("no", _("No"))]
class ModelForm(forms.ModelForm):
"""
Customized ModelForm class with additional functionality for field customization
based on the type of widget and setting initial values based on the current request.
"""
def __init__(self, *args, **kwargs):
"""
Initializes the ModelForm instance.
This method customizes field attributes such as CSS classes and placeholders
based on the type of widget. It also sets initial values for specific fields
based on the current request, particularly for 'employee_id' and 'company_id' fields.
"""
super().__init__(*args, **kwargs)
request = getattr(thread_local_middleware._thread_locals, "request", None)
reload_queryset(self.fields)
for field_name, field in self.fields.items():
widget = field.widget
if isinstance(widget, (forms.DateInput)):
field.widget.attrs.update({"class": "oh-input oh-calendar-input w-100"})
field.initial = date.today()
elif isinstance(
widget, (forms.NumberInput, forms.EmailInput, forms.TextInput)
):
field.widget.attrs.update(
{"class": "oh-input w-100", "placeholder": field.label}
)
elif isinstance(widget, (forms.Select,)):
field.widget.attrs.update(
{"class": "oh-select oh-select-2 select2-hidden-accessible"}
)
elif isinstance(widget, (forms.Textarea)):
field.widget.attrs.update(
{
"class": "oh-input w-100",
"placeholder": field.label,
"rows": 2,
"cols": 40,
}
)
elif isinstance(
widget,
(
forms.CheckboxInput,
forms.CheckboxSelectMultiple,
),
):
field.widget.attrs.update({"class": "oh-switch__checkbox"})
try:
self.fields["employee_id"].initial = request.user.employee_get
except:
pass
try:
self.fields["company_id"].initial = request.user.employee_get.get_company
except:
pass
class ConditionForm(forms.ModelForm):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
request = getattr(thread_local_middleware._thread_locals, "request", None)
reload_queryset(self.fields)
for field_name, field in self.fields.items():
widget = field.widget
if isinstance(widget, (forms.Select,)):
field.widget.attrs["style"] = (
"width:100%; height:50px;border: 1px solid hsl(213deg,22%,84%);border-radius: 0rem;padding: 0.8rem 1.25rem;"
)
elif isinstance(widget, (forms.DateInput)):
field.widget.attrs.update({"class": "oh-input oh-calendar-input w-100"})
field.initial = date.today()
elif isinstance(
widget, (forms.NumberInput, forms.EmailInput, forms.TextInput)
):
field.widget.attrs.update(
{"class": "oh-input w-100", "placeholder": field.label}
)
elif isinstance(widget, (forms.Textarea)):
field.widget.attrs.update(
{
"class": "oh-input w-100",
"placeholder": field.label,
"rows": 2,
"cols": 40,
}
)
elif isinstance(
widget,
(
forms.CheckboxInput,
forms.CheckboxSelectMultiple,
),
):
field.widget.attrs.update({"class": "oh-switch__checkbox"})
try:
self.fields["employee_id"].initial = request.user.employee_get
except:
pass
try:
self.fields["company_id"].initial = request.user.employee_get.get_company
except:
pass
class LeaveTypeForm(ConditionForm):
class Meta:
model = LeaveType
fields = "__all__"
exclude = ["is_active"]
labels = {
"name": _("Name"),
}
widgets = {
"color": TextInput(attrs={"type": "color", "style": "height:40px;"}),
"period_in": forms.HiddenInput(),
"total_days": forms.HiddenInput(),
}
def clean(self):
cleaned_data = super().clean()
if "employee_id" in self.errors:
del self.errors["employee_id"]
if "exceed_days" in self.errors:
del self.errors["exceed_days"]
return cleaned_data
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
class UpdateLeaveTypeForm(ConditionForm):
def __init__(self, *args, **kwargs):
super(UpdateLeaveTypeForm, self).__init__(*args, **kwargs)
empty_fields = []
for field_name, field_value in self.instance.__dict__.items():
if field_value is None or field_value == "":
if field_name.endswith("_id"):
foreign_key_field_name = re.sub("_id$", "", field_name)
empty_fields.append(foreign_key_field_name)
empty_fields.append(field_name)
for index, visible in enumerate(self.visible_fields()):
if list(self.fields.keys())[index] in empty_fields:
visible.field.widget.attrs["style"] = (
"display:none;width:100%; height:50px;border: 1px solid hsl(213deg,22%,84%);border-radius: 0rem;padding: 0.8rem 1.25rem;"
)
visible.field.widget.attrs["data-hidden"] = True
class Meta:
model = LeaveType
fields = "__all__"
exclude = ["period_in", "total_days", "is_active"]
widgets = {
"color": TextInput(attrs={"type": "color", "style": "height:40px;"}),
}
def clean(self):
cleaned_data = super().clean()
if "exceed_days" in self.errors:
del self.errors["exceed_days"]
return cleaned_data
def cal_effective_requested_days(start_date, end_date, leave_type_id, requested_days):
requested_dates = leave_requested_dates(start_date, end_date)
holidays = Holiday.objects.all()
holiday_dates = holiday_dates_list(holidays)
company_leaves = CompanyLeave.objects.all()
company_leave_dates = company_leave_dates_list(company_leaves, start_date)
if (
leave_type_id.exclude_company_leave == "yes"
and leave_type_id.exclude_holiday == "yes"
):
total_leaves = list(set(holiday_dates + company_leave_dates))
total_leave_count = sum(
requested_date in total_leaves for requested_date in requested_dates
)
requested_days = requested_days - total_leave_count
else:
holiday_count = 0
if leave_type_id.exclude_holiday == "yes":
for requested_date in requested_dates:
if requested_date in holiday_dates:
holiday_count += 1
requested_days = requested_days - holiday_count
if leave_type_id.exclude_company_leave == "yes":
company_leave_count = sum(
requested_date in company_leave_dates
for requested_date in requested_dates
)
requested_days = requested_days - company_leave_count
return requested_days
class LeaveRequestCreationForm(ModelForm):
start_date = forms.DateField(widget=forms.DateInput(attrs={"type": "date"}))
end_date = forms.DateField(widget=forms.DateInput(attrs={"type": "date"}))
def clean(self):
cleaned_data = super().clean()
start_date = cleaned_data.get("start_date")
end_date = cleaned_data.get("end_date")
employee_id = cleaned_data.get("employee_id")
leave_type_id = cleaned_data.get("leave_type_id")
start_date_breakdown = cleaned_data.get("start_date_breakdown")
end_date_breakdown = cleaned_data.get("end_date_breakdown")
attachment = cleaned_data.get("attachment")
overlapping_requests = LeaveRequest.objects.filter(
employee_id=employee_id, start_date__lte=end_date, end_date__gte=start_date
).exclude(
id=self.instance.id,
)
if leave_type_id.require_attachment == "yes":
if attachment is None:
raise forms.ValidationError(
{
"attachment": _(
"An attachment is required for this leave request"
)
}
)
if not start_date <= end_date:
raise forms.ValidationError(
_("End date should not be less than start date.")
)
if start_date == end_date:
if start_date_breakdown != end_date_breakdown:
raise forms.ValidationError(
_(
"There is a mismatch in the breakdown of the start date and end date."
)
)
if not AvailableLeave.objects.filter(
employee_id=employee_id, leave_type_id=leave_type_id
).exists():
raise forms.ValidationError(_("Employee has no leave type.."))
if overlapping_requests.exclude(status__in=["cancelled", "rejected"]).exists():
raise forms.ValidationError(
_("Employee has already a leave request for this date range..")
)
available_leave = AvailableLeave.objects.get(
employee_id=employee_id, leave_type_id=leave_type_id
)
total_leave_days = (
available_leave.available_days + available_leave.carryforward_days
)
requested_days = calculate_requested_days(
start_date, end_date, start_date_breakdown, end_date_breakdown
)
effective_requested_days = cal_effective_requested_days(
start_date=start_date,
end_date=end_date,
leave_type_id=leave_type_id,
requested_days=requested_days,
)
leave_dates = leave_requested_dates(start_date, end_date)
month_year = [f"{date.year}-{date.strftime('%m')}" for date in leave_dates]
today = datetime.today()
unique_dates = list(set(month_year))
if f"{today.month}-{today.year}" in unique_dates:
unique_dates.remove(f"{today.strftime('%m')}-{today.year}")
forcasted_leaves = available_leave.forcasted_leaves()
if leave_type_id.reset_based == "monthly":
if f"{today.year}-{today.strftime('%m')}" not in unique_dates:
for item in unique_dates:
total_leave_days += forcasted_leaves[item]
if not effective_requested_days <= total_leave_days:
raise forms.ValidationError(_("Employee doesn't have enough leave days.."))
return cleaned_data
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.fields["leave_type_id"].widget.attrs.update(
{
"onchange": "empleavetypeChange($(this))",
}
)
self.fields["employee_id"].widget.attrs.update(
{
"hx-target": "#id_leave_type_id_parent_div",
"hx-trigger": "change",
"hx-get": "/leave/get-employee-leave-types",
}
)
self.fields["start_date"].widget.attrs.update(
{
"onchange": "dateChange($(this))",
}
)
def as_p(self, *args, **kwargs):
"""
Render the form fields as HTML table rows with Bootstrap styling.
"""
context = {"form": self}
table_html = render_to_string("attendance_form.html", context)
return table_html
class Meta:
model = LeaveRequest
fields = [
"employee_id",
"leave_type_id",
"start_date",
"start_date_breakdown",
"end_date",
"end_date_breakdown",
"attachment",
"description",
]
class LeaveRequestUpdationForm(ModelForm):
start_date = forms.DateField(widget=forms.DateInput(attrs={"type": "date"}))
end_date = forms.DateField(widget=forms.DateInput(attrs={"type": "date"}))
def clean(self):
cleaned_data = super().clean()
start_date = cleaned_data.get("start_date")
end_date = cleaned_data.get("end_date")
employee_id = cleaned_data.get("employee_id")
leave_type_id = cleaned_data.get("leave_type_id")
start_date_breakdown = cleaned_data.get("start_date_breakdown")
end_date_breakdown = cleaned_data.get("end_date_breakdown")
overlapping_requests = LeaveRequest.objects.filter(
employee_id=employee_id, start_date__lte=end_date, end_date__gte=start_date
).exclude(id=self.instance.id)
if not start_date <= end_date:
raise forms.ValidationError(
_("End date should not be less than start date.")
)
if start_date == end_date:
if start_date_breakdown != end_date_breakdown:
raise forms.ValidationError(
_(
"There is a mismatch in the breakdown of the start date and end date."
)
)
if not AvailableLeave.objects.filter(
employee_id=employee_id, leave_type_id=leave_type_id
).exists():
raise forms.ValidationError(_("Employee has no leave type.."))
if overlapping_requests.exclude(status__in=["cancelled", "rejected"]).exists():
raise forms.ValidationError(
_("Employee has already a leave request for this date range..")
)
available_leave = AvailableLeave.objects.get(
employee_id=employee_id, leave_type_id=leave_type_id
)
total_leave_days = (
available_leave.available_days + available_leave.carryforward_days
)
requested_days = calculate_requested_days(
start_date, end_date, start_date_breakdown, end_date_breakdown
)
effective_requested_days = cal_effective_requested_days(
start_date=start_date,
end_date=end_date,
leave_type_id=leave_type_id,
requested_days=requested_days,
)
leave_dates = leave_requested_dates(start_date, end_date)
month_year = [f"{date.year}-{date.strftime('%m')}" for date in leave_dates]
today = datetime.today()
unique_dates = list(set(month_year))
if f"{today.month}-{today.year}" in unique_dates:
unique_dates.remove(f"{today.strftime('%m')}-{today.year}")
forcasted_leaves = available_leave.forcasted_leaves()
if leave_type_id.reset_based == "monthly":
if f"{today.year}-{today.strftime('%m')}" not in unique_dates:
for item in unique_dates:
total_leave_days += forcasted_leaves[item]
if not effective_requested_days <= total_leave_days:
raise forms.ValidationError(_("Employee doesn't have enough leave days.."))
return cleaned_data
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.fields["leave_type_id"].widget.attrs.update(
{
"onchange": "empleavetypeChange($(this))",
}
)
self.fields["employee_id"].widget.attrs.update(
{
"hx-target": "#id_leave_type_id_parent_div",
"hx-trigger": "change",
"hx-get": "/leave/get-employee-leave-types",
}
)
self.fields["start_date"].widget.attrs.update(
{
"onchange": "dateChange($(this))",
}
)
def as_p(self, *args, **kwargs):
"""
Render the form fields as HTML table rows with Bootstrap styling.
"""
context = {"form": self}
table_html = render_to_string("attendance_form.html", context)
return table_html
class Meta:
model = LeaveRequest
fields = [
"leave_type_id",
"employee_id",
"start_date",
"start_date_breakdown",
"end_date",
"end_date_breakdown",
"attachment",
"description",
]
class AvailableLeaveForm(ModelForm):
"""
Form for managing available leave data.
This form allows users to manage available leave data by specifying details such as
the leave type and employee.
Attributes:
- leave_type_id: A ModelChoiceField representing the leave type associated with the available leave.
- employee_id: A ModelChoiceField representing the employee associated with the available leave.
"""
leave_type_id = forms.ModelChoiceField(
queryset=LeaveType.objects.all(),
widget=forms.SelectMultiple,
empty_label=None,
)
employee_id = forms.ModelChoiceField(
queryset=Employee.objects.all(),
widget=forms.SelectMultiple,
empty_label=None,
)
class Meta:
model = AvailableLeave
fields = ["leave_type_id", "employee_id", "is_active"]
class HolidayForm(ModelForm):
"""
Form for creating or updating a holiday.
This form allows users to create or update holiday data by specifying details such as
the start date and end date.
Attributes:
- start_date: A DateField representing the start date of the holiday.
- end_date: A DateField representing the end date of the holiday.
"""
start_date = forms.DateField(
widget=forms.DateInput(attrs={"type": "date"}),
)
end_date = forms.DateField(
widget=forms.DateInput(attrs={"type": "date"}),
)
def clean_end_date(self):
start_date = self.cleaned_data.get("start_date")
end_date = self.cleaned_data.get("end_date")
if start_date and end_date and end_date < start_date:
raise ValidationError(
_("End date should not be earlier than the start date.")
)
return end_date
class Meta:
"""
Meta class for additional options
"""
model = Holiday
fields = "__all__"
exclude = ["is_active"]
labels = {
"name": _("Name"),
}
def __init__(self, *args, **kwargs):
super(HolidayForm, self).__init__(*args, **kwargs)
self.fields["name"].widget.attrs["autocomplete"] = "name"
class LeaveOneAssignForm(HorillaModelForm):
"""
Form for assigning available leave to employees.
This form allows administrators to assign available leave to a single employee
by specifying the employee and setting the is_active flag.
Attributes:
- employee_id: A HorillaMultiSelectField representing the employee to assign leave to.
"""
employee_id = HorillaMultiSelectField(
queryset=Employee.objects.all(),
widget=HorillaMultiSelectWidget(
filter_route_name="employee-widget-filter",
filter_class=EmployeeFilter,
filter_instance_contex_name="f",
filter_template_path="employee_filters.html",
required=True,
),
label="Employee",
)
class Meta:
"""
Meta class for additional options
"""
model = AvailableLeave
fields = ["employee_id", "is_active"]
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
reload_queryset(self.fields)
class AvailableLeaveUpdateForm(ModelForm):
"""
Form for updating available leave data.
This form allows users to update available leave data by modifying fields such as
available_days, carryforward_days, and is_active.
Attributes:
- Meta: Inner class defining metadata options.
- model: The model associated with the form (AvailableLeave).
- fields: A list of fields to include in the form.
"""
class Meta:
"""
Meta class for additional options
"""
model = AvailableLeave
fields = ["available_days", "carryforward_days", "is_active"]
class CompanyLeaveForm(ModelForm):
"""
Form for managing company leave data.
This form allows users to manage company leave data by including all fields from
the CompanyLeave model except for is_active.
Attributes:
- Meta: Inner class defining metadata options.
- model: The model associated with the form (CompanyLeave).
- fields: A special value indicating all fields should be included in the form.
- exclude: A list of fields to exclude from the form (is_active).
"""
class Meta:
"""
Meta class for additional options
"""
model = CompanyLeave
fields = "__all__"
exclude = ["is_active"]
class UserLeaveRequestForm(ModelForm):
start_date = forms.DateField(widget=forms.DateInput(attrs={"type": "date"}))
end_date = forms.DateField(widget=forms.DateInput(attrs={"type": "date"}))
description = forms.CharField(label=_("Description"), widget=forms.Textarea)
def clean(self):
cleaned_data = super().clean()
start_date = cleaned_data.get("start_date")
end_date = cleaned_data.get("end_date")
employee_id = cleaned_data.get("employee_id")
start_date_breakdown = cleaned_data.get("start_date_breakdown")
end_date_breakdown = cleaned_data.get("end_date_breakdown")
leave_type_id = cleaned_data.get("leave_type_id")
overlapping_requests = LeaveRequest.objects.filter(
employee_id=employee_id, start_date__lte=end_date, end_date__gte=start_date
).exclude(id=self.instance.id)
assigned_leave_types = AvailableLeave.objects.filter(employee_id=employee_id)
if not assigned_leave_types:
raise forms.ValidationError(
_("You dont have enough leave days to update the request.")
)
if leave_type_id.require_attachment == "yes":
attachment = cleaned_data.get("attachment")
if attachment is None:
raise forms.ValidationError(
{
"attachment": _(
"An attachment is required for this leave request"
)
}
)
if start_date == end_date:
if start_date_breakdown != end_date_breakdown:
raise forms.ValidationError(
_(
"There is a mismatch in the breakdown of the start date and end date."
)
)
if not start_date <= end_date:
raise forms.ValidationError(
_("End date should not be less than start date.")
)
if overlapping_requests.exclude(status__in=["cancelled", "rejected"]).exists():
raise forms.ValidationError(
_("Employee has already a leave request for this date range.....")
)
requested_days = calculate_requested_days(
start_date, end_date, start_date_breakdown, end_date_breakdown
)
available_leave = AvailableLeave.objects.get(
employee_id=employee_id, leave_type_id=leave_type_id
)
total_leave_days = (
available_leave.available_days + available_leave.carryforward_days
)
effective_requested_days = cal_effective_requested_days(
start_date=start_date,
end_date=end_date,
leave_type_id=leave_type_id,
requested_days=requested_days,
)
if not effective_requested_days <= total_leave_days:
raise forms.ValidationError(_("Employee doesn't have enough leave days.."))
return cleaned_data
def __init__(self, *args, **kwargs):
leave_type = kwargs.pop("initial", None)
employee = kwargs.pop("employee", None)
super(UserLeaveRequestForm, self).__init__(*args, **kwargs)
if employee:
available_leaves = employee.available_leave.all()
assigned_leave_types = LeaveType.objects.filter(
id__in=available_leaves.values_list("leave_type_id", flat=True)
)
self.fields["leave_type_id"].queryset = assigned_leave_types
if leave_type:
self.fields["leave_type_id"].queryset = LeaveType.objects.filter(
id=leave_type["leave_type_id"].id
)
self.fields["leave_type_id"].initial = leave_type["leave_type_id"].id
self.fields["leave_type_id"].empty_label = None
def as_p(self, *args, **kwargs):
"""
Render the form fields as HTML table rows with Bootstrap styling.
"""
context = {"form": self}
table_html = render_to_string("attendance_form.html", context)
return table_html
class Meta:
"""
Meta class for additional options
"""
model = LeaveRequest
fields = [
"employee_id",
"leave_type_id",
"start_date",
"start_date_breakdown",
"end_date",
"end_date_breakdown",
"attachment",
"description",
]
widgets = {
"employee_id": forms.HiddenInput(),
}
excluded_fields = [
"id",
"approved_available_days",
"approved_carryforward_days",
"created_at",
"attachment",
]
class AvailableLeaveColumnExportForm(forms.Form):
"""
Form for selecting columns to export in available leave data.
This form allows users to select specific columns from the AvailableLeave model
for export. The available columns are dynamically generated based on the
model's meta information, excluding specified excluded_fields.
Attributes:
- model_fields: A list of fields in the AvailableLeave model.
- field_choices: A list of field choices for the form, consisting of field names
and their verbose names, excluding specified excluded_fields.
- selected_fields: A MultipleChoiceField representing the selected columns
to be exported.
"""
model_fields = AvailableLeave._meta.get_fields()
field_choices = [
(field.name, field.verbose_name)
for field in model_fields
if hasattr(field, "verbose_name") and field.name not in excluded_fields
]
selected_fields = forms.MultipleChoiceField(
choices=field_choices,
widget=forms.CheckboxSelectMultiple,
initial=[
"employee_id",
"leave_type_id",
"available_days",
"carryforward_days",
"total_leave_days",
],
)
class HolidaysColumnExportForm(forms.Form):
"""
Form for selecting columns to export in holiday data.
This form allows users to select specific columns from the Holiday model
for export. The available columns are dynamically generated based on the
model's meta information, excluding specified excluded_fields.
Attributes:
- model_fields: A list of fields in the Holiday model.
- field_choices: A list of field choices for the form, consisting of field names
and their verbose names, excluding specified excluded_fields.
- selected_fields: A MultipleChoiceField representing the selected columns
to be exported.
"""
model_fields = Holiday._meta.get_fields()
field_choices = [
(field.name, field.verbose_name)
for field in model_fields
if hasattr(field, "verbose_name") and field.name not in excluded_fields
]
selected_fields = forms.MultipleChoiceField(
choices=field_choices,
widget=forms.CheckboxSelectMultiple,
initial=[
"name",
"start_date",
"end_date",
"recurring",
],
)
class RejectForm(forms.Form):
"""
Form for rejecting a leave request.
This form allows administrators to provide a rejection reason when rejecting
a leave request.
Attributes:
- reason: A CharField representing the reason for rejecting the leave request.
"""
reason = forms.CharField(
label=_("Rejection Reason"),
widget=forms.Textarea(attrs={"rows": 4, "class": "p-4 oh-input w-100"}),
)
class Meta:
"""
Meta class for additional options
"""
model = LeaveRequest
fields = ["reject_reason"]
class UserLeaveRequestCreationForm(ModelForm):
start_date = forms.DateField(widget=forms.DateInput(attrs={"type": "date"}))
end_date = forms.DateField(widget=forms.DateInput(attrs={"type": "date"}))
def as_p(self, *args, **kwargs):
"""
Render the form fields as HTML table rows with Bootstrap styling.
"""
context = {"form": self}
table_html = render_to_string("attendance_form.html", context)
return table_html
def __init__(self, *args, **kwargs):
employee = kwargs.pop("employee", None)
super().__init__(*args, **kwargs)
if employee:
available_leaves = employee.available_leave.all()
assigned_leave_types = LeaveType.objects.filter(
id__in=available_leaves.values_list("leave_type_id", flat=True)
)
self.fields["leave_type_id"].queryset = assigned_leave_types
self.fields["leave_type_id"].widget.attrs.update(
{
"onchange": "typeChange($(this))",
}
)
self.fields["employee_id"].initial = employee
def clean(self):
cleaned_data = super().clean()
start_date = cleaned_data.get("start_date")
end_date = cleaned_data.get("end_date")
employee_id = cleaned_data.get("employee_id")
leave_type_id = cleaned_data.get("leave_type_id")
start_date_breakdown = cleaned_data.get("start_date_breakdown")
end_date_breakdown = cleaned_data.get("end_date_breakdown")
overlapping_requests = LeaveRequest.objects.filter(
employee_id=employee_id, start_date__lte=end_date, end_date__gte=start_date
).exclude(id=self.instance.id)
if not start_date <= end_date:
raise forms.ValidationError(
_("End date should not be less than start date.")
)
if leave_type_id.require_attachment == "yes":
attachment = cleaned_data.get("attachment")
if attachment is None:
raise forms.ValidationError(
{
"attachment": _(
"An attachment is required for this leave request"
)
}
)
if start_date == end_date:
if start_date_breakdown != end_date_breakdown:
raise forms.ValidationError(
_(
"There is a mismatch in the breakdown of the start date and end date."
)
)
if not AvailableLeave.objects.filter(
employee_id=employee_id, leave_type_id=leave_type_id
).exists():
raise forms.ValidationError(_("Employee has no leave type.."))
if overlapping_requests.exclude(status__in=["cancelled", "rejected"]).exists():
raise forms.ValidationError(
_("Employee has already a leave request for this date range..")
)
available_leave = AvailableLeave.objects.get(
employee_id=employee_id, leave_type_id=leave_type_id
)
total_leave_days = (
available_leave.available_days + available_leave.carryforward_days
)
requested_days = (end_date - start_date).days + 1
cleaned_data["requested_days"] = requested_days
if not requested_days <= total_leave_days:
raise forms.ValidationError(_("Employee doesn't have enough leave days.."))
return cleaned_data
class Meta:
"""
Meta class for additional options
"""
model = LeaveRequest
fields = [
"leave_type_id",
"employee_id",
"start_date",
"start_date_breakdown",
"end_date",
"end_date_breakdown",
"attachment",
"description",
"requested_days",
]
widgets = {
"employee_id": forms.HiddenInput(),
"requested_days": forms.HiddenInput(),
}
class LeaveAllocationRequestForm(ModelForm):
"""
Form for creating a leave allocation request.
This form allows users to create a leave allocation request by specifying
details such as leave type, employee, requested days, description, and attachment.
Methods:
- as_p: Render the form fields as HTML table rows with Bootstrap styling.
"""
def as_p(self, *args, **kwargs):
"""
Render the form fields as HTML table rows with Bootstrap styling.
"""
context = {"form": self}
table_html = render_to_string("attendance_form.html", context)
return table_html
class Meta:
"""
Meta class for additional options
"""
model = LeaveAllocationRequest
fields = [
"leave_type_id",
"employee_id",
"requested_days",
"description",
"attachment",
]
class LeaveAllocationRequestRejectForm(forms.Form):
"""
Form for rejecting a leave allocation request.
This form allows administrators to provide a rejection reason when rejecting
a leave allocation request.
Attributes:
- reason: A CharField representing the reason for rejecting the leave allocation request.
"""
reason = forms.CharField(
label=_("Rejection Reason"),
widget=forms.Textarea(attrs={"rows": 4, "class": "p-4 oh-input w-100"}),
)
class Meta:
model = LeaveAllocationRequest
fields = ["reject_reason"]
class LeaveRequestExportForm(forms.Form):
"""
Form for selecting fields to export in a leave request export.
This form allows users to select specific fields from the LeaveRequest model
for export. The available fields are dynamically generated based on the
model's meta information, excluding certain fields specified in 'excluded_fields'.
Attributes:
- model_fields: A list of fields in the LeaveRequest model.
- field_choices: A list of field choices for the form, consisting of field names
and their verbose names, excluding specified excluded_fields.
- selected_fields: A MultipleChoiceField representing the selected fields
to be exported.
"""
model_fields = LeaveRequest._meta.get_fields()
field_choices = [
(field.name, field.verbose_name)
for field in model_fields
if hasattr(field, "verbose_name") and field.name not in excluded_fields
]
selected_fields = forms.MultipleChoiceField(
choices=field_choices,
widget=forms.CheckboxSelectMultiple,
initial=[
"employee_id",
"leave_type_Assignid",
"start_date",
"start_date_breakdown",
"end_date",
"end_date_breakdown",
"requested_days",
"description",
"status",
],
)
class AssignLeaveForm(HorillaForm):
"""
Form for Payslip
"""
leave_type_id = forms.ModelChoiceField(
queryset=LeaveType.objects.all(),
widget=forms.SelectMultiple(
attrs={"class": "oh-select oh-select-2 mb-2", "required": True}
),
empty_label=None,
label="Leave Type",
required=False,
)
employee_id = HorillaMultiSelectField(
queryset=Employee.objects.all(),
widget=HorillaMultiSelectWidget(
filter_route_name="employee-widget-filter",
filter_class=EmployeeFilter,
filter_instance_contex_name="f",
filter_template_path="employee_filters.html",
required=True,
),
label="Employee",
)
def clean(self):
cleaned_data = super().clean()
employee_id = cleaned_data.get("employee_id")
leave_type_id = cleaned_data.get("leave_type_id")
if not employee_id:
raise forms.ValidationError({"employee_id": "This field is required"})
if not leave_type_id:
raise forms.ValidationError({"leave_type_id": "This field is required"})
return cleaned_data
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
reload_queryset(self.fields)
self.fields["employee_id"].widget.attrs.update(
{"required": True, "id": uuid.uuid4()}
),
self.fields["leave_type_id"].label = "Leave Type"
class LeaverequestcommentForm(ModelForm):
"""
LeaverequestComment form
"""
class Meta:
"""
Meta class for additional options
"""
model = LeaverequestComment
fields = ("comment",)
class LeaveCommentForm(ModelForm):
"""
Leave request comment model form
"""
verbose_name = "Add Comment"
class Meta:
"""
Meta class for additional options
"""
model = LeaverequestComment
fields = "__all__"
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.fields["files"] = MultipleFileField(label="files")
self.fields["files"].required = False
def as_p(self):
"""
Render the form fields as HTML table rows with Bootstrap styling.
"""
context = {"form": self}
table_html = render_to_string("common_form.html", context)
return table_html
def save(self, commit: bool = ...) -> Any:
multiple_files_ids = []
files = None
if self.files.getlist("files"):
files = self.files.getlist("files")
self.instance.attachemnt = files[0]
multiple_files_ids = []
for attachemnt in files:
file_instance = LeaverequestFile()
file_instance.file = attachemnt
file_instance.save()
multiple_files_ids.append(file_instance.pk)
instance = super().save(commit)
if commit:
instance.files.add(*multiple_files_ids)
return instance, files
class LeaveallocationrequestcommentForm(ModelForm):
"""
Leave Allocation Requestcomment form
"""
class Meta:
"""
Meta class for additional options
"""
model = LeaveallocationrequestComment
fields = ("comment",)
class LeaveAllocationCommentForm(ModelForm):
"""
Leave request comment model form
"""
verbose_name = "Add Comment"
class Meta:
"""
Meta class for additional options
"""
model = LeaveallocationrequestComment
fields = "__all__"
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.fields["files"] = MultipleFileField(label="files")
self.fields["files"].required = False
def as_p(self):
"""
Render the form fields as HTML table rows with Bootstrap styling.
"""
context = {"form": self}
table_html = render_to_string("common_form.html", context)
return table_html
def save(self, commit: bool = ...) -> Any:
multiple_files_ids = []
files = None
if self.files.getlist("files"):
files = self.files.getlist("files")
self.instance.attachemnt = files[0]
multiple_files_ids = []
for attachemnt in files:
file_instance = LeaverequestFile()
file_instance.file = attachemnt
file_instance.save()
multiple_files_ids.append(file_instance.pk)
instance = super().save(commit)
if commit:
instance.files.add(*multiple_files_ids)
return instance, files
class RestrictLeaveForm(ModelForm):
start_date = forms.DateField(
widget=forms.DateInput(attrs={"type": "date"}),
)
end_date = forms.DateField(
widget=forms.DateInput(attrs={"type": "date"}),
)
def clean_end_date(self):
start_date = self.cleaned_data.get("start_date")
end_date = self.cleaned_data.get("end_date")
if start_date and end_date and end_date < start_date:
raise ValidationError(
_("End date should not be earlier than the start date.")
)
return end_date
class Meta:
model = RestrictLeave
fields = "__all__"
exclude = ["is_active"]
labels = {
"title": _("Title"),
}
def __init__(self, *args, **kwargs):
super(RestrictLeaveForm, self).__init__(*args, **kwargs)
self.fields["title"].widget.attrs["autocomplete"] = "title"
class CompensatoryLeaveForm(ModelForm):
"""
Form for creating a leave allocation request.
This form allows users to create a leave allocation request by specifying
details such as leave type, employee, requested days, description, and attachment.
Methods:
- as_p: Render the form fields as HTML table rows with Bootstrap styling.
"""
class Meta:
"""
Meta class for additional options
"""
attendance_id = forms.MultipleChoiceField(required=True)
model = CompensatoryLeaveRequest
fields = [
# "leave_type_id",
"employee_id",
"attendance_id",
# "requested_days",
"description",
]
def __init__(self, *args, **kwargs):
from attendance.models import Attendance
super(CompensatoryLeaveForm, self).__init__(*args, **kwargs)
request = getattr(thread_local_middleware._thread_locals, "request", None)
instance_id = None
if self.instance:
instance_id = self.instance.id
if (
request
and hasattr(request, "user")
and hasattr(request.user, "employee_get")
):
employee = request.user.employee_get
holiday_attendance = get_leave_day_attendance(employee, comp_id=instance_id)
# Get a list of tuples containing (id, attendance_date)
attendance_dates = list(
holiday_attendance.values_list("id", "attendance_date")
)
# Set the queryset of attendance_id to the attendance_dates
self.fields["attendance_id"].choices = attendance_dates
queryset = (
filtersubordinatesemployeemodel(
request, Employee.objects.filter(is_active=True)
)
| Employee.objects.filter(employee_user_id=request.user)
).distinct()
self.fields["employee_id"].queryset = queryset
self.fields["employee_id"].widget.attrs.update(
{
"hx-target": "#id_attendance_id_parent_div",
"hx-trigger": "change",
"hx-get": "/leave/get-leave-attendance-dates",
}
)
def as_p(self, *args, **kwargs):
"""
Render the form fields as HTML table rows with Bootstrap styling.
"""
context = {"form": self}
table_html = render_to_string("attendance_form.html", context)
return table_html
def clean(self):
cleaned_data = super().clean()
attendance_id = cleaned_data.get("attendance_id")
employee = cleaned_data.get("employee_id")
attendance_repeat = False
instance_id = None
if self.instance:
instance_id = self.instance.id
for attendance in attendance_id:
if (
CompensatoryLeaveRequest.objects.filter(
employee_id=employee, attendance_id=attendance
)
.exclude(Q(id=instance_id) | Q(status="rejected"))
.exists()
):
attendance_repeat = True
break
if attendance_repeat:
raise forms.ValidationError(
{
"attendance_id": "This attendance is already converted to complimentory leave"
}
)
return cleaned_data
class CompensatoryLeaveRequestRejectForm(forms.Form):
"""
Form for rejecting a compensatory leave request.
This form allows administrators to provide a rejection reason when rejecting
a compensatory leave request.
Attributes:
- reason: A CharField representing the reason for rejecting the compensatory leave request.
"""
reason = forms.CharField(
label=_("Rejection Reason"),
widget=forms.Textarea(attrs={"rows": 4, "class": "p-4 oh-input w-100"}),
)
class Meta:
model = CompensatoryLeaveRequest
fields = ["reject_reason"]
class CompensatoryLeaveRequestcommentForm(ModelForm):
"""
LeaverequestComment form
"""
class Meta:
"""
Meta class for additional options
"""
model = CompensatoryLeaverequestComment
fields = ("comment",)