def fake_company(self): for company in self.lagou.company: c = User( name=company['companyShortName'], email=fake.email(), role=User.ROLE_COMPANY ) c.password = '******' db.session.add(c) try: db.session.commit() except: db.session.rollback() continue d = CompanyDetail( logo='https://static.lagou.com/thumbnail_160x160/' + company['companyLogo'], site='https://shiyanlou.com', location=company['city'], field=company['industryField'], finance_stage=company['financeStage'] ) d.user_id = c.id db.session.add(d) db.session.commit()
def updated_profile(self, user): user.name = self.name.data user.email = self.email.data if self.password.data: user.password = self.password.data if user.company_detail: company_detail = user.company_detail else: company_detail = CompanyDetail() company_detail.user_id = user.id self.populate_obj(company_detail) db.session.add(user) db.session.add(company_detail) db.session.commit()
def create_company(self): user = User(name=self.name.data, email=self.email.data, password=self.password.data, role=User.ROLE_COMPANY) db.session.add(user) db.session.commit() company = CompanyDetail(site=self.site.data, logo=self.logo.data, location=self.location.data, description=self.description.data) company.user_id = user.id db.session.add(company) db.session.commit()
def update(self, company): company.name = self.name.data company.email = self.email.data if self.password.data: company.password = self.password.data if company.detail: detail = company.detail else: detail = CompanyDetail() detail.user_id = company.id detail.site = self.site.data detail.description = self.description.data db.session.add(company) db.session.add(detail) db.session.commit()
def update(self, company): company.name = self.name.data company.email = self.email.data if self.password.data: company.password = self.password.data if company.detail: detail = company.detail else: detail = CompanyDetail() detail.user_id = company.id detail.site = self.site.data detail.description = self.description.data db.session.add(company) db.session.add(detail) db.session.commit()
def updated_profile(self, user): user.name = self.name.data user.email = self.email.data if self.password.data: user.password = self.password.data if user.detail: detail = user.detail else: detail = CompanyDetail() detail.user_id = user.id self.populate_obj(detail) db.session.add(user) db.session.add(detail) db.session.commit()
def updated_profile(self, company: Company): company.name = self.name.data company.email = self.email.data company.phone = self.phone.data if self.password.data: company.password = self.password.data if company.detail: detail = company.detail else: detail = CompanyDetail() detail.user_id = company.id self.populate_obj(detail) db.session.add(company) db.session.add(detail) db.session.commit()