def generate_test_data(session): # create teams team_list = [] for team in dummy.teams: t = Team(**team) t.monthly_goal = random.randrange(50000, 100000, 1000) session.add(t) team_list.append(t) session.flush() user_chunks = list(chunks(dummy.users, 20)) user_list = [] for idx, team in enumerate(team_list): for user in user_chunks[idx]: u = User(**user) u.team_id = team.id u.username = u.name.lower().replace(' ', '_') u.password = '******' u.user_type = 'User' u.monthly_goal = random.randrange(1000, 5000, 1000) session.add(u) user_list.append(u) admin = User(username="******", name="John Kida", email='*****@*****.**', team_id=1, user_type='Admin') admin.password = '******' user = User(username="******", name="John Kida", email='*****@*****.**', team_id=1, user_type='User') user.password = '******' session.add(admin) session.add(user) session.flush() days_ago = date.today() - timedelta(days=30) for x in range(1, 30): d = days_ago + timedelta(days=x) for u in user_list: s = Sale(user_id=u.id, value=random.randint(1, 200), date=d) session.add(s)
def generate_init_data(session): admin = User(username="******", name="John Kida", email='*****@*****.**', user_type='Admin') admin.password = '******' session.add(admin) session.flush()