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

api/child_rewards.py

@@ -0,0 +1,15 @@
+# api/child_rewards.py
+class ChildReward:
+    def __init__(self, name: str, cost: int, image_id: str, reward_id: str):
+        self.name = name
+        self.cost = cost
+        self.image_id = image_id
+        self.id = reward_id
+
+    def to_dict(self):
+        return {
+            'id': self.id,
+            'name': self.name,
+            'cost': self.cost,
+            'image_id': self.image_id
+        }

api/reward_api.py

@@ -19,6 +19,8 @@ def add_reward():
     reward_db.insert(reward.to_dict())
     return jsonify({'message': f'Reward {name} added.'}), 201
 
+
+
 @reward_api.route('/reward/<id>', methods=['GET'])
 def get_reward(id):
     RewardQuery = Query()
@@ -45,4 +47,44 @@ def delete_reward(id):
                 rewards.remove(id)
                 child_db.update({'rewards': rewards}, ChildQuery.id == child.get('id'))
         return jsonify({'message': f'Reward {id} deleted.'}), 200
-    return jsonify({'error': 'Reward not found'}), 404
+    return jsonify({'error': 'Reward not found'}), 404
+
+@reward_api.route('/reward/<id>/edit', methods=['PUT'])
+def edit_reward(id):
+    RewardQuery = Query()
+    existing = reward_db.get(RewardQuery.id == id)
+    if not existing:
+        return jsonify({'error': 'Reward not found'}), 404
+
+    data = request.get_json(force=True) or {}
+    updates = {}
+
+    if 'name' in data:
+        name = (data.get('name') or '').strip()
+        if not name:
+            return jsonify({'error': 'Name cannot be empty'}), 400
+        updates['name'] = name
+
+    if 'description' in data:
+        desc = (data.get('description') or '').strip()
+        if not desc:
+            return jsonify({'error': 'Description cannot be empty'}), 400
+        updates['description'] = desc
+
+    if 'cost' in data:
+        cost = data.get('cost')
+        if not isinstance(cost, int):
+            return jsonify({'error': 'Cost must be an integer'}), 400
+        if cost <= 0:
+            return jsonify({'error': 'Cost must be a positive integer'}), 400
+        updates['cost'] = cost
+
+    if 'image_id' in data:
+        updates['image_id'] = data.get('image_id', '')
+
+    if not updates:
+        return jsonify({'error': 'No valid fields to update'}), 400
+
+    reward_db.update(updates, RewardQuery.id == id)
+    updated = reward_db.get(RewardQuery.id == id)
+    return jsonify(updated), 200