SimpleTimeTracker/app/forms.py

from django import forms
from django.core.validators import RegexValidator, MaxLengthValidator, MinLengthValidator, ValidationError
from . models import Project, Setting
from calendar import monthrange

numeric = RegexValidator(r'^[0-9+]', 'Only numeric characters.')
time_max_length = MaxLengthValidator(4, 'Length limit exceeds 4 characters.')
max_daily_hours_length = MaxLengthValidator(2, 'Only 2 place values allowed.')
session_timeout_length = MaxLengthValidator(4, 'Only 4 place values allowed.')
min_pin_length = MinLengthValidator(4, 'Must be at least 4 characters long.')
max_pin_length = MaxLengthValidator(4, 'May not be longer then 4 characters long.')


def pin_blacklist(value):
    blacklist = ['1234', '4321', '0000', '1111',
                 '2222', '3333', '4444', '5555',
                 '6666', '7777', '8888', '9999',
                 '2345', '5432', '3456', '6543',
                 '4567', '7654', '5678', '8765',
                 '6789', '9876', '7890', '0987']
    if value in blacklist:
        raise ValidationError("Please provide a more complex PIN")
    else:
        return value


hours = []
try:
    max_daily_hours = Setting.objects.get(setting='Max Daily Hours')
    for hour in range(0, int(max_daily_hours.value)+1):
        hours.append(("%i" % hour, "%i" % hour))
except Exception as e:
    print(e)
    for hour in range(0, 9):
        hours.append(("%i" % hour, "%i" % hour))

days = (
    ('1', '1'),
    ('2', '2'),
    ('3', '3'),
    ('4', '4'),
    ('5', '5'),
    ('6', '6'),
    ('7', '7'),
    ('8', '8'),
    ('9', '9'),
    ('10', '10'),
    ('11', '11'),
    ('12', '12'),
    ('13', '13'),
    ('14', '14'),
    ('15', '15'),
    ('16', '16'),
    ('17', '17'),
    ('18', '18'),
    ('19', '19'),
    ('20', '20'),
    ('21', '21'),
    ('22', '22'),
    ('23', '23'),
    ('24', '24'),
    ('25', '25'),
    ('26', '26'),
    ('27', '27'),
    ('28', '28'),
    ('29', '29'),
    ('30', '30'),
    ('31', '31')
)

minutes = (
    ('0', '00'),
    ('0.25', '15'),
    ('0.50', '30'),
    ('0.75', '45')
)


class LoginForm(forms.Form):
    pin = forms.CharField(strip=True, widget=forms.PasswordInput(attrs={
        'class': 'form-control form-control-lg p-4 text-center',
        'id': 'pin',
        'placeholder': 'PIN',
    }), validators=[numeric])


class CreateUserForm(forms.Form):
    first_name = forms.CharField(strip=True, required=True)
    last_name = forms.CharField(strip=True, required=True)
    pin = forms.CharField(strip=True, required=True, validators=[numeric, min_pin_length, max_pin_length, pin_blacklist],
                          help_text="4 character numeric PIN")


class TimeEntryForm(forms.Form):
    day_of_month = forms.ChoiceField(choices=days, widget=forms.Select(attrs={'class': 'form-control form-control-lg'}))
    project = forms.ModelChoiceField(Project.objects.all(), required=False,
                                     label="Select a project if applicable (not required)",
                                     widget=forms.Select(attrs={'class': 'form-control form-control-lg'}))

    hours = forms.ChoiceField(required=True, choices=hours,
                              widget=forms.Select(attrs={'class': 'form-control form-control-lg'}))
    minutes = forms.ChoiceField(required=True, choices=minutes,
                                widget=forms.Select(attrs={'class': 'form-control form-control-lg'}))


class SettingsForm(forms.Form):
    max_daily_hours = forms.CharField(
        required=True,
        validators=[max_daily_hours_length],
        help_text="Sets the maximum hour value for the Hours select box."
    )

    max_daily_entries = forms.CharField(
        required=True,
        help_text="Sets the maximum number of time entries for a given calendar day."
    )

    session_timeout = forms.CharField(
        required=True,
        validators=[session_timeout_length],
        label="Session Timeout (Minutes)",
        help_text="Sets the automatic logout time limit in minutes"
    )

    projects = forms.BooleanField(required=False, label="Enable Projects",
                                  help_text="Allows for time entries to be attached to projects.")