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()
