Bumblebee-JennyXu

app/__init__.py

@@ -1,7 +1,12 @@
 from flask import Flask
 from .db import db, migrate
 from .models import task, goal
+from .routes.task_routes import tasks_bp
+from .routes.goal_routes import goals_bp
 import os
+from dotenv import load_dotenv
+
+load_dotenv()
 
 def create_app(config=None):
     app = Flask(__name__)
@@ -18,5 +23,9 @@ def create_app(config=None):
     migrate.init_app(app, db)
 
     # Register Blueprints here
+    app.register_blueprint(tasks_bp)
+    app.register_blueprint(goals_bp)
 
     return app
+
+SLACK_API_TOKEN = os.environ.get("SLACK_API_TOKEN")

app/models/goal.py

@@ -1,5 +1,21 @@
-from sqlalchemy.orm import Mapped, mapped_column
+from sqlalchemy.orm import Mapped, mapped_column, relationship
 from ..db import db
 
 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"])
+
+    from typing import TYPE_CHECKING
+    if TYPE_CHECKING:
+        from .task import Task

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 datetime import datetime
+from sqlalchemy import 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[datetime] = mapped_column (nullable=True)
+    goal_id: Mapped[Optional[int]] = mapped_column (ForeignKey("goal.id"))
+    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": bool(self.completed_at)
+    }
+
+        if self.goal:
+            task_dict["goal_id"] = self.goal_id
+        return task_dict
+
+
+    @classmethod
+    def from_dict(cls, task_data):
+        goal_id = task_data.get("goal_id")
+        new_task = cls(title=task_data["title"], description=task_data["description"], completed_at=task_data.get("completed_at"), goal_id=goal_id)
+        return new_task
+
+

app/routes/goal_routes.py

@@ -1 +1,98 @@
-from flask import Blueprint
+from flask import  Blueprint, request, Response, abort, make_response
+from .routes_utilities import validate_model, create_model, get_models_with_filters
+from ..db import db
+from app.models.goal import Goal
+from app.models.task import Task
+
+
+goals_bp = Blueprint("goals_bp", __name__, url_prefix = "/goals")
+
+@goals_bp.get("")
+def get_all_goals():
+    # query = db.select(Goal)
+
+    # sort_param = request.args.get("sort")
+    # if sort_param == "desc":
+    #     query = query.order_by(Goal.title.desc())
+
+    # if sort_param == "asc":
+    #     query = query.order_by(Goal.title)
+
+    # goals = db.session.scalars(query)
+
+    # goal_response = [goal.to_dict() for goal in goals]
+
+    # return goal_response
+    return get_models_with_filters(Goal, request.args)
+
+
+@goals_bp.get("/<id>")
+def get_one_Goal(id):
+    goal = validate_model(Goal, id)
+    return {"goal":goal.to_dict()}
+
+@goals_bp.put("/<id>")
+def update_Goal(id):
+    goal = validate_model(Goal, id)
+    request_body = request.get_json()
+
+    goal.title = request_body["title"]
+
+    db.session.commit()
+    return Response(status=204, mimetype="application/json")
+
+
+@goals_bp.delete("/<id>")
+def delete_Goal(id):
+    goal = validate_model(Goal, id)
+
+    db.session.delete(goal)
+    db.session.commit()
+
+    return Response(status=204, mimetype="application/json")
+
+@goals_bp.post("")
+def post_new_Goal():
+    request_body = request.get_json()
+
+    # try: 
+    #     new_goal = Goal.from_dict(request_body)
+    # except KeyError as error:
+    #     response = {"details": "Invalid data"}
+    #     abort(make_response(response, 400))
+
+    # db.session.add(new_goal)
+    # db.session.commit()
+
+    # return make_response({"goal": new_goal.to_dict()}, 201)
+    return create_model(Goal, request_body)
+
+@goals_bp.post("/<goal_id>/tasks")
+def create_tasks_with_goal(goal_id):
+    goal = validate_model(Goal, goal_id)
+    request_body = request.get_json()
+
+    if not request_body or "task_ids" not in request_body:
+        return make_response({"details": "Invalid data"}, 400)
+
+    task_ids = request_body["task_ids"]
+
+    goal.tasks.clear()
+
+    for task_id in task_ids:
+        task = validate_model(Task, task_id)
+        task.goal_id = goal.id
+
+    db.session.commit()
+
+    return {"id": goal.id, "task_ids": task_ids}, 200
+
+
+@goals_bp.get("/<goal_id>/tasks")
+def get_tasks_by_goal(goal_id):
+    goal = validate_model(Goal, goal_id)
+    response = {"id": goal.id, "title": goal.title, "tasks": [task.to_dict() for task in goal.tasks]}
+    return response
+
+
+

app/routes/routes_utilities.py

@@ -0,0 +1,41 @@
+from ..db import db
+from flask import abort, make_response
+
+def validate_model(cls, id):
+    try: 
+        id = int(id)
+    except: 
+        abort(make_response({"message": f"{cls.__name__} id {id} is invalid."}, 400))
+
+    query = db.select(cls).where(cls.id == id)
+    model = db.session.scalar(query)
+    if not model:
+        abort(make_response({"message": f"{cls.__name__} id {id} is not found."}, 404))
+    return model
+
+def create_model(cls, model_data):
+    try:
+        new_model = cls.from_dict(model_data)
+    except KeyError as error:
+        response = {"details": "Invalid data"}
+        abort(make_response(response, 400))
+
+    db.session.add(new_model)
+    db.session.commit()
+
+    return make_response({cls.__name__.lower(): new_model.to_dict()}, 201)
+
+
+
+def get_models_with_filters(cls, filters=None):
+    query = db.select(cls)
+
+    if filters:
+        for attribute, value in filters.items():
+            if hasattr(cls, attribute):
+                query = query.where(getattr(cls, attribute).ilike(f"%{value}%"))
+
+    models = db.session.scalars(query.order_by(cls.id))
+    models_response = [model.to_dict() for model in models]
+    return models_response
+

app/routes/task_routes.py

@@ -1 +1,107 @@
-from flask import Blueprint
+from flask import  Blueprint, request, Response, abort, make_response
+from .routes_utilities import validate_model, create_model, get_models_with_filters
+from ..db import db
+from app.models.task import Task
+from datetime import datetime
+import os
+import requests
+
+tasks_bp = Blueprint("tasks_bp", __name__, url_prefix = "/tasks")
+
+@tasks_bp.get("")
+def get_all_tasks():
+    query = db.select(Task)
+
+    sort_param = request.args.get("sort")
+    if sort_param == "desc":
+        query = query.order_by(Task.title.desc())
+
+    if sort_param == "asc":
+        query = query.order_by(Task.title)
+
+    tasks = db.session.scalars(query)
+
+    tasks_response = [task.to_dict() for task in tasks]
+
+    return tasks_response
+
+@tasks_bp.get("/<id>")
+def get_one_task(id):
+    task = validate_model(Task, id)
+    return {"task":task.to_dict()}
+
+@tasks_bp.put("/<id>")
+def update_task(id):
+    task = validate_model(Task, id)
+    request_body = request.get_json()
+
+    task.title = request_body["title"]
+    task.description = request_body["description"]
+
+    db.session.commit()
+    return Response(status=204, mimetype="application/json")
+
+@tasks_bp.patch("/<id>/mark_complete")
+def mark_completed(id):
+    task = validate_model(Task, id)
+
+    task.completed_at = datetime.now()
+
+    db.session.commit()
+
+    send_slack_msg(task_title=task.title)
+
+    return Response(status=204, mimetype="application/json")
+
+@tasks_bp.patch("/<id>/mark_incomplete")
+def is_incompleted(id):
+    task = validate_model(Task, id)
+
+    task.completed_at = None
+
+    db.session.commit()
+
+    return Response(status=204, mimetype="application/json")
+
+
+@tasks_bp.delete("/<id>")
+def delete_task(id):
+    task = validate_model(Task, id)
+
+    db.session.delete(task)
+    db.session.commit()
+
+    return Response(status=204, mimetype="application/json")
+
+@tasks_bp.post("")
+def post_new_task():
+    request_body = request.get_json()
+    return create_model(Task, request_body)
+
+    # try: 
+    #     new_task = Task.from_dict(request_body)
+    # except KeyError as error:
+    #     response = {"details": "Invalid data"}
+    #     abort(make_response(response, 400))
+
+    # db.session.add(new_task)
+    # db.session.commit()
+
+    # return make_response({"task": new_task.to_dict()}, 201)
+
+def send_slack_msg(task_title):
+    token = os.environ.get("SLACK_API_TOKEN")
+    url = "https://slack.com/api/chat.postMessage"
+    headers = {
+        "Authorization": f"Bearer {token}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "channel": "task-notifications",
+        "text": f"Someone just completed the task {task_title}"
+    }
+
+    response = requests.post(url, headers=headers, json=data)
+
+    if response.status_code != 200 or not response.json().get("ok"):
+        print("Slack API error:", response.text)

migrations/README

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

migrations/alembic.ini

@@ -0,0 +1,50 @@
+# A generic, single database configuration.
+
+[alembic]
+# template used to generate migration files
+# file_template = %%(rev)s_%%(slug)s
+
+# set to 'true' to run the environment during
+# the 'revision' command, regardless of autogenerate
+# revision_environment = false
+
+
+# Logging configuration
+[loggers]
+keys = root,sqlalchemy,alembic,flask_migrate
+
+[handlers]
+keys = console
+
+[formatters]
+keys = generic
+
+[logger_root]
+level = WARN
+handlers = console
+qualname =
+
+[logger_sqlalchemy]
+level = WARN
+handlers =
+qualname = sqlalchemy.engine
+
+[logger_alembic]
+level = INFO
+handlers =
+qualname = alembic
+
+[logger_flask_migrate]
+level = INFO
+handlers =
+qualname = flask_migrate
+
+[handler_console]
+class = StreamHandler
+args = (sys.stderr,)
+level = NOTSET
+formatter = generic
+
+[formatter_generic]
+format = %(levelname)-5.5s [%(name)s] %(message)s
+datefmt = %H:%M:%S