def create_company(self):
company = Company()
company.user_id = current_user.id
company.company_name = self.company_name.data
company.offical_websit = self.offical_websit.data
company.address = self.offical_websit.data
company.company_TEL = self.company_TEL.data
company.description = self.description.data
db.session.add(company)
db.session.commit()
return company
def create_companyProfile(self):
user = User()
company = Company()
company.location = self.location.data
company.number_of_people = dict(PEOPLE_CHOICES).get(
self.number_of_people.data)
company.company_name = self.companyName.data
company.save()
user.company_id = company.id
user.username = self.username.data
user.email = self.email.data
user.password = self.password.data
user.save()