Use StringField.process_formdata() if valuelist is empty.

This commit is contained in:
Berker Peksag 2015-08-25 03:44:02 +03:00
parent 75dc254c65
commit 2a1082e3dc

View File

@ -27,10 +27,12 @@ from mediagoblin.auth.tools import normalize_user_or_email_field
class WebsiteField(wtforms.StringField): class WebsiteField(wtforms.StringField):
"""A field that expects a website URL but adds http:// if not provided.""" """A field that expects a website URL but adds http:// if not provided."""
def process_formdata(self, valuelist): def process_formdata(self, valuelist):
data = valuelist[0] if valuelist:
if not data.startswith((u'http://', u'https://')): data = valuelist[0]
data = u'http://' + data if not data.startswith((u'http://', u'https://')):
self.data = data self.data = u'http://' + data
else:
super(WebsiteField, self).process_formdata(valuelist)
class EditForm(wtforms.Form): class EditForm(wtforms.Form):