WYGIWYH/app/apps/transactions/views/entities.py

from django.contrib import messages
from django.contrib.auth.decorators import login_required
from django.http import HttpResponse
from django.shortcuts import render, get_object_or_404
from django.utils.translation import gettext_lazy as _
from django.views.decorators.http import require_http_methods

from apps.common.decorators.htmx import only_htmx
from apps.transactions.forms import TransactionEntityForm
from apps.transactions.models import TransactionEntity


@login_required
@require_http_methods(["GET"])
def entities_index(request):
    return render(
        request,
        "entities/pages/index.html",
    )


@only_htmx
@login_required
@require_http_methods(["GET"])
def entities_list(request):
    return render(
        request,
        "entities/fragments/list.html",
    )


@only_htmx
@login_required
@require_http_methods(["GET"])
def entities_table_active(request):
    entities = TransactionEntity.objects.filter(active=True).order_by("id")
    return render(
        request,
        "entities/fragments/table.html",
        {"entities": entities, "active": True},
    )


@only_htmx
@login_required
@require_http_methods(["GET"])
def entities_table_archived(request):
    entities = TransactionEntity.objects.filter(active=False).order_by("id")
    return render(
        request,
        "entities/fragments/table.html",
        {"entities": entities, "active": False},
    )


@only_htmx
@login_required
@require_http_methods(["GET", "POST"])
def entity_add(request, **kwargs):
    if request.method == "POST":
        form = TransactionEntityForm(request.POST)
        if form.is_valid():
            form.save()
            messages.success(request, _("Entity added successfully"))

            return HttpResponse(
                status=204,
                headers={
                    "HX-Trigger": "updated, hide_offcanvas",
                },
            )
    else:
        form = TransactionEntityForm()

    return render(
        request,
        "entities/fragments/add.html",
        {"form": form},
    )


@only_htmx
@login_required
@require_http_methods(["GET", "POST"])
def entity_edit(request, entity_id):
    entity = get_object_or_404(TransactionEntity, id=entity_id)

    if request.method == "POST":
        form = TransactionEntityForm(request.POST, instance=entity)
        if form.is_valid():
            form.save()
            messages.success(request, _("Entity updated successfully"))

            return HttpResponse(
                status=204,
                headers={
                    "HX-Trigger": "updated, hide_offcanvas",
                },
            )
    else:
        form = TransactionEntityForm(instance=entity)

    return render(
        request,
        "entities/fragments/edit.html",
        {"form": form, "entity": entity},
    )


@only_htmx
@login_required
@require_http_methods(["DELETE"])
def entity_delete(request, entity_id):
    entity = get_object_or_404(TransactionEntity, id=entity_id)

    entity.delete()

    messages.success(request, _("Entity deleted successfully"))

    return HttpResponse(
        status=204,
        headers={
            "HX-Trigger": "updated, hide_offcanvas",
        },
    )