from typing import Any from django.contrib.auth.decorators import login_required from django.core.paginator import Paginator from django.http import ( HttpRequest, HttpResponse, HttpResponseRedirect, ) from django.shortcuts import get_object_or_404, redirect, render from django.template.loader import render_to_string from django.urls import reverse from django.utils import timezone from django.views.decorators.http import require_POST from common.components import A, Button, Icon, LinkedPurchase, PurchasePrice from common.time import dateformat from games.forms import PurchaseForm from games.models import Game, Purchase from games.views.general import use_custom_redirect @login_required def list_purchases(request: HttpRequest) -> HttpResponse: context: dict[Any, Any] = {} page_number = request.GET.get("page", 1) limit = request.GET.get("limit", 10) purchases = Purchase.objects.order_by("-date_purchased", "-created_at") page_obj = None if int(limit) != 0: paginator = Paginator(purchases, limit) page_obj = paginator.get_page(page_number) purchases = page_obj.object_list context = { "title": "Manage purchases", "page_obj": page_obj or None, "elided_page_range": ( page_obj.paginator.get_elided_page_range( page_number, on_each_side=1, on_ends=1 ) if page_obj else None ), "data": { "header_action": A([], Button([], "Add purchase"), url="add_purchase"), "columns": [ "Name", "Type", "Price", "Infinite", "Purchased", "Refunded", "Created", "Actions", ], "rows": [ [ LinkedPurchase(purchase), purchase.get_type_display(), PurchasePrice(purchase), purchase.infinite, purchase.date_purchased.strftime(dateformat), ( purchase.date_refunded.strftime(dateformat) if purchase.date_refunded else "-" ), purchase.created_at.strftime(dateformat), render_to_string( "cotton/button_group.html", { "buttons": [ { "href": reverse( "refund_purchase", args=[purchase.pk] ), "slot": Icon("refund"), "title": "Mark as refunded", } if not purchase.date_refunded else {}, { "href": reverse( "edit_purchase", args=[purchase.pk] ), "slot": Icon("edit"), "title": "Edit", "color": "gray", }, { "href": reverse( "delete_purchase", args=[purchase.pk] ), "slot": Icon("delete"), "title": "Delete", "color": "red", }, ] }, ), ] for purchase in purchases ], }, } return render(request, "list_purchases.html", context) @login_required def add_purchase(request: HttpRequest, game_id: int = 0) -> HttpResponse: context: dict[str, Any] = {} initial = {"date_purchased": timezone.now()} if request.method == "POST": form = PurchaseForm(request.POST or None, initial=initial) if form.is_valid(): purchase = form.save() if "submit_and_redirect" in request.POST: return HttpResponseRedirect( reverse( "add_session_for_game", kwargs={"game_id": purchase.first_game.id}, ) ) else: return redirect("list_purchases") else: if game_id: game = Game.objects.get(id=game_id) form = PurchaseForm( initial={ **initial, "games": [game], "platform": game.platform, } ) else: form = PurchaseForm(initial=initial) context["form"] = form context["title"] = "Add New Purchase" context["script_name"] = "add_purchase.js" return render(request, "add_purchase.html", context) @login_required @use_custom_redirect def edit_purchase(request: HttpRequest, purchase_id: int) -> HttpResponse: context = {} purchase = get_object_or_404(Purchase, id=purchase_id) form = PurchaseForm(request.POST or None, instance=purchase) if form.is_valid(): form.save() return redirect("list_sessions") context["title"] = "Edit Purchase" context["form"] = form context["purchase_id"] = str(purchase_id) context["script_name"] = "add_purchase.js" return render(request, "add_purchase.html", context) @login_required def delete_purchase(request: HttpRequest, purchase_id: int) -> HttpResponse: purchase = get_object_or_404(Purchase, id=purchase_id) purchase.delete() return redirect("list_purchases") @login_required def view_purchase(request: HttpRequest, purchase_id: int) -> HttpResponse: purchase = get_object_or_404(Purchase, id=purchase_id) return render( request, "view_purchase.html", {"purchase": purchase, "title": f"Purchase: {purchase.full_name}"}, ) @login_required def drop_purchase(request: HttpRequest, purchase_id: int) -> HttpResponse: purchase = get_object_or_404(Purchase, id=purchase_id) purchase.date_dropped = timezone.now() purchase.save() return redirect("list_purchases") @login_required def refund_purchase(request: HttpRequest, purchase_id: int) -> HttpResponse: purchase = get_object_or_404(Purchase, id=purchase_id) purchase.date_refunded = timezone.now() purchase.save() return redirect("list_purchases") @login_required def finish_purchase(request: HttpRequest, purchase_id: int) -> HttpResponse: purchase = get_object_or_404(Purchase, id=purchase_id) purchase.date_finished = timezone.now() purchase.save() return redirect("list_purchases") def related_purchase_by_game(request: HttpRequest) -> HttpResponse: games: list[str] = [] games = request.GET.getlist("games") context = {} if games: form = PurchaseForm() qs = Purchase.objects.filter(games__in=games, type=Purchase.GAME).order_by( "games__sort_name" ) form.fields["related_purchase"].queryset = qs first_option = qs.first() if first_option: form.fields["related_purchase"].initial = first_option.id context["form"] = form return render(request, "partials/related_purchase_field.html", context) else: # abort swap return HttpResponse(status=204)