91 lines
3.1 KiB
Python
91 lines
3.1 KiB
Python
import pendulum
|
|
from copy import deepcopy
|
|
from wtforms.fields.html5 import DateField, EmailField, TelField
|
|
from wtforms.fields import RadioField, StringField
|
|
from wtforms.validators import Email, DataRequired, Optional
|
|
|
|
from .fields import SelectField
|
|
from .forms import CacheableForm
|
|
from .data import SERVICE_BRANCHES
|
|
from atst.models.user import User
|
|
|
|
from .validators import Name, DateRange, PhoneNumber
|
|
|
|
|
|
USER_FIELDS = {
|
|
"first_name": StringField("First Name", validators=[Name()]),
|
|
"last_name": StringField("Last Name", validators=[Name()]),
|
|
"email": EmailField(
|
|
"E-mail Address",
|
|
description="Enter your preferred contact e-mail address",
|
|
validators=[Email()],
|
|
),
|
|
"phone_number": TelField("Phone Number", validators=[PhoneNumber()]),
|
|
"phone_ext": StringField("Extension"),
|
|
"service_branch": SelectField(
|
|
"Service Branch or Agency",
|
|
description="Which service or organization do you belong to within the DoD?",
|
|
choices=SERVICE_BRANCHES,
|
|
),
|
|
"citizenship": RadioField(
|
|
description="What is your citizenship status?",
|
|
choices=[
|
|
("United States", "United States"),
|
|
("Foreign National", "Foreign National"),
|
|
("Other", "Other"),
|
|
],
|
|
),
|
|
"designation": RadioField(
|
|
"Designation of Person",
|
|
description="What is your designation within the DoD?",
|
|
choices=[
|
|
("military", "Military"),
|
|
("civilian", "Civilian"),
|
|
("contractor", "Contractor"),
|
|
],
|
|
),
|
|
"date_latest_training": DateField(
|
|
"Latest Information Assurance (IA) Training Completion Date",
|
|
description='To complete the training, you can find it in <a class="icon-link" href="https://iatraining.disa.mil/eta/disa_cac2018/launchPage.htm" target="_blank">Information Assurance Cyber Awareness Challange</a> website.',
|
|
validators=[
|
|
DateRange(
|
|
lower_bound=pendulum.duration(years=1),
|
|
upper_bound=pendulum.duration(days=0),
|
|
message="Must be a date within the last year.",
|
|
)
|
|
],
|
|
format="%m/%d/%Y",
|
|
),
|
|
}
|
|
|
|
|
|
def inherit_field(unbound_field, required=True):
|
|
kwargs = deepcopy(unbound_field.kwargs)
|
|
if not "validators" in kwargs:
|
|
kwargs["validators"] = []
|
|
|
|
if required:
|
|
kwargs["validators"].append(DataRequired())
|
|
else:
|
|
kwargs["validators"].append(Optional())
|
|
|
|
return unbound_field.field_class(*unbound_field.args, **kwargs)
|
|
|
|
|
|
def inherit_user_field(field_name):
|
|
required = field_name in User.REQUIRED_FIELDS
|
|
return inherit_field(USER_FIELDS[field_name], required=required)
|
|
|
|
|
|
class EditUserForm(CacheableForm):
|
|
|
|
first_name = inherit_user_field("first_name")
|
|
last_name = inherit_user_field("last_name")
|
|
email = inherit_user_field("email")
|
|
phone_number = inherit_user_field("phone_number")
|
|
phone_ext = inherit_user_field("phone_ext")
|
|
service_branch = inherit_user_field("service_branch")
|
|
citizenship = inherit_user_field("citizenship")
|
|
designation = inherit_user_field("designation")
|
|
date_latest_training = inherit_user_field("date_latest_training")
|