def updated_profile(self, user): user.username = self.username.data user.email = self.email.data if self.password.data: user.password = self.password.data if user.company_detail: company = user.company_detail else: company = Company() company.user_id = user.id if self.slug.data: company.slug = self.slug.data if self.location.data: company.location = self.location.data if self.site.data: company.site = self.site.data if self.logo.data: company.logo = self.logo.data if self.description.data: company.description = self.description.data if self.about.data: company.about = self.about.data db.session.add(user) db.session.add(company) db.session.commit()
def update(self, company): company.name = self.name.data company.email = self.email.data company.mobile = self.mobile.data if self.password.data: company.password = self.password.data if company.company_detail: company_detail = company.company_detail else: company_detail = Company() company_detail.user_id = company.id company_detail.site = self.site.data company_detail.description = self.description.data db.session.add(company) db.session.add(company_detail) db.session.commit()