Sela Crow HoneyBees Cohort 23

app/__init__.py

@@ -1,13 +1,19 @@
 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
 
+
 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')
+    app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get(
+        'SQLALCHEMY_DATABASE_URI')
+    # app.config['SLACK_BOT_TOKEN'] = os.environ.get("SLACK_BOT_TOKEN")
+    # app.config['SLACK_CHANNEL'] = os.environ.get("SLACK_CHANNEL")
 
     if config:
         # Merge `config` into the app's configuration
@@ -18,5 +24,6 @@ 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,21 @@
-from sqlalchemy.orm import Mapped, mapped_column
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+# from sqlalchemy import ForeignKe
+from typing import List
 from ..db import db
 
+
 class Goal(db.Model):
     id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
+    title: Mapped[str] = mapped_column(nullable=False)
+    tasks: Mapped[List["Task"]] = relationship("Task", back_populates="goal")
+
+    def to_dict(self):
+        return {
+            "id": self.id,
+            "title": self.title,
+            # "tasks": [task.to_dict() for task in self.tasks]
+        }
+
+    @classmethod
+    def from_dict(cls, goal_data):
+        return cls(title=goal_data["title"])

app/models/task.py

@@ -1,5 +1,46 @@
-from sqlalchemy.orm import Mapped, mapped_column
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+from datetime import datetime
+from typing import Optional
+from sqlalchemy import ForeignKey
+# from sqlalchemy import Datetime
 from ..db import db
 
+
 class Task(db.Model):
     id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
+    title: Mapped[str] = mapped_column(nullable=False)
+    goal_id: Mapped[Optional[int]] = mapped_column(ForeignKey("goal.id"))
+    goal: Mapped[Optional["Goal"]] = relationship(
+        "Goal", back_populates="tasks")
+    description: Mapped[str] = mapped_column(nullable=False)
+    completed_at: Mapped[Optional[datetime]] = mapped_column(nullable=True)
+
+    def to_dict(self, include_completed_at=False, include_goal_id=False):
+        task_dict = {
+            "id": self.id,
+            "title": self.title,
+            "description": self.description,
+            "is_complete": self.completed_at is not None,
+            # "goal_id": self.goal.id if self.goal else None,  # Default to None
+            # "goal_title": self.goal.title if self.goal else None  # Default to None
+
+        }
+        if include_goal_id and self.goal_id:
+            task_dict["goal_id"] = self.goal_id
+        return task_dict
+
+    # if include_completed_at:
+    #     task_dict["completed_at"] = self.completed_at
+
+    # if include_goal_id and self.goal:
+    #     task_dict["goal_id"] = self.goal.id
+    #     task_dict["goal_title"] = self.goal.title
+    # return task_dict
+
+    @classmethod
+    def from_dict(cls, task_data):
+        return cls(
+            title=task_data["title"],
+            description=task_data["description"],
+            completed_at=task_data.get("completed_at")
+        )

app/routes/goal_routes.py

@@ -1 +1,92 @@
-from flask import Blueprint
+from flask import Blueprint, abort, make_response, request, Response
+from app.models.goal import Goal
+from app.models.task import Task
+from ..db import db
+from .route_utilities import validate_model, create_model
+
+goal_bp = Blueprint("goal_bp", __name__, url_prefix="/goals")
+
+
+@goal_bp.get("")
+def get_all_goals():
+    query = db.select(Goal)
+    goals = db.session.scalars(query)
+    goal_response = [goal.to_dict() for goal in goals]
+
+    return goal_response
+
+
+@goal_bp.get("/<id>")
+def get_one_goal(id):
+    goal = validate_model(Goal, id)
+    return {"goal": goal.to_dict()}
+
+
+@goal_bp.post("")
+def create_goal():
+    request_body = request.get_json()
+
+    if not request_body.get("title"):
+        return {"details": "Invalid data"}, 400
+
+    new_goal = Goal.from_dict(request_body)
+
+    db.session.add(new_goal)
+    db.session.commit()
+
+    return {"goal": new_goal.to_dict()}, 201
+
+
+@goal_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")
+
+
+@goal_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")
+
+
+@goal_bp.get("/<id>/tasks")
+def get_all_goal_tasks(id):
+    goal = validate_model(Goal, id)
+    tasks = [task.to_dict(include_goal_id=True) for task in goal.tasks]
+
+    return {
+        "id": goal.id,
+        "title": goal.title,
+        "tasks": tasks
+    }, 200
+
+
+@goal_bp.post("/<id>/tasks")
+def create_task_with_goal_id(id):
+    goal = validate_model(Goal, id)
+    request_body = request.get_json()
+
+    task_ids = request_body.get("task_ids")
+    if not task_ids or not isinstance(task_ids, list):
+        return {"message": "'task_ids' field is required and must be a list"}, 400
+
+    # Query all tasks by the given IDs
+    tasks = Task.query.filter(Task.id.in_(task_ids)).all()
+
+    if len(tasks) != len(task_ids):
+        return {"message": "One or more task_ids are invalid"}, 404
+
+    # Replace the tasks associated with the goal
+    goal.tasks = tasks
+    db.session.commit()
+
+    return {
+        "id": goal.id,
+        "task_ids": [task.id for task in tasks]
+    }, 200

app/routes/route_utilities.py

@@ -0,0 +1,56 @@
+from flask import abort, make_response, jsonify
+import requests
+from ..db import db
+import os
+
+
+SLACK_BOT_TOKEN = os.environ.get("SLACK_BOT_TOKEN")
+SLACK_CHANNEL = os.environ.get("SLACK_CHANNEL")
+
+
+def validate_model(cls, modelid):
+    try:
+        model_id = int(modelid)
+    except ValueError:
+        invalid_response = {
+            "message": f"{cls.__name__} id ({model_id}) is invalid."}
+        abort(make_response(invalid_response, 400))
+
+    query = db.select(cls).where(cls.id == model_id)
+    model = db.session.scalar(query)
+
+    if not model:
+        not_found = {
+            "message": f"{cls.__name__} with id ({model_id}) not found."}
+        abort(make_response(not_found, 404))
+    return model
+
+
+def create_model(cls, model_data):
+    try:
+        new_model = cls.from_dict(model_data)
+    except KeyError as e:
+        response = {"message": f"Invalid request: missing {e.args[0]}"}
+        abort(make_response(response, 400))
+
+    db.session.add(new_model)
+    db.session.commit()
+    return new_model.to_dict(), 201
+
+
+def send_slack_message(task_title):
+    url = "https://slack.com/api/chat.postMessage"
+
+    headers = {
+        "Authorization": f"Bearer {SLACK_BOT_TOKEN}",
+        "Content-Type": "application/json; charset=UTF-8"
+    }
+    payload = {
+        "channel": SLACK_CHANNEL,
+        "text": f"Someone just completed the task {task_title}"
+
+    }
+    response = requests.post(url, headers=headers, json=payload)
+
+    if not response.ok or not response.json().get("ok"):
+        print("Failed to send slack message:", response.text)

app/routes/task_routes.py

@@ -1 +1,83 @@
-from flask import Blueprint
+from flask import Blueprint, abort, make_response, request, Response
+from app.models.task import Task
+from ..db import db
+from .route_utilities import validate_model, send_slack_message
+from datetime import datetime, timezone
+
+task_bp = Blueprint("task_bp", __name__, url_prefix="/tasks")
+
+
+@task_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")
+
+
+@task_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")
+
+
+@task_bp.post("")
+def create_task():
+    request_body = request.get_json()
+
+    if not request_body.get("title") or not request_body.get("description"):
+        return {"details": "Invalid data"}, 400
+
+    new_task = Task.from_dict(request_body)
+
+    db.session.add(new_task)
+    db.session.commit()
+    return {"task": new_task.to_dict()}, 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(Task.title.asc())
+    elif sort_param == "desc":
+        query = query.order_by(Task.title.desc())
+    title_param = request.args.get("title")
+
+    if title_param:
+        query = query.where(Task.name == title_param)
+
+    tasks = db.session.scalars(query)
+    tasks_response = [task.to_dict() for task in tasks]
+    return tasks_response
+
+
+@task_bp.get("/<id>")
+def get_one_task(id):
+    task = validate_model(Task, id)
+    return {"task": task.to_dict(include_goal_id=True)}
+
+
+@task_bp.patch("/<id>/mark_complete")
+def mark_task_complete(id):
+    task = validate_model(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("/<id>/mark_incomplete")
+def mark_task_incomplete(id):
+    task = validate_model(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.

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