initial commit

api/task_api.py

@@ -0,0 +1,48 @@
+from flask import Blueprint, request, jsonify
+from tinydb import Query
+from models.task import Task
+from db.db import task_db, child_db
+
+task_api = Blueprint('task_api', __name__)
+
+# Task endpoints
+@task_api.route('/task/add', methods=['PUT'])
+def add_task():
+    data = request.get_json()
+    name = data.get('name')
+    points = data.get('points')
+    is_good = data.get('is_good')
+    image = data.get('image_id', '')
+    if not name or points is None or is_good is None:
+        return jsonify({'error': 'Name, points, and is_good are required'}), 400
+    task = Task(name, points, is_good, image_id=image)
+    task_db.insert(task.to_dict())
+    return jsonify({'message': f'Task {name} added.'}), 201
+
+@task_api.route('/task/<id>', methods=['GET'])
+def get_task(id):
+    TaskQuery = Query()
+    result = task_db.search(TaskQuery.id == id)
+    if not result:
+        return jsonify({'error': 'Task not found'}), 404
+    return jsonify(result[0]), 200
+
+@task_api.route('/task/list', methods=['GET'])
+def list_tasks():
+    tasks = task_db.all()
+    return jsonify({'tasks': tasks}), 200
+
+@task_api.route('/task/<id>', methods=['DELETE'])
+def delete_task(id):
+    TaskQuery = Query()
+    removed = task_db.remove(TaskQuery.id == id)
+    if removed:
+        # remove the task id from any child's task list
+        ChildQuery = Query()
+        for child in child_db.all():
+            tasks = child.get('tasks', [])
+            if id in tasks:
+                tasks.remove(id)
+                child_db.update({'tasks': tasks}, ChildQuery.id == child.get('id'))
+        return jsonify({'message': f'Task {id} deleted.'}), 200
+    return jsonify({'error': 'Task not found'}), 404