aboutsummaryrefslogtreecommitdiffstats
path: root/ctrack/organisations/forms.py
blob: 2609a6b8350d361c1675cdde1bf684e66b2ea73d (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
from crispy_forms.helper import FormHelper
from crispy_forms.layout import Layout, Fieldset, ButtonHolder, Submit, Button, Field
from django import forms
from django.forms import inlineformset_factory
from django.urls import reverse

from ctrack.organisations.models import Organisation, Address


class OrganisationCreateForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        cancel_redirect = reverse("organisations:list")
        self.helper = FormHelper(self)
        self.helper.layout = Layout(
            Fieldset(
                "",
                Field("name", css_class="form-control-lg"),
                "submode",
                "oes",
                "designation_type",
                "registered_company_name",
                "registered_company_number",
                "updated_by",
                "comments",
                "active"
            ),
            ButtonHolder(
                Submit("submit", "Submit", css_class="btn-primary"),
                Button("cancel", "Cancel", onclick=f"location.href='{cancel_redirect}';", css_class="btn-danger")
            )
        )

    class Meta:
        model = Organisation
        fields = ["name", "submode", "oes", "designation_type",
                  "registered_company_name", "registered_company_number",
                  "updated_by", "comments", "active"]
        labels = {
            "oes": "OES"
        }
        help_texts = {
            "submode": "e.g. Rail Maintenance, TOC, etc...",
            "updated_by": "Name of staff member/inspector creating this record",
            "active": "Is this company an active participant in the NIS compliance regime?",
            "designation_type": "This is probably defined in the Reguation",
        }


class AddressCreateForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        cancel_redirect = reverse("organisations:list")
        self.helper = FormHelper(self)
        self.helper.layout = Layout(
            Fieldset(
                "Add an address",
                "type",
                "line1",
                "line2",
                "line3",
                "city",
                "county",
                "postcode",
                "country",
                "other_details"
            )
        )

    class Meta:
        model = Address
        fields = ('type', 'line1', 'line2', 'line3', 'city', 'county', 'postcode',
                  'country', 'other_details')

    # def __init__(self, *args, **kwargs):
    #     self.org = kwargs.pop("org")
    #     super().__init__(*args, **kwargs)
    #
    # def save(self):
    #     address = super().save(commit=False)
    #     address.organisation = self.org
    #     address.organisation.save()
    #     return address


AddressInlineForm = inlineformset_factory(Organisation, Address, exclude=(), can_delete=False,
                                          form=AddressCreateForm, extra=1)