def get_team(self): sess = db.session() has_default_team = sess.query(Team.query.filter(Team.name == 'default'). exists()).scalar() if not has_default_team: print("Creating default Team") team = Team(name='default', events_token='424242') sess.add(team) sess.commit() else: team = Team.query.filter(Team.name == 'default').one() return team
def run(self, password): team = self.get_team() sess = db.session() has_admin = sess.query(User.query.filter(User.username == 'admin'). exists()).scalar() if not has_admin: user = User(username='******', password=hash_password(password), admin=True, team_id=team.id) sess.add(user) sess.commit() else: print("User admin is already exists")