round 4

api/reward_api.py

@@ -4,9 +4,7 @@ from tinydb import Query
 from events.sse import send_event_to_user
 from events.types.event import Event
 from events.types.event_types import EventType
-from events.types.reward_created import RewardCreated
-from events.types.reward_deleted import RewardDeleted
-from events.types.reward_edited import RewardEdited
+from events.types.reward_modified import RewardModified
 from models.reward import Reward
 from db.db import reward_db, child_db
 
@@ -24,8 +22,8 @@ def add_reward():
         return jsonify({'error': 'Name, description, and cost are required'}), 400
     reward = Reward(name=name, description=description, cost=cost, image_id=image)
     reward_db.insert(reward.to_dict())
-    send_event_to_user("user123", Event(EventType.REWARD_CREATED.value,
-                                        RewardCreated(reward.id, "created")))
+    send_event_to_user("user123", Event(EventType.REWARD_MODIFIED.value,
+                                        RewardModified(reward.id, RewardModified.OPERATION_ADD)))
 
     return jsonify({'message': f'Reward {name} added.'}), 201
 
@@ -56,8 +54,8 @@ def delete_reward(id):
             if id in rewards:
                 rewards.remove(id)
                 child_db.update({'rewards': rewards}, ChildQuery.id == child.get('id'))
-                send_event_to_user("user123", Event(EventType.REWARD_DELETED.value,
-                                                    RewardDeleted(id, "created")))
+                send_event_to_user("user123", Event(EventType.REWARD_MODIFIED.value,
+                                                    RewardModified(id, RewardModified.OPERATION_DELETE)))
 
         return jsonify({'message': f'Reward {id} deleted.'}), 200
     return jsonify({'error': 'Reward not found'}), 404
@@ -100,7 +98,7 @@ def edit_reward(id):
 
     reward_db.update(updates, RewardQuery.id == id)
     updated = reward_db.get(RewardQuery.id == id)
-    send_event_to_user("user123", Event(EventType.REWARD_EDITED.value,
-                                        RewardEdited(id, "created")))
+    send_event_to_user("user123", Event(EventType.REWARD_MODIFIED.value,
+                                        RewardModified(id, RewardModified.OPERATION_EDIT)))
 
     return jsonify(updated), 200