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 from apps.common.models import SharedObject from apps.common.forms import SharedObjectForm @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 entity.owner and entity.owner != request.user: messages.error(request, _("Only the owner can edit this")) return HttpResponse( status=204, headers={ "HX-Trigger": "updated, hide_offcanvas", }, ) 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) if entity.owner != request.user and request.user in entity.shared_with.all(): entity.shared_with.remove(request.user) messages.success(request, _("Item no longer shared with you")) else: entity.delete() messages.success(request, _("Entity deleted successfully")) return HttpResponse( status=204, headers={ "HX-Trigger": "updated, hide_offcanvas", }, ) @only_htmx @login_required @require_http_methods(["GET"]) def entity_take_ownership(request, entity_id): entity = get_object_or_404(TransactionEntity, id=entity_id) if not entity.owner: entity.owner = request.user entity.visibility = SharedObject.Visibility.private entity.save() messages.success(request, _("Ownership taken successfully")) return HttpResponse( status=204, headers={ "HX-Trigger": "updated, hide_offcanvas", }, ) @only_htmx @login_required @require_http_methods(["GET", "POST"]) def entity_share(request, pk): obj = get_object_or_404(TransactionEntity, id=pk) if obj.owner and obj.owner != request.user: messages.error(request, _("Only the owner can edit this")) return HttpResponse( status=204, headers={ "HX-Trigger": "updated, hide_offcanvas", }, ) if request.method == "POST": form = SharedObjectForm(request.POST, instance=obj, user=request.user) if form.is_valid(): form.save() messages.success(request, _("Configuration saved successfully")) return HttpResponse( status=204, headers={ "HX-Trigger": "updated, hide_offcanvas", }, ) else: form = SharedObjectForm(instance=obj, user=request.user) return render( request, "entities/fragments/share.html", {"form": form, "object": obj}, )