04/30 Web Checker web app ver. 1.0

routes/admin_shifts.py

@@ -0,0 +1,131 @@
+"""
+routes/admin_shifts.py — Shift management CRUD routes.
+"""
+
+import logging
+from flask import Blueprint, render_template, request, redirect, url_for, flash, session, jsonify
+from models import (
+    get_all_shifts, get_shift_by_id, get_shift_assigned_users,
+    get_shift_assigned_websites, create_shift, update_shift, delete_shift,
+    get_all_users, get_all_websites,
+)
+from utils.decorators import admin_required
+
+logger = logging.getLogger("routes.admin_shifts")
+admin_shifts_bp = Blueprint("admin_shifts", __name__, url_prefix="/admin/shifts")
+
+# MySQL DAYOFWEEK: 1=Sun 2=Mon … 7=Sat
+DAY_MAP = [
+    ("Mon", "2"), ("Tue", "3"), ("Wed", "4"),
+    ("Thu", "5"), ("Fri", "6"), ("Sat", "7"), ("Sun", "1"),
+]
+
+
+@admin_shifts_bp.route("/")
+@admin_required
+def shifts_list():
+    shifts    = get_all_shifts()
+    all_users = get_all_users()
+    all_sites = get_all_websites()
+    return render_template("admin/shifts.html",
+                           shifts=shifts, all_users=all_users,
+                           all_sites=all_sites, day_map=DAY_MAP)
+
+
+@admin_shifts_bp.route("/<int:shift_id>/detail")
+@admin_required
+def detail(shift_id):
+    shift    = get_shift_by_id(shift_id)
+    users    = get_shift_assigned_users(shift_id)
+    websites = get_shift_assigned_websites(shift_id)
+    if not shift:
+        return jsonify({"error": "Not found"}), 404
+
+    # Convert time objects to HH:MM strings for form inputs
+    def _fmt(t):
+        if t is None:
+            return ""
+        if hasattr(t, "seconds"):          # timedelta from MySQL
+            h, rem = divmod(int(t.total_seconds()), 3600)
+            return f"{h:02d}:{rem // 60:02d}"
+        return str(t)[:5]
+
+    return jsonify({
+        "id":           shift["id"],
+        "name":         shift["name"],
+        "days_of_week": shift["days_of_week"],
+        "start_time":   _fmt(shift["start_time"]),
+        "end_time":     _fmt(shift["end_time"]),
+        "note":         shift["note"] or "",
+        "is_active":    shift["is_active"],
+        "user_ids":     [u["id"] for u in users],
+        "website_ids":  [w["id"] for w in websites],
+    })
+
+
+@admin_shifts_bp.route("/create", methods=["POST"])
+@admin_required
+def create():
+    admin_id     = session["user"]["id"]
+    name         = request.form.get("name", "").strip()
+    days_of_week = "".join(request.form.getlist("days_of_week[]"))
+    start_time   = request.form.get("start_time", "08:00")
+    end_time     = request.form.get("end_time", "17:00")
+    note         = request.form.get("note", "").strip()
+    user_ids     = [int(x) for x in request.form.getlist("user_ids[]") if x]
+    website_ids  = [int(x) for x in request.form.getlist("website_ids[]") if x]
+
+    if not name:
+        flash("Shift name is required.", "danger")
+        return redirect(url_for("admin_shifts.shifts_list"))
+
+    try:
+        create_shift(admin_id, name, days_of_week, start_time, end_time,
+                     note, user_ids, website_ids)
+        flash(f"Shift '{name}' created successfully.", "success")
+        logger.info(f"Shift '{name}' created by admin_id={admin_id}.")
+    except Exception as e:
+        logger.error(f"create_shift error: {e}")
+        flash(f"Error creating shift: {e}", "danger")
+
+    return redirect(url_for("admin_shifts.shifts_list"))
+
+
+@admin_shifts_bp.route("/<int:shift_id>/edit", methods=["POST"])
+@admin_required
+def edit(shift_id):
+    admin_id     = session["user"]["id"]
+    name         = request.form.get("name", "").strip()
+    days_of_week = "".join(request.form.getlist("days_of_week[]"))
+    start_time   = request.form.get("start_time", "08:00")
+    end_time     = request.form.get("end_time", "17:00")
+    note         = request.form.get("note", "").strip()
+    is_active    = int(request.form.get("is_active", 1))
+    user_ids     = [int(x) for x in request.form.getlist("user_ids[]") if x]
+    website_ids  = [int(x) for x in request.form.getlist("website_ids[]") if x]
+
+    try:
+        update_shift(admin_id, shift_id, name, days_of_week, start_time, end_time,
+                     note, is_active, user_ids, website_ids)
+        flash(f"Shift '{name}' updated successfully.", "success")
+        logger.info(f"Shift id={shift_id} updated by admin_id={admin_id}.")
+    except Exception as e:
+        logger.error(f"update_shift error: {e}")
+        flash(f"Error updating shift: {e}", "danger")
+
+    return redirect(url_for("admin_shifts.shifts_list"))
+
+
+@admin_shifts_bp.route("/<int:shift_id>/delete", methods=["POST"])
+@admin_required
+def delete(shift_id):
+    admin_id = session["user"]["id"]
+    try:
+        delete_shift(admin_id, shift_id)
+        flash("Shift deactivated successfully.", "success")
+        logger.info(f"Shift id={shift_id} soft-deleted by admin_id={admin_id}.")
+    except Exception as e:
+        logger.error(f"delete_shift error: {e}")
+        flash(f"Error deleting shift: {e}", "danger")
+
+    return redirect(url_for("admin_shifts.shifts_list"))