67 lines
2.8 KiB
Python
67 lines
2.8 KiB
Python
from iso3166 import countries
|
||
import string
|
||
import random
|
||
from ..models import User, Role
|
||
|
||
from flask_wtf import FlaskForm, RecaptchaField
|
||
from wtforms import StringField, PasswordField, BooleanField, SubmitField, SelectField, DecimalField
|
||
from wtforms import validators, ValidationError
|
||
from wtforms.fields.html5 import EmailField
|
||
|
||
class EditProfileForm(FlaskForm):
|
||
name = StringField('Name:', [validators.DataRequired(), validators.Length(3, 60)])
|
||
address = StringField('Address:', [validators.DataRequired(), validators.Length(2, 50)])
|
||
city = StringField('City:', [validators.DataRequired(), validators.Length(2,40)])
|
||
|
||
postcode = StringField('Postcode:')
|
||
|
||
clist = []
|
||
for c in countries:
|
||
clist.append((c.alpha2, c.name))
|
||
country = SelectField('Country:', choices=clist, default='BG')
|
||
|
||
phone = StringField('Phone:')
|
||
org_account = BooleanField('This is a business account.')
|
||
org_companyname = StringField('Company Name:')
|
||
org_regaddress = StringField('Company Address:')
|
||
org_responsible = StringField('Accountable Person:')
|
||
org_vatnum = StringField('VAT Number:')
|
||
twofactor = BooleanField('Enable 2-factor authentication')
|
||
submit = SubmitField('Update')
|
||
|
||
class EditProfileAdminForm(FlaskForm):
|
||
email = StringField('Електроннa поща (логин):', [validators.DataRequired(), validators.Length(1, 64), validators.Email()])
|
||
confirmed = BooleanField('Activated')
|
||
role = SelectField('Role', coerce=int)
|
||
|
||
name = StringField('Лице за контакт:', [validators.DataRequired(), validators.Length(3, 60)])
|
||
address = StringField('Адрес:', [validators.DataRequired(), validators.Length(2, 50)])
|
||
city = StringField('Град:', [validators.DataRequired(), validators.Length(2,40)])
|
||
postcode = DecimalField('Пощенски Код:')
|
||
|
||
clist = []
|
||
for c in countries:
|
||
clist.append((c.alpha2, c.name))
|
||
country = SelectField('Държава:', choices=clist)
|
||
|
||
phone = DecimalField('Телефон:', [validators.DataRequired()])
|
||
org_account = BooleanField('This is a business account')
|
||
org_companyname = StringField('Company Name:')
|
||
org_regaddress = StringField('Company Address:')
|
||
org_responsible = StringField('Primary Contact:')
|
||
org_vatnum = StringField('ДДС Номер:')
|
||
twofactor = BooleanField('2-factor authentication')
|
||
submit = SubmitField('Обнови')
|
||
|
||
def __init__(self, user, *args, **kwargs):
|
||
super(EditProfileAdminForm, self).__init__(*args, **kwargs)
|
||
self.role.choices = [(role.pid, role.name)
|
||
for role in Role.query.order_by(Role.name).all()]
|
||
self.user = user
|
||
|
||
def validate_email(self, field):
|
||
if field.data != self.user.email and User.query.filter_by(email=field.data).first():
|
||
raise ValidationError('Email-а е вече регистриран.')
|
||
|
||
|