def test_get_prizes(self, mock_get_db):
expected_query = "SELECT * FROM prizes WHERE user_id = ? AND is_active = 1 ORDER BY points"
db_mock = Mock()
mock_get_db.return_value = db_mock
prizes.get_prizes(child_id=1)
db_mock.execute.assert_called_with(expected_query, (1,))
def index():
role = g.user_data['role']
logged_user_id = g.user_data['id']
template = None
context = {}
if role == 'caregiver':
children = users.get_caregivers_children(logged_user_id)
template = 'index_caregiver.html'
context.update({
"firstname": g.user_data['firstname'],
"role": role,
"children": children
})
elif role == 'child':
template = 'index_child.html'
child = users.get_child_data(logged_user_id)
child_points = points.get_child_points(child['id'])
childs_tasks = tasks.get_tasks(logged_user_id)
childs_prizes = prizes.get_prizes(logged_user_id)
context = {
'child': child,
'child_points': child_points,
'childs_tasks': childs_tasks,
'childs_prizes': childs_prizes
}
messages = get_flashed_messages()
return render_template(template, **context, messages=messages)
def child(child_id):
child = users.get_child_data(child_id)
childs_tasks = tasks.get_tasks(child_id)
childs_prizes = prizes.get_prizes(child_id)
role = g.user_data['role']
child_points = points.get_child_points(child['id'])
info = bans.get_most_important_warn_ban(child_id).split(';')
info_str = info[0]
info_bck = info[1]
context = {
'child': child,
'childs_tasks': childs_tasks,
'childs_prizes': childs_prizes,
'role': role,
'child_points': child_points,
'info_str': info_str,
'info_bck': info_bck,
"firstname": g.user_data['firstname']
}
messages = get_flashed_messages()
return render_template('base/index_child.html',
**context,
messages=messages)
def child(child_id):
child = users.get_child_data(child_id)
childs_tasks = tasks.get_tasks(child_id)
childs_prizes = prizes.get_prizes(child_id)
role = g.user_data['role']
child_points = points.get_child_points(child['id'])
context = {'child': child,
'childs_tasks': childs_tasks,
'childs_prizes': childs_prizes,
'role': role,
'child_points': child_points}
messages = get_flashed_messages()
return render_template('caregiver_panel.html', **context, messages=messages)
def prizes_detail(child_id):
role = g.user_data['role']
try:
child = users.get_child_data(child_id)
except TypeError:
return abort(403)
childs_prizes = prizes.get_prizes(child_id)
child_points = points.get_child_points(child['id'])
context = {'child': child,
'childs_prizes': childs_prizes,
'role': role,
'child_points': child_points}
return render_template('prizes_detail.html', **context)
