trustchain-oc1-orchestral/devicehub-teal
Archived
10
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

use populate_obj

Browse source
This commit is contained in:
Cayo Puigdefabregas 2023-03-07 16:39:52 +01:00
parent 1e62af56a6
commit 0ff2bcae92
1 changed files with 11 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
ereuse_devicehub/forms.py
View file

@ -141,15 +141,17 @@ class SanitizationEntityForm(FlaskForm):
return True return True
def save(self, commit=True): def save(self, commit=True):
sanitation_data = SanitizationEntity( if isinstance(self.logo.data, str):
logo=URL(self.logo.data), self.logo.data = URL(self.logo.data)
company_name=self.company_name.data,
location=self.location.data, sanitation_data = SanitizationEntity.query.filter_by(user_id=g.user.id).first()
responsable_person=self.responsable_person.data,
supervisor_person=self.supervisor_person.data, if not sanitation_data:
user=g.user, sanitation_data = SanitizationEntity(user_id=g.user.id)
) self.populate_obj(sanitation_data)
db.session.add(sanitation_data) db.session.add(sanitation_data)
else:
self.populate_obj(sanitation_data)
if commit: if commit:
db.session.commit() db.session.commit()