def create_boss(company): user = User(username="******", email='*****@*****.**', password='******', role=User.ROLE_BOSS) user.company = company return user
def create_user(self): user = User() self.populate_obj(user) user.role = 20 user.company = Company(name=self.name.data) db.session.add(user) db.session.commit() return user
def add_company_user(self): company = Company() user = User() self.populate_obj(company) self.populate_obj(user) user.company = company user.role = user.ROLE_COMPANY db.session.add(user) db.session.add(company) db.session.commit() flash("更新成功", 'success') return [user, company]