round 2

api/child_api.py

@@ -3,6 +3,7 @@ from tinydb import Query
 from db.db import child_db, task_db, reward_db
 from api.reward_status import RewardStatus
 from api.child_tasks import ChildTask
+from api.child_rewards import ChildReward
 
 from models.child import Child
 from models.task import Task
@@ -227,6 +228,51 @@ def remove_reward_from_child(id):
         return jsonify({'message': f'Reward {reward_id} removed from {child["name"]}.'}), 200
     return jsonify({'error': 'Reward not assigned to child'}), 400
 
+@child_api.route('/child/<id>/list-rewards', methods=['GET'])
+def list_child_rewards(id):
+    ChildQuery = Query()
+    result = child_db.search(ChildQuery.id == id)
+    if not result:
+        return jsonify({'error': 'Child not found'}), 404
+
+    child = result[0]
+    reward_ids = child.get('rewards', [])
+
+    RewardQuery = Query()
+    child_rewards = []
+    for rid in reward_ids:
+        reward = reward_db.get(RewardQuery.id == rid)
+        if not reward:
+            continue
+        cr = ChildReward(reward.get('name'), reward.get('cost'), reward.get('image_id'), reward.get('id'))
+        child_rewards.append(cr.to_dict())
+
+    return jsonify({'rewards': child_rewards}), 200
+
+@child_api.route('/child/<id>/list-assignable-rewards', methods=['GET'])
+def list_assignable_rewards(id):
+    ChildQuery = Query()
+    result = child_db.search(ChildQuery.id == id)
+    if not result:
+        return jsonify({'error': 'Child not found'}), 404
+
+    child = result[0]
+    assigned_ids = set(child.get('rewards', []))
+
+    all_reward_ids = [r.get('id') for r in reward_db.all() if r and r.get('id')]
+    assignable_ids = [rid for rid in all_reward_ids if rid not in assigned_ids]
+
+    RewardQuery = Query()
+    assignable_rewards = []
+    for rid in assignable_ids:
+        reward = reward_db.get(RewardQuery.id == rid)
+        if not reward:
+            continue
+        cr = ChildReward(reward.get('name'), reward.get('cost'), reward.get('image_id'), reward.get('id'))
+        assignable_rewards.append(cr.to_dict())
+
+    return jsonify({'rewards': assignable_rewards, 'count': len(assignable_rewards)}), 200
+
 @child_api.route('/child/<id>/trigger-reward', methods=['POST'])
 def trigger_child_reward(id):
     data = request.get_json()