128 lines
4.7 KiB
Python
128 lines
4.7 KiB
Python
"""
|
|
horilla_automations/forms.py
|
|
"""
|
|
|
|
from typing import Any
|
|
|
|
from django import forms
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from base.forms import ModelForm
|
|
from employee.filters import EmployeeFilter
|
|
from employee.models import Employee
|
|
from horilla_automations.methods.methods import generate_choices
|
|
from horilla_automations.models import MODEL_CHOICES, MailAutomation
|
|
from horilla_widgets.forms import default_select_option_template
|
|
from horilla_widgets.widgets.horilla_multi_select_field import HorillaMultiSelectField
|
|
from horilla_widgets.widgets.select_widgets import HorillaMultiSelectWidget
|
|
|
|
|
|
class AutomationForm(ModelForm):
|
|
"""
|
|
AutomationForm
|
|
"""
|
|
|
|
condition_html = forms.CharField(widget=forms.HiddenInput())
|
|
condition_querystring = forms.CharField(widget=forms.HiddenInput())
|
|
|
|
cols = {"template_attachments": 12}
|
|
|
|
class Meta:
|
|
model = MailAutomation
|
|
fields = "__all__"
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
# --- Field: also_sent_to ---
|
|
self.fields["also_sent_to"] = HorillaMultiSelectField(
|
|
queryset=Employee.objects.all(),
|
|
required=False,
|
|
widget=HorillaMultiSelectWidget(
|
|
filter_route_name="employee-widget-filter",
|
|
filter_class=EmployeeFilter,
|
|
filter_instance_contex_name="f",
|
|
filter_template_path="employee_filters.html",
|
|
instance=self.instance,
|
|
),
|
|
label="Also Sent to",
|
|
help_text=_("The employees selected here will receive the email as Cc."),
|
|
)
|
|
|
|
# --- Determine model for generate_choices ---
|
|
model = getattr(self.instance, "model", None) or self.data.get("model")
|
|
mail_to, mail_details_choice = [], []
|
|
|
|
if model:
|
|
choices = generate_choices(model)
|
|
mail_to, mail_details_choice = choices[0], choices[1]
|
|
|
|
# --- Field: mail_to ---
|
|
self.fields["mail_to"] = forms.MultipleChoiceField(
|
|
choices=mail_to,
|
|
initial=self.data.get("mail_to"),
|
|
widget=forms.SelectMultiple(attrs={"class": "oh-select oh-select-2 w-100"}),
|
|
)
|
|
|
|
# --- Field: mail_details ---
|
|
self.fields["mail_details"] = forms.ChoiceField(
|
|
choices=mail_details_choice,
|
|
help_text=_(
|
|
"Fill mail template details (receiver/instance, `self` will be the person who triggers the automation)"
|
|
),
|
|
)
|
|
self.fields["mail_details"].widget.attrs = {
|
|
"class": "oh-select oh-select-2 w-100"
|
|
}
|
|
|
|
# --- Field: model ---
|
|
self.fields["model"].choices = [("", "Select model")] + sorted(
|
|
set(MODEL_CHOICES)
|
|
)
|
|
self.fields["model"].widget.attrs["onchange"] = "getToMail($(this))"
|
|
|
|
# --- Field: mail_template ---
|
|
self.fields["mail_template"].empty_label = "----------"
|
|
|
|
# --- Field: condition fields ---
|
|
self.fields["condition"].initial = getattr(
|
|
self.instance, "condition", None
|
|
) or self.data.get("condition")
|
|
self.fields["condition_html"].initial = getattr(
|
|
self.instance, "condition_html", None
|
|
) or self.data.get("condition_html")
|
|
self.fields["condition_querystring"].initial = getattr(
|
|
self.instance, "condition", None
|
|
) or self.data.get("condition_html")
|
|
|
|
# --- Apply option template name for all select fields ---
|
|
for field in self.fields.values():
|
|
if isinstance(field.widget, forms.Select):
|
|
field.widget.option_template_name = default_select_option_template
|
|
|
|
# --- Re-insert is_active field to ensure order ---
|
|
self.fields["is_active"] = self.fields.pop("is_active")
|
|
|
|
def clean(self):
|
|
cleaned_data = super().clean()
|
|
if isinstance(self.fields["also_sent_to"], HorillaMultiSelectField):
|
|
self.errors.pop("also_sent_to", None)
|
|
|
|
employee_data = self.fields["also_sent_to"].queryset.filter(
|
|
id__in=self.data.getlist("also_sent_to")
|
|
)
|
|
cleaned_data["also_sent_to"] = employee_data
|
|
|
|
return cleaned_data
|
|
|
|
def save(self, commit: bool = ...) -> Any:
|
|
self.instance: MailAutomation = self.instance
|
|
condition_querystring = self.cleaned_data["condition_querystring"]
|
|
condition_html = self.cleaned_data["condition_html"]
|
|
mail_to = self.data.getlist("mail_to")
|
|
self.instance.mail_to = str(mail_to)
|
|
self.instance.mail_details = self.data["mail_details"]
|
|
self.instance.condition_querystring = condition_querystring
|
|
self.instance.condition_html = condition_html
|
|
return super().save(commit)
|