Bumblebee - Gitika K

app/__init__.py

@@ -1,7 +1,11 @@
 from flask import Flask
 from .db import db, migrate
 from .models import task, goal
+from .routes.task_routes import task_bp
+from .routes.goal_routes import goal_bp
 import os
+from dotenv import load_dotenv
+load_dotenv() # Reads .env file and Loads all key-value pairs into environment
 
 def create_app(config=None):
     app = Flask(__name__)
@@ -18,5 +22,8 @@ def create_app(config=None):
     migrate.init_app(app, db)
 
     # Register Blueprints here
-
+    app.register_blueprint(task_bp)
+    app.register_blueprint(goal_bp)
+
     return app
+

app/models/goal.py

@@ -1,5 +1,20 @@
 from sqlalchemy.orm import Mapped, mapped_column
 from ..db import db
+from sqlalchemy.orm import relationship
 
 class Goal(db.Model):
     id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
+    title : Mapped[str]
+    tasks: Mapped[list["Task"]] = relationship(back_populates="goal")
+
+    def to_dict(self):
+        return{
+            "id" : self.id,
+            "title" : self.title
+        }
+
+    @classmethod
+    def from_dict(cls, goal_data):
+        return cls(
+            title = goal_data["title"]
+        )

app/models/task.py

@@ -1,5 +1,34 @@
-from sqlalchemy.orm import Mapped, mapped_column
+from sqlalchemy.orm import Mapped, mapped_column, relationship
 from ..db import db
+from typing import Optional
+from datetime import datetime
+from sqlalchemy import ForeignKey
 
 class Task(db.Model):
     id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
+    title : Mapped[str]
+    description : Mapped[str]
+    completed_at : Mapped[Optional[datetime]] = mapped_column(nullable=True)
+    goal_id: Mapped[Optional[int]] = mapped_column(ForeignKey("goal.id"), nullable=True)
+    goal: Mapped[Optional["Goal"]] = relationship(back_populates="tasks")
+
+    def to_dict(self):
+        task_dict = {
+            "id": self.id,
+            "title": self.title,
+            "description": self.description,
+            "is_complete": self.completed_at is not None,
+        }
+
+        if self.goal_id is not None:
+            task_dict["goal_id"] = self.goal_id
+
+        return task_dict
+
+    @classmethod
+    def from_dict(cls, task_data):
+        return cls(
+            title=task_data["title"], # this will raise KeyError if missing
+            description=task_data["description"],
+            completed_at=task_data.get("completed_at") # this will not raise KeyError if missing
+        )

app/routes/goal_routes.py

@@ -1 +1,114 @@
-from flask import Blueprint
+from flask import Blueprint,make_response, abort, request, Response
+from ..models.goal import Goal
+from ..models.task import Task
+from app.routes.route_utilities import validate_model
+from app import db
+
+goal_bp = Blueprint("goal", __name__, url_prefix="/goals")
+
+@goal_bp.post("")
+def create_goal():
+
+    request_body = request.get_json()
+
+    try:
+        new_goal = Goal.from_dict(request_body)
+    except KeyError as error:
+        message = {
+            # "message": f"Missing '{error.args[0]}' attribute"
+            "details": "Invalid data"
+        }
+        abort(make_response(message, 400))
+    db.session.add(new_goal)
+    db.session.commit()
+
+    response =  {"goal": new_goal.to_dict()}
+    return response, 201
+
+@goal_bp.get("")
+def get_all_goals():
+    query = db.select(Goal)
+    goals = db.session.scalars(query) #This executes the SQL query
+
+    goals_response = []
+    for goal in goals:
+        goals_response.append(goal.to_dict())
+    return goals_response
+
+@goal_bp.get("/<goal_id>")
+def get_one_goal(goal_id):
+    goal = validate_model(Goal, goal_id)
+    return {"goal": goal.to_dict()}
+
+@goal_bp.put("/<goal_id>")
+def update_goal(goal_id):
+
+    goal = validate_model(Goal,goal_id)
+
+    request_body = request.get_json()
+
+    try:
+        goal.title = request_body["title"]
+    except KeyError as error:
+        message = {
+            "message": f"Missing '{error.args[0]}' attribute"
+        }
+        abort(make_response(message, 400))
+
+    db.session.commit()
+
+    return Response(status=204, mimetype="application/json")
+
+@goal_bp.delete("/<goal_id>")
+def remove_task(goal_id):
+
+    goal = validate_model(Goal,goal_id)
+
+    db.session.delete(goal)
+    db.session.commit()
+
+    return Response(status=204, mimetype="application/json")
+
+@goal_bp.post("/<goal_id>/tasks")
+def assign_tasks_to_goal(goal_id):
+    goal = validate_model(Goal, goal_id)
+
+    request_body = request.get_json()
+    task_ids = request_body.get("task_ids", [])
+
+    # Clear existing task associations first
+    for task in goal.tasks:
+        task.goal_id = None
+
+    tasks = []
+    for task_id in task_ids:
+        task = validate_model(Task, task_id)
+        task.goal_id = goal.id
+        tasks.append(task)
+
+    db.session.commit()
+
+    return {
+        "id": goal.id,
+        "task_ids": [task.id for task in tasks]
+    }, 200
+
+@goal_bp.get("/<goal_id>/tasks")
+def get_tasks_of_one_goal(goal_id):
+    goal = validate_model(Goal, goal_id)
+
+    task_list = [task.to_dict() for task in goal.tasks]
+    tasks_response = []
+    for task in goal.tasks:
+        task_dict = task.to_dict()
+        task_dict["goal_id"] = goal.id  # add goal_id if not in to_dict
+        tasks_response.append(task_dict)
+
+    return {
+        "id": goal.id,
+        "title": goal.title,
+        "tasks": tasks_response
+    }, 200
+
+
+

app/routes/route_utilities.py

@@ -0,0 +1,17 @@
+from flask import abort, make_response
+from app import db
+
+def validate_model(cls, model_id):
+    try:
+        model_id = int(model_id)
+    except ValueError:
+        response = {"message" : f"id {model_id} invalid"}
+        abort(make_response(response, 400))
+
+    query = db.select(cls).where(cls.id == model_id)
+    instance = db.session.scalar(query)
+
+    if instance is None:
+        abort(make_response({"message": f"{cls.__name__} ID ({model_id}) not found."}, 404))
+    return instance
+

app/routes/task_routes.py

@@ -1 +1,139 @@
-from flask import Blueprint
+from flask import Blueprint,make_response, abort, request, Response
+from ..models.task import Task
+from app.routes.route_utilities import validate_model
+from app import db
+from sqlalchemy import asc, desc
+from datetime import datetime, timezone
+import os
+import requests
+
+task_bp = Blueprint("task", __name__, url_prefix="/tasks")
+
+@task_bp.post("")
+def create_task():
+
+    request_body = request.get_json()
+
+    try:
+        new_task = Task.from_dict(request_body)
+    except KeyError as error:
+        message = {
+            # "message": f"Missing '{error.args[0]}' attribute"
+            "details": "Invalid data"
+        }
+        abort(make_response(message, 400))
+    db.session.add(new_task)
+    db.session.commit()
+
+    response =  {"task": new_task.to_dict()}
+    return response, 201
+
+@task_bp.get("")
+def get_all_tasks():
+    sort_param = request.args.get("sort")
+
+    query = db.select(Task)
+    if sort_param == "asc":
+        query = query.order_by(asc(Task.title))
+    elif sort_param == "desc":
+        query = query.order_by(desc(Task.title))
+
+    tasks = db.session.scalars(query) #This executes the SQL query
+
+    tasks_response = []
+    for task in tasks:
+        tasks_response.append(task.to_dict())
+    return tasks_response
+
+@task_bp.get("/<task_id>")
+def get_one_task(task_id):
+    task = validate_model(Task,task_id)
+    # return task.to_dict()
+    return {"task": task.to_dict()}
+
+
+# def validate_task(task_id):
+#     try:
+#         task_id = int(task_id)
+#     except ValueError:
+#         response = {"message" : f"task {task_id} invalid"}
+#         abort(make_response(response, 400))
+
+#     # task = Task.query.get(task_id)
+#     query = db.select(Task).where(Task.id == task_id)
+#     task = db.session.scalar(query)
+#     # db.session.get(Task, task_id)
+
+#     if task is None:
+#         message = {
+#             "message": f"task ID ({task_id}) not found."
+#         }
+#         abort(make_response(message, 404))
+#     return task
+
+@task_bp.put("/<task_id>")
+def update_task(task_id):
+
+    task = validate_model(Task,task_id)
+
+    request_body = request.get_json()
+
+    try:
+        task.title = request_body["title"]
+        task.description = request_body["description"]
+    except KeyError as error:
+        message = {
+            "message": f"Missing '{error.args[0]}' attribute"
+        }
+        abort(make_response(message, 400))
+
+    db.session.commit()
+
+    return Response(status=204, mimetype="application/json")
+
+@task_bp.delete("/<task_id>")
+def remove_task(task_id):
+
+    task = validate_model(Task,task_id)
+
+    db.session.delete(task)
+    db.session.commit()
+
+    return Response(status=204, mimetype="application/json")
+
+def send_slack_message(task_title):
+    slack_token = os.environ.get("SLACK_BOT_TOKEN")
+    channel = os.environ.get("SLACK_CHANNEL", "task-notifications")
+
+    url = "https://slack.com/api/chat.postMessage"
+    headers = {
+        "Authorization": f"Bearer {slack_token}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "channel": f"#{channel}",
+        "text": f"Someone just completed the task {task_title}"
+    }
+
+    requests.post(url, json=payload, headers=headers)
+
+@task_bp.patch("/<task_id>/mark_complete")
+def mark_complete(task_id):
+    task = validate_model(Task,task_id)
+
+    task.completed_at = datetime.now(timezone.utc)
+    db.session.commit()
+
+    send_slack_message(task.title)
+
+    return Response(status=204, mimetype="application/json")
+
+@task_bp.patch("/<task_id>/mark_incomplete")
+def mark_incomplete(task_id):
+    task = validate_model(Task,task_id)
+
+    task.completed_at = None
+    db.session.commit()
+
+    return Response(status=204, mimetype="application/json")
+

migrations/README

@@ -0,0 +1 @@
+Single-database configuration for Flask.