Source code for intranet.apps.preferences.forms

import logging

from django import forms
from django.contrib.auth import get_user_model

from ..bus.models import Route
from ..users.models import Email, Grade, Phone, Website

logger = logging.getLogger(__name__)


[docs]class BusRouteForm(forms.Form): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.BUS_ROUTE_CHOICES = [(None, "Set bus route...")] routes = Route.objects.all().order_by("route_name") for route in routes: self.BUS_ROUTE_CHOICES += [(route.route_name, route.route_name)] self.fields["bus_route"] = forms.ChoiceField(choices=self.BUS_ROUTE_CHOICES, widget=forms.Select, required=False)
[docs]class PreferredPictureForm(forms.Form): def __init__(self, user, *args, **kwargs): super().__init__(*args, **kwargs) self.PREFERRED_PICTURE_CHOICES = [("AUTO", "Auto-select the most recent photo")] for i in range(4): try: grade = Grade.names[i] user.photos.get(grade_number=(i + 9)) # Only display option if the photo exists self.PREFERRED_PICTURE_CHOICES += [(i + 9, grade.title() + " Photo")] except Exception: pass self.fields["preferred_photo"] = forms.ChoiceField(choices=self.PREFERRED_PICTURE_CHOICES, widget=forms.RadioSelect(), required=True)
[docs]class PrivacyOptionsForm(forms.Form): def __init__(self, user, *args, **kwargs): super().__init__(*args, **kwargs) def flag(label, default): return forms.BooleanField(initial=default, label=label, required=False) self.fields["show_address"] = flag(None, False) self.fields["show_address-self"] = flag("Show Address", False) self.fields["show_telephone"] = flag(None, False) self.fields["show_telephone-self"] = flag("Show Phone", False) self.fields["show_birthday"] = flag(None, False) self.fields["show_birthday-self"] = flag("Show Birthday", False) pictures_label = "Show Pictures" if user.is_student: pictures_label += " on Import" self.fields["show_pictures"] = flag(None, False) self.fields["show_pictures-self"] = flag(pictures_label, False) # photos = user.photo_permissions["self"] # for i in range(4): # grade = Grade.names[i] # if photos[grade] is not None: # self.fields["photoperm-{}".format(grade)] = flag(None, False) # self.fields["photoperm-{}-self".format(grade)] = flag("Show {} Photo".format(grade.capitalize()), False) self.fields["show_eighth"] = flag(None, False) self.fields["show_eighth-self"] = flag("Show Eighth Period Schedule", False) self.fields["show_schedule"] = flag(None, False) self.fields["show_schedule-self"] = flag("Show Class Schedule", False) if not user.has_admin_permission("preferences"): for name in self.fields: if not name.endswith("-self"): self.fields[name].widget.attrs["class"] = "disabled"
[docs]class NotificationOptionsForm(forms.Form): def __init__(self, user, *args, **kwargs): super().__init__(*args, **kwargs) def flag(label, default): return forms.BooleanField(initial=default, label=label, required=False) self.fields["receive_news_emails"] = flag("Receive News Emails", False) self.fields["receive_eighth_emails"] = flag("Receive Eighth Period Emails", False) label = "Primary Email" if user.emails.all().count() == 0: label = "You can set a primary email after adding emails below." self.fields["primary_email"] = forms.ModelChoiceField( queryset=Email.objects.filter(user=user), required=False, label=label, disabled=(user.emails.all().count() == 0) )
[docs]class DarkModeForm(forms.Form): def __init__(self, user, *args, **kwargs): super().__init__(*args, **kwargs) self.fields["dark_mode_enabled"] = forms.BooleanField( initial=user.dark_mode_properties.dark_mode_enabled, label="Enable dark mode?", required=False )
[docs]class PhoneForm(forms.ModelForm): """Represents a phone number (number + purpose)""" _number = forms.CharField(max_length=14)
[docs] class Meta: model = Phone fields = ["purpose", "_number"]
[docs]class EmailForm(forms.ModelForm):
[docs] def clean_address(self): data = self.cleaned_data["address"] if data.lower().strip().endswith("@fcpsschools.net"): raise forms.ValidationError("You cannot provide a fcpsschools.net address.", code="invalid") return data
[docs] class Meta: model = Email fields = ["address"]
[docs]class WebsiteForm(forms.ModelForm):
[docs] class Meta: model = Website fields = ["url"]
PhoneFormset = forms.inlineformset_factory(get_user_model(), Phone, form=PhoneForm, extra=1) EmailFormset = forms.inlineformset_factory(get_user_model(), Email, form=EmailForm, extra=1) WebsiteFormset = forms.inlineformset_factory(get_user_model(), Website, form=WebsiteForm, extra=1)