routes.py

from flask import current_app, Blueprint, render_template, request, redirect, url_for
import datetime
import uuid

pages = Blueprint("habits", __name__, template_folder="templates", static_folder="static")


@pages.context_processor
def add_calc_date_range():
    def date_range(start: datetime.datetime):
        dates = [start + datetime.timedelta(days=diff) for diff in range(-3, 4)]
        return dates

    return {"date_range": date_range}


def at_midnight(date_obj):
    return datetime.datetime(date_obj.year, date_obj.month, date_obj.day)


def get_selected_date(date_str):
    if date_str:
        selected_date = at_midnight(datetime.datetime.fromisoformat(date_str))
    else:
        selected_date = at_midnight(datetime.datetime.today())
    return selected_date


@pages.route("/")
def index():
    date_str = request.args.get("date")
    selected_date = get_selected_date(date_str)

    habits_on_date = current_app.db.habits.find({"added": {"$lte": selected_date}})
    completions = [
        habit["habit"]
        for habit in current_app.db.completions.find({"date": selected_date})
    ]

    return render_template(
        "index.html",
        habits=habits_on_date,
        title="Habit Tracker - Home",
        selected_date=selected_date,
        completions=completions
    )


@pages.route("/add", methods=["GET", "POST"])
def add_habit():
    date_str = request.args.get("date")
    selected_date = get_selected_date(date_str)

    if request.method == "POST":
        current_app.db.habits.insert_one(
            {"_id": uuid.uuid4().hex, "added": selected_date, "name": request.form.get("habit")}
        )

    return render_template(
        "add_habit.html",
        title="Habit Tracker - Add Habit",
        selected_date=selected_date
    )


@pages.route("/complete", methods=["POST"])
def complete():
    date_str = request.form.get("date")
    habit = request.form.get("habitId")
    date = at_midnight(datetime.datetime.fromisoformat(date_str))
    current_app.db.completions.insert_one({"date": date, "habit": habit})

    return redirect(url_for("habits.index", date=date_str))


@pages.route("/delete", methods=["GET", "POST"])
def delete_habit():
    date_str = request.args.get("date")
    habit_id = request.form.get("habitId")

    if request.method == "POST":
        current_app.db.habits.delete_many({"_id": habit_id})

    return redirect(url_for("habits.index", date=date_str))