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]
