This commit is contained in:
Cayo Puigdefabregas 2022-04-13 13:28:22 +02:00
parent 0b5d8d46d1
commit febff54c6b
1 changed files with 3 additions and 2 deletions

View File

@ -1,5 +1,6 @@
from flask import g from flask import g
from flask_wtf import FlaskForm from flask_wtf import FlaskForm
from teal.enums import Country
from werkzeug.security import generate_password_hash from werkzeug.security import generate_password_hash
from wtforms import ( from wtforms import (
BooleanField, BooleanField,
@ -11,7 +12,6 @@ from wtforms import (
) )
from ereuse_devicehub.db import db from ereuse_devicehub.db import db
from ereuse_devicehub.enums import Country
from ereuse_devicehub.resources.user.models import User from ereuse_devicehub.resources.user.models import User
COUNTRY = [(x.name, x.value) for x in Country] COUNTRY = [(x.name, x.value) for x in Country]
@ -104,7 +104,8 @@ class ProfileForm(FlaskForm):
self.last_name.data = user.last_name self.last_name.data = user.last_name
self.email.data = user.email self.email.data = user.email
self.telephone.data = user.telephone self.telephone.data = user.telephone
self.country.data = user.country.name if user.country:
self.country.data = user.country.name
def save(self, commit=True): def save(self, commit=True):
agent = g.user.individual agent = g.user.individual