Skip to content
5 changes: 5 additions & 0 deletions app/__init__.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
from flask import Flask
from .db import db, migrate
from .models import task, goal
from .routes.goal_routes import bp as goal_bp
from .routes.task_routes import bp as tasks_bp
import os

def create_app(config=None):
Expand All @@ -18,5 +20,8 @@ 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
26 changes: 25 additions & 1 deletion app/models/goal.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,29 @@
from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy.orm import Mapped, mapped_column, relationship
from ..db import db
from typing import TYPE_CHECKING

if TYPE_CHECKING:
from .task import Task

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")

@classmethod
def from_dict(cls, goal_data):
new_goal = cls(
title = goal_data["title"]
)
return new_goal

def to_dict(self):
goal_dict = {
"id": self.id,
"title": self.title,
}

if self.tasks:
goal_dict["tasks"] = [task.to_dict() for task in self.tasks]

return goal_dict
35 changes: 34 additions & 1 deletion app/models/task.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,38 @@
from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy.orm import Mapped, mapped_column, relationship
from sqlalchemy import DateTime, ForeignKey
from ..db import db
from datetime import datetime
from typing import Optional, TYPE_CHECKING

if TYPE_CHECKING:
from .goal import Goal

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(DateTime)
goal_id: Mapped[Optional[int]] = mapped_column(ForeignKey("goal.id"))
goal: Mapped[Optional["Goal"]] = relationship(back_populates="tasks")

@classmethod
def from_dict(cls, task_data):
new_task = cls(
title = task_data["title"],
description = task_data["description"],
completed_at = task_data.get("completed_at")
)
return new_task

def to_dict(self):
task_dict = {
"id": self.id,
"title": self.title,
"description": self.description,
"is_complete": bool(self.completed_at)
}

if self.goal_id:
task_dict["goal_id"] = self.goal_id

return task_dict
65 changes: 64 additions & 1 deletion app/routes/goal_routes.py
Original file line number Diff line number Diff line change
@@ -1 +1,64 @@
from flask import Blueprint
from flask import Blueprint, 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, get_models_with_filters, send_slack_bot_message

bp = Blueprint("goals_bp", __name__, url_prefix="/goals")

@bp.post("")
def create_goal():
request_body = request.get_json()
return create_model(Goal, request_body)

@bp.get("")
def get_all_goals():
return get_models_with_filters(Goal, request.args)

@bp.get("/<goal_id>")
def get_one_goal(goal_id):
goal = validate_model(Goal, goal_id)
return 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 Response(status=204, mimetype="application/json")


@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")

@bp.post("/<goal_id>/tasks")
def create_task_list_for_goal(goal_id):
goal = validate_model(Goal, goal_id)
request_body = request.get_json()

for task in goal.tasks:
task.goal_id = None

task_ids = request_body.get("task_ids")
for task_id in task_ids:
task = validate_model(Task, task_id)
task.goal_id = goal.id
db.session.commit()

response = {
"id": goal.id,
"task_ids": task_ids
}
return response, 200

@bp.get("/<goal_id>/tasks")
def get_task_of_one_goal(goal_id):
goal = validate_model(Goal, goal_id)
goal_dict = goal.to_dict()
goal_dict["tasks"] = [task.to_dict() for task in goal.tasks]
return goal_dict
63 changes: 63 additions & 0 deletions app/routes/route_utilities.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,63 @@
from flask import abort, make_response
from ..db import db
import requests, os
from dotenv import load_dotenv

load_dotenv()

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": "Invalid data"}
abort(make_response(response, 400))

db.session.add(new_model)
db.session.commit()

return 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

def send_slack_bot_message(message):
SLACK_API_URL = "https://slack.com/api/chat.postMessage"
SLACK_API_TOKEN = os.environ.get("TOKEN")

headers = {
"Content-type": "application/json",
"Authorization": f"Bearer {SLACK_API_TOKEN}"
}

body = {
"channel": "task-notifications",
"text": f"{message}"
}

return requests.post(SLACK_API_URL, headers=headers, json=body)
70 changes: 69 additions & 1 deletion app/routes/task_routes.py
Original file line number Diff line number Diff line change
@@ -1 +1,69 @@
from flask import Blueprint
from flask import Blueprint, request, Response
from app.models.task import Task
from ..db import db
from datetime import datetime
from sqlalchemy import desc
from .route_utilities import validate_model, create_model, get_models_with_filters, send_slack_bot_message

bp = Blueprint("tasks_bp", __name__, url_prefix="/tasks")

@bp.post("")
def create_task():
request_body = request.get_json()
return create_model(Task, request_body)

@bp.get("")
def get_all_tasks():
sort_parameter = request.args.get('sort')
query = db.select(Task)

if sort_parameter == "desc":
query = query.order_by(desc(Task.title))
else:
query = query.order_by(Task.title)

tasks = db.session.scalars(query)
task_response = [task.to_dict() for task in tasks]
return task_response

@bp.get("/<task_id>")
def get_one_task(task_id):
task = validate_model(Task, task_id)
return task.to_dict()

@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")


@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")

@bp.patch("/<task_id>/mark_complete")
def patch_task_to_complete(task_id):
task = validate_model(Task, task_id)
task.completed_at = datetime.now()
db.session.commit()

slack_message = f"Someone just completed the tast {task.title}"
send_slack_bot_message(slack_message)

return Response(status=204, mimetype="application/json")

@bp.patch("/<task_id>/mark_incomplete")
def patch_task_to_incomplete(task_id):
task = validate_model(Task, task_id)
task.completed_at = None
db.session.commit()
return Response(status=204, mimetype="application/json")
1 change: 1 addition & 0 deletions migrations/README
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Single-database configuration for Flask.
50 changes: 50 additions & 0 deletions migrations/alembic.ini
Original file line number Diff line number Diff line change
@@ -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
Loading