Cohort22 - Sphinx - TatianaSnook

app/__init__.py

@@ -1,11 +1,14 @@
 from flask import Flask
 from .db import db, migrate
-from .models import task, goal
+from .models import task
+from .models import goal
+from .routes.task_routes import tasks_bp
+from .routes.goal_routes import bp as goal_bp
 import os
 
 def create_app(config=None):
     app = Flask(__name__)
-
+    
     app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
     app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get('SQLALCHEMY_DATABASE_URI')
 
@@ -18,5 +21,7 @@ def create_app(config=None):
     migrate.init_app(app, db)
 
     # Register Blueprints here
+    app.register_blueprint(tasks_bp)
+    app.register_blueprint(goal_bp)
 
     return app

app/models/goal.py

@@ -1,5 +1,20 @@
-from sqlalchemy.orm import Mapped, mapped_column
+from sqlalchemy.orm import Mapped, mapped_column, relationship
 from ..db import db
+from app.routes.route_utilities import create_model
 
 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):
+        response = {
+            "id": self.id,
+            "title": self.title
+        }
+        return response
+
+    @classmethod
+    def from_dict(cls, goal_data):
+        new_goal = cls(title=goal_data["title"])
+        return new_goal

app/models/task.py

@@ -1,5 +1,33 @@
-from sqlalchemy.orm import Mapped, mapped_column
+from sqlalchemy.orm import Mapped, mapped_column, relationship
 from ..db import db
+from sqlalchemy import String, Boolean, ForeignKey
+from typing import Optional
+
 
 class Task(db.Model):
     id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
+    title: Mapped[str]
+    description: Mapped[str]
+    completed_at: Mapped[str | None] = mapped_column(String, nullable=True)
+    is_complete: Mapped[bool] = mapped_column(Boolean, default=False)
+    goal_id: Mapped[Optional[int]] = mapped_column(ForeignKey("goal.id"))
+    goal: Mapped[Optional["Goal"]] = relationship(back_populates="tasks")
+
+    def to_dict(self, include_goal_id=False):
+        response = {
+            "id": self.id,
+            "title": self.title,
+            "description": self.description,
+            "is_complete": self.completed_at is not None,
+        }
+        if include_goal_id and self.goal_id is not None:
+            response["goal_id"] = self.goal_id
+        return response
+
+    @classmethod
+    def from_dict(cls, task_data):
+        return cls(
+            title=task_data['title'],
+            description=task_data['description'],
+            goal_id=task_data.get("goal_id", None)
+        )

app/routes/goal_routes.py

@@ -1 +1,110 @@
-from flask import Blueprint
+from flask import Blueprint, request, abort, make_response
+from app.models.goal import Goal
+from app.models.task import Task
+from app.db import db
+from app.routes.route_utilities import validate_model
+
+bp = Blueprint("goal_bp", __name__, url_prefix="/goals")
+
+@bp.post("")
+def create_goal():
+    request_body = request.get_json()
+
+    if "title" not in request_body:
+        return {"details": "Invalid data"}, 400
+
+    title = request_body["title"]
+
+    new_goal = Goal(title=title)
+    db.session.add(new_goal)
+    db.session.commit()
+
+    return {"goal": new_goal.to_dict()}, 201
+
+
+@bp.get("")
+def get_all_goals():
+    query = db.select(Goal).order_by(Goal.id)
+    goals = db.session.scalars(query)
+
+    return [goal.to_dict() for goal in goals]
+
+
+@bp.get("/<goal_id>")
+def get_single_goal(goal_id):
+
+    goal = validate_model(Goal, goal_id)
+    return {"goal": goal.to_dict()}
+
+
+@bp.put("/<goal_id>")
+def update_goal(goal_id):
+    goal = validate_model(Goal, goal_id)
+    request_body = request.get_json()
+
+    goal.title = request_body["title"]
+    db.session.commit()
+
+    return {"goal": goal.to_dict()}
+
+
+@bp.delete("/<goal_id>")
+def delete_task(goal_id):
+    goal = validate_model(Goal, goal_id)
+
+    db.session.delete(goal)
+    db.session.commit()
+
+    return {"details": f'Goal {goal_id} "Build a habit of going outside daily" successfully deleted'}
+
+
+@bp.post("/<goal_id>/tasks")
+def post_tasks_ids_to_goal(goal_id):
+    goal = validate_model(Goal, goal_id)
+    request_body = request.get_json()
+    task_ids = request_body.get("task_ids", [])
+
+    if not task_ids:
+        response = {"message": "Invalid request: missing task_ids"}
+        abort(make_response(response, 400))
+
+    tasks_to_add = []
+    for task_id in task_ids:
+        task = validate_model(Task, task_id)
+
+        if task not in goal.tasks:
+            tasks_to_add.append(task)
+
+    if tasks_to_add:
+        goal.tasks.extend(tasks_to_add)
+        try:
+            db.session.add(goal)
+            db.session.commit()
+        except Exception as e:
+            response = {
+                "message": f"Failed to update goal with tasks: {str(e)}"}
+            abort(make_response(response, 500))
+
+    return {"id": goal.id, "task_ids": task_ids}, 200
+
+
+@bp.get("/<goal_id>/tasks")
+def get_tasks_for_specific_goal(goal_id):
+    goal = validate_model(Goal, goal_id)
+
+    response = {
+        "id": goal.id,
+        "title": goal.title,
+        "tasks": [
+            {
+                "id": task.id,
+                "goal_id": goal.id,  
+                "title": task.title,
+                "description": task.description,
+                "is_complete": task.completed_at is not None
+            }
+            for task in goal.tasks
+        ]
+    }
+
+    return response, 200

app/routes/route_utilities.py

@@ -0,0 +1,33 @@
+from flask import abort, make_response
+from app.db import db
+
+
+def validate_model(cls, model_id):
+    try:
+        model_id = int(model_id)
+    except:
+        response = {"message": f"{cls.__name__} {model_id} invalid"}
+        abort(make_response(response, 400))
+
+    query = db.select(cls).where(cls.id == model_id)
+    model = db.session.scalar(query)
+
+    if not model:
+        response = {"message": f"{cls.__name__} {model_id} not found"}
+        abort(make_response(response, 404))
+
+    return model
+
+
+def create_model(cls, model_data):
+    try:
+        new_model = cls.from_dict(model_data)
+
+    except KeyError as error:
+        response = {"message": f"Invalid request: missing {error.args[0]}"}
+        abort(make_response(response, 400))
+
+    db.session.add(new_model)
+    db.session.commit()
+
+    return new_model.to_dict(), 201

app/routes/task_routes.py

@@ -1 +1,117 @@
-from flask import Blueprint
+from flask import Blueprint, request, abort, make_response, Response
+from app.models.task import Task
+from app.db import db
+from datetime import datetime
+import os
+import requests
+from app.routes.route_utilities import validate_model
+
+tasks_bp = Blueprint("task_bp", __name__, url_prefix="/tasks")
+
+@tasks_bp.post("")
+def create_task():
+    request_body = request.get_json()
+
+    if "title" not in request_body or "description" not in request_body:
+        return {"details": "Invalid data"}, 400
+
+    title = request_body["title"]
+    description = request_body["description"]
+    completed_at = request_body.get("completed_at", None)
+    is_complete = request_body.get("is_complete", False)
+    goal_id = request_body.get("goal_id", None)
+
+    new_task = Task(
+        title=title,
+        description=description,
+        is_complete=is_complete,
+        goal_id=goal_id
+    )
+    db.session.add(new_task)
+    db.session.commit()
+
+    response = {"task": new_task.to_dict()}
+    return response, 201
+
+
+@tasks_bp.get("")
+def get_all_tasks():
+
+    sort_order = request.args.get("sort")
+
+    if sort_order == "asc":
+        query = db.select(Task).order_by(Task.title.asc())
+    elif sort_order == "desc":
+        query = db.select(Task).order_by(Task.title.desc())
+    else:
+        query = db.select(Task).order_by(Task.id)
+
+    tasks = db.session.scalars(query)
+    tasks_response = [task.to_dict(include_goal_id=True) for task in tasks]
+
+    return tasks_response
+
+@tasks_bp.get("/<task_id>")
+def get_single_task(task_id):
+    task = validate_model(Task, task_id)
+
+    include_goal_id = task.goal_id is not None
+    return {"task": task.to_dict(include_goal_id=include_goal_id)}
+
+
+@tasks_bp.put("/<task_id>")
+def update_task(task_id):
+    task = validate_model(Task, task_id)
+    request_body = request.get_json()
+
+    task.title = request_body["title"]
+    task.description = request_body["description"]
+    task.completed_at = request_body.get("completed_at", None)
+    task.is_complete = request_body.get("is_complete", False)
+
+    db.session.commit()
+
+    return {"task": task.to_dict()}
+
+
+@tasks_bp.patch("/<task_id>/mark_complete")
+def mark_task_complete(task_id):
+    task = validate_model(Task, task_id)
+
+    task.completed_at = datetime.utcnow()
+    db.session.commit()
+
+    url = "https://slack.com/api/chat.postMessage"
+    api_key = os.environ.get("SLACK_BOT_TOKEN")
+    headers = {"Authorization": f"Bearer {api_key}"}
+    request_body = {
+        "channel": "task-notifications",
+        "text": f"Task '{task.title}' has been completed!"
+    }
+
+    response = requests.post(url, headers=headers, data=request_body)
+
+    if response.status_code != 200:
+        return {"error": "Failed to send Slack notification"}, 500
+
+    return {"task": task.to_dict()}, 200
+
+
+@tasks_bp.patch("<task_id>/mark_incomplete")
+def mark_task_incomplete(task_id):
+    task = validate_model(Task, task_id)
+
+    task.completed_at = None
+    db.session.commit()
+
+    return {"task": task.to_dict()}, 200
+
+
+@tasks_bp.delete("/<task_id>")
+def delete_task(task_id):
+    task = validate_model(Task, task_id)
+
+    db.session.delete(task)
+    db.session.commit()
+
+    return {"details": f'Task {task_id} "Go on my daily walk 🏞" successfully deleted'}

migrations/README

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