Flask（Part.27）｜ 【注文管理アプリケーションのプログラミング（３）】

import os

from flask import Blueprint, redirect, render_template, request, session, url_for

from apps.common.db import db
from apps.ordersapp.forms import OrderForm
from apps.ordersapp.models import Order
from apps.productsapp.models import Category, Product
from apps.tablesapp.models import Table

orders_bp = Blueprint(
    "orders",
    __name__,
    template_folder=os.path.join(os.getcwd(), "apps", "ordersapp", "templates"),
    static_folder=os.path.join(os.getcwd(), "apps", "ordersapp", "static"),
)


# カートをテーブルIDごとにセッションから取得するヘルパー関数
def get_cart(table_id):
    if f"cart_{table_id}" not in session:
        session[f"cart_{table_id}"] = []
    return session[f"cart_{table_id}"]


def create_orders_app(app):
    """Blueprintを新規作成し、登録する関数"""
    return orders_bp


@orders_bp.route("/order/table/<int:table_id>", methods=["GET"])
def order_menu(table_id):
    table = Table.query.get_or_404(table_id)
    categories = Category.query.all()
    products = Product.query.all()
    cart = get_cart(table_id)

    # 現在のテーブルの注文履歴を取得
    orders = Order.query.filter_by(table_id=table.id).all()

    # 合計額を計算する
    total_amount = sum(order.total for order in orders)

    return render_template(
        "order_menu.html",
        products=products,
        cart=cart,
        table=table,
        categories=categories,
        total_amount=total_amount,
        orders=orders,  # 注文履歴を渡す
        title="メニュー一覧",
    )


@orders_bp.route(
    "/order/table/<int:table_id>/product/<int:product_id>", methods=["GET", "POST"]
)
def order_form(table_id, product_id):
    table = Table.query.get_or_404(table_id)
    product = Product.query.get_or_404(product_id)
    form = OrderForm()

    if form.validate_on_submit():
        quantity = form.quantity.data
        total = product.price * quantity

        # カートに追加（テーブルごとにセッションを使用）
        cart = get_cart(table_id)
        cart.append(
            {
                "product_id": product.id,
                "product_name": product.name,
                "quantity": quantity,
                "total": total,
            }
        )
        session.modified = True  # セッションを変更したことを通知
        return redirect(url_for("orders.order_menu", table_id=table.id))

    return render_template(
        "order_form.html",
        form=form,
        product=product,
        table=table,
        title="注文フォーム",
    )


@orders_bp.route("/order/cart/<int:table_id>", methods=["GET", "POST"])
def order_cart(table_id):
    table = Table.query.get_or_404(table_id)
    cart = get_cart(table_id)
    total_amount = sum(item["total"] for item in cart)

    if request.method == "POST":
        # 注文確定処理
        for item in cart:
            # 注文をデータベースに保存
            order = Order(
                table_id=table.id,
                product_id=item["product_id"],
                product_name=item["product_name"],
                quantity=item["quantity"],
                total=item["total"],
            )
            db.session.add(order)

        db.session.commit()

        # 注文後、カートをクリア
        session[f"cart_{table_id}"] = []
        session.modified = True

        # 確定後、再度注文情報を表示するためにリダイレクト
        return redirect(url_for("orders.order_menu", table_id=table.id))

    return render_template(
        "order_cart.html",
        cart=cart,
        total_amount=total_amount,
        table=table,
        title="注文確認フォーム",
    )