def test_select_date_widget():
"""Test the SelectDateWidget."""
assert SelectDateWidget.FORMAT_CHOICES['%d'] == [(x, str(x))
for x in range(1, 32)]
assert SelectDateWidget.FORMAT_CHOICES['%m'] == [(x, str(x))
for x in range(1, 13)]
assert SelectDateWidget.FORMAT_CLASSES == {
'%d': 'select_date_day',
'%m': 'select_date_month',
'%Y': 'select_date_year'
}
select_date_widget = SelectDateWidget(years=[0, 1])
assert select_date_widget.FORMAT_CHOICES['%Y'] == [(0, '0'), (1, '1')]
class Field(object):
id = 'world'
name = 'helloWorld'
format = '%d %m %Y'
data = None
html = select_date_widget(field=Field())
assert 'world' in html
assert 'helloWorld' in html
assert 'class="select_date_day"' in html
assert 'class="select_date_month"' in html
assert 'class="select_date_year"' in html
class ChangeUserDetailsForm(Form):
# TODO: Better birthday field
birthday = DateField(_("Your Birthday"), format="%d %m %Y",
widget=SelectDateWidget(), validators=[Optional()])
gender = SelectField(_("Gender"), default="None", choices=[
("None", ""),
("Male", _("Male")),
("Female", _("Female"))])
location = StringField(_("Location"), validators=[
Optional()])
website = StringField(_("Website"), validators=[
Optional(), URL()])
avatar = StringField(_("Avatar"), validators=[
Optional(), URL()])
signature = TextAreaField(_("Forum Signature"), validators=[
Optional()])
notes = TextAreaField(_("Notes"), validators=[
Optional(), Length(min=0, max=5000)])
submit = SubmitField(_("Save"))
class UserForm(Form):
username = TextField("Username", validators=[
Required(),
is_username])
email = TextField("E-Mail", validators=[
Required(),
Email(message="This email is invalid")])
password = PasswordField("Password", validators=[
Optional()])
birthday = DateField("Birthday", format="%d %m %Y",
widget=SelectDateWidget(),
validators=[Optional()])
gender = SelectField("Gender", default="None", choices=[
("None", ""),
("Male", "Male"),
("Female", "Female")])
location = TextField("Location", validators=[
Optional()])
website = TextField("Website", validators=[
Optional(), URL()])
avatar = TextField("Avatar", validators=[
Optional(), URL()])
signature = TextAreaField("Forum Signature", validators=[
Optional(), Length(min=0, max=250)])
notes = TextAreaField("Notes", validators=[
Optional(), Length(min=0, max=5000)])
primary_group = QuerySelectField(
"Primary Group",
query_factory=select_primary_group,
get_label="name")
secondary_groups = QuerySelectMultipleField(
"Secondary Groups",
# TODO: Template rendering errors "NoneType is not callable"
# without this, figure out why.
query_factory=select_primary_group,
allow_blank=True,
get_label="name")
def validate_username(self, field):
if hasattr(self, "user"):
user = User.query.filter(
db.and_(User.username.like(field.data),
db.not_(User.id == self.user.id)
)
).first()
else:
user = User.query.filter(User.username.like(field.data)).first()
if user:
raise ValidationError("This username is taken")
def validate_email(self, field):
if hasattr(self, "user"):
user = User.query.filter(
db.and_(User.email.like(field.data),
db.not_(User.id == self.user.id)
)
).first()
else:
user = User.query.filter(User.email.like(field.data)).first()
if user:
raise ValidationError("This email is taken")
def save(self):
user = User(**self.data)
return user.save()