82 lines
3.3 KiB
Python
82 lines
3.3 KiB
Python
from django import forms
|
|
from .models import Voter, Interaction, Donation, VoterLikelihood, InteractionType, DonationMethod, ElectionType, Event, EventParticipation
|
|
|
|
class VoterForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Voter
|
|
fields = [
|
|
'first_name', 'last_name', 'street_address', 'city', 'state', 'zip_code',
|
|
'latitude', 'longitude', 'phone', 'email',
|
|
'voter_id', 'district', 'precinct', 'registration_date',
|
|
'candidate_support', 'yard_sign'
|
|
]
|
|
widgets = {
|
|
'registration_date': forms.DateInput(attrs={'type': 'date'}),
|
|
}
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
for field in self.fields.values():
|
|
field.widget.attrs.update({'class': 'form-control'})
|
|
self.fields['candidate_support'].widget.attrs.update({'class': 'form-select'})
|
|
self.fields['yard_sign'].widget.attrs.update({'class': 'form-select'})
|
|
|
|
class InteractionForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Interaction
|
|
fields = ['type', 'date', 'description', 'notes']
|
|
widgets = {
|
|
'date': forms.DateInput(attrs={'type': 'date'}),
|
|
'notes': forms.Textarea(attrs={'rows': 2}),
|
|
}
|
|
|
|
def __init__(self, *args, tenant=None, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
if tenant:
|
|
self.fields['type'].queryset = InteractionType.objects.filter(tenant=tenant)
|
|
for field in self.fields.values():
|
|
field.widget.attrs.update({'class': 'form-control'})
|
|
self.fields['type'].widget.attrs.update({'class': 'form-select'})
|
|
|
|
class DonationForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Donation
|
|
fields = ['date', 'method', 'amount']
|
|
widgets = {
|
|
'date': forms.DateInput(attrs={'type': 'date'}),
|
|
}
|
|
|
|
def __init__(self, *args, tenant=None, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
if tenant:
|
|
self.fields['method'].queryset = DonationMethod.objects.filter(tenant=tenant)
|
|
for field in self.fields.values():
|
|
field.widget.attrs.update({'class': 'form-control'})
|
|
self.fields['method'].widget.attrs.update({'class': 'form-select'})
|
|
|
|
class VoterLikelihoodForm(forms.ModelForm):
|
|
class Meta:
|
|
model = VoterLikelihood
|
|
fields = ['election_type', 'likelihood']
|
|
|
|
def __init__(self, *args, tenant=None, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
if tenant:
|
|
self.fields['election_type'].queryset = ElectionType.objects.filter(tenant=tenant)
|
|
for field in self.fields.values():
|
|
field.widget.attrs.update({'class': 'form-control'})
|
|
self.fields['election_type'].widget.attrs.update({'class': 'form-select'})
|
|
self.fields['likelihood'].widget.attrs.update({'class': 'form-select'})
|
|
|
|
class EventParticipationForm(forms.ModelForm):
|
|
class Meta:
|
|
model = EventParticipation
|
|
fields = ['event']
|
|
|
|
def __init__(self, *args, tenant=None, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
if tenant:
|
|
self.fields['event'].queryset = Event.objects.filter(tenant=tenant)
|
|
for field in self.fields.values():
|
|
field.widget.attrs.update({'class': 'form-control'})
|
|
self.fields['event'].widget.attrs.update({'class': 'form-select'}) |