Bumblebee Class - Jeslyn Lai

app/__init__.py

@@ -2,9 +2,14 @@
 from .db import db, migrate
 from .models import task, goal
 import os
+from .routes.task_routes import bp as tasks_bp
+from .routes.goal_routes import bp as goals_bp
+from flask_cors import CORS
 
 def create_app(config=None):
     app = Flask(__name__)
+    CORS(app)
+    app.config['CORS_HEADERS'] = 'Content-Type'
 
     app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
     app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get('SQLALCHEMY_DATABASE_URI')
@@ -18,5 +23,7 @@ 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

app/models/goal.py

@@ -1,5 +1,32 @@
-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):
+        goal_as_dict = {}
+        goal_as_dict["id"] = self.id
+        goal_as_dict["title"] = self.title
+
+        if self.tasks:
+            goal_as_dict["tasks"] = [task.to_dict() for task in self.tasks]
+
+        return goal_as_dict
+
+    def nested_category(self):
+        updated_goal_as_dict = {}
+        updated_goal_as_dict["goal"] = self.to_dict()
+        return updated_goal_as_dict
+
+
+    @classmethod
+    def from_dict(cls, goal_data):
+
+        new_goal = cls(
+            title=goal_data["title"]
+        )
+
+        return new_goal

app/models/task.py

@@ -1,5 +1,51 @@
-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]] 
+    is_complete : Mapped[Optional[bool]] 
+    goal_id: Mapped[Optional[int]] = mapped_column(ForeignKey("goal.id"))
+    goal: Mapped[Optional["Goal"]] = relationship(back_populates="tasks")
+
+    def to_dict(self):
+        task_as_dict = {}
+        task_as_dict["id"] = self.id
+        task_as_dict["title"] = self.title
+        task_as_dict["description"] = self.description
+        if self.completed_at:
+            task_as_dict["completed_at"] = self.completed_at
+        if self.is_complete:
+            task_as_dict["is_complete"] = self.is_complete
+        else:
+            task_as_dict["is_complete"] = False
+
+        if self.goal_id:
+            task_as_dict["goal_id"] = self.goal_id
+
+        return task_as_dict
+
+    def nested_category(self):
+        updated_task_as_dict = {}
+        updated_task_as_dict["task"] = self.to_dict()
+        return updated_task_as_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", None),
+            is_complete=task_data.get("is_complete", False),
+            goal_id=goal_id
+        )
+
+        return new_task

app/routes/goal_routes.py

@@ -1 +1,75 @@
-from flask import Blueprint
+from flask import Blueprint, request, Response
+from app.models.goal import Goal
+from app.models.task import Task
+from .route_utilities import validate_model, create_model, get_models_with_sort
+from ..db import db
+
+bp = Blueprint("goals_bp", __name__, url_prefix="/goals")
+
+#Create a Goal: Valid Goal
+@bp.post("")
+def create_goal():
+    request_body = request.get_json()
+    return create_model(Goal, request_body)
+
+#Get Goals: Getting Saved Goals
+@bp.get("")
+def get_all_goals():
+    return get_models_with_sort(Goal, request.args)
+
+#Get One Goal: One Saved Goal
+@bp.get("/<goal_id>")
+def get_one_goal(goal_id):
+    goal = validate_model(Goal, goal_id)
+    return goal.nested_category()
+
+#Update Goal
+@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 Response(status=204, mimetype="application/json") 
+
+#Delete Goal: Deleting a Goal
+@bp.delete("/<goal_id>")
+def delete_goal(goal_id):
+    goal = validate_model(Goal, goal_id)
+    db.session.delete(goal)
+    db.session.commit()
+
+    return Response(status=204, mimetype="application/json")
+
+#Sending a List of Task IDs to a Goal
+@bp.post("/<goal_id>/tasks")
+def create_tasks_with_goal(goal_id):
+    goal = validate_model(Goal, goal_id)
+
+    request_body = request.get_json()
+    task_ids = request_body.get("task_ids", [])
+
+    for task in goal.tasks:  #Add this to pass the test test_post_task_ids_to_goal_already_with_goals
+        task.goal_id = None
+
+    for task_id in task_ids:
+        task = validate_model(Task, task_id)
+        if task.goal_id is None:
+            task.goal_id = goal.id
+
+    db.session.commit()
+    return {
+        "id": goal.id,
+        "task_ids": task_ids
+    }, 200
+
+#Getting Tasks of One Goal
+@bp.get("/<goal_id>/tasks")
+def get_tasks_by_goal(goal_id):
+    goal = validate_model(Goal, goal_id)
+    tasks = [task.to_dict() for task in goal.tasks]
+    response = goal.to_dict()
+    response["tasks"] = tasks
+    return response

app/routes/route_utilities.py

@@ -0,0 +1,47 @@
+from flask import abort, make_response
+from ..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 = {"details": f"Invalid data"}
+        abort(make_response(response, 400))
+
+    db.session.add(new_model)
+    db.session.commit()
+
+    return new_model.nested_category(), 201
+
+def get_models_with_sort(cls, args):
+    query = db.select(cls)
+    sort = args.get("sort")
+
+    if sort == "asc":
+        query = query.order_by(cls.title)
+    elif sort == "desc":
+        query = query.order_by(cls.title.desc())
+    else:
+        query = query.order_by(cls.id)
+
+    models = db.session.scalars(query)
+    models_response = [model.to_dict() for model in models]
+
+    return models_response

app/routes/task_routes.py

@@ -1 +1,79 @@
-from flask import Blueprint
+from flask import Blueprint, request, Response
+from app.models.task import Task
+from .route_utilities import validate_model, create_model, get_models_with_sort
+from ..db import db
+from datetime import date
+import os
+import requests
+
+bp = Blueprint("tasks_bp", __name__, url_prefix="/tasks")
+
+#Create a Task: Valid Task With null completed_at
+@bp.post("")
+def create_task():
+    request_body = request.get_json()
+    return create_model(Task, request_body)
+
+#Get Tasks: Getting Saved Tasks
+@bp.get("")
+def get_all_tasks():
+    return get_models_with_sort(Task, request.args)
+
+#Get One Task: One Saved Task
+@bp.get("/<task_id>")
+def get_one_task(task_id):
+    task = validate_model(Task, task_id)
+    return task.nested_category()
+
+#Update Task
+@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"]
+    db.session.commit()
+
+    return Response(status=204, mimetype="application/json") 
+
+#Delete Task: Deleting a Task
+@bp.delete("/<task_id>")
+def delete_task(task_id):
+    task = validate_model(Task, task_id)
+    db.session.delete(task)
+    db.session.commit()
+
+    return Response(status=204, mimetype="application/json")
+
+#Mark Complete on an Incomplete Task
+@bp.patch("/<task_id>/mark_complete")
+def complete_an_incomplete_task(task_id):
+    task = validate_model(Task, task_id)
+
+    task.is_complete = True
+    task.completed_at = date.today()
+    db.session.commit()
+
+    slack_token = os.environ.get("SLACK_API_TOKEN")
+    slack_url = "https://slack.com/api/chat.postMessage"
+    slack_headers = {"Authorization": f"Bearer {slack_token}"}
+    slack_data = {
+        "channel": "task-notifications",
+        "text": f"Someone just completed the task {task.title}"
+    }
+
+    requests.post(slack_url, headers=slack_headers, json=slack_data)
+
+    return Response(status=204, mimetype="application/json") 
+
+#Mark Incomplete on a Completed Task
+@bp.patch("/<task_id>/mark_incomplete")
+def incomplete_an_complete_task(task_id):
+    task = validate_model(Task, task_id)
+
+    task.is_complete = False
+    task.completed_at = None
+    db.session.commit()
+
+    return Response(status=204, mimetype="application/json") 

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