nummi/nummi/main/views.py

from django.shortcuts import render, get_object_or_404, redirect
from django.http import HttpResponse
from django.contrib.auth.decorators import login_required
from django.contrib.auth import views as auth_views
from django.contrib.auth.mixins import LoginRequiredMixin
from django.views.generic import ListView
from django.core.paginator import Paginator

from .models import (
    Transaction,
    TransactionForm,
    Invoice,
    InvoiceForm,
    Category,
    CategoryForm,
    Snapshot,
    SnapshotForm,
)


@login_required
def index(request):
    _transactions = Transaction.objects.all()[:10]
    _categories = Category.objects.all()
    _snapshots = Snapshot.objects.all()[:5]

    context = {
        "transactions": _transactions,
        "categories": _categories,
        "snapshots": _snapshots,
    }
    return render(request, "main/index.html", context)


class LoginView(auth_views.LoginView):
    template_name = "main/login.html"
    next_page = "index"


class LogoutView(auth_views.LogoutView):
    next_page = "login"


class TransactionListView(LoginRequiredMixin, ListView):
    paginate_by = 20
    model = Transaction
    template_name = "main/transactions.html"
    context_object_name = "transactions"


@login_required
def transaction(request, uuid=None):
    if uuid is None:
        _transaction = Transaction()
        _invoices = []
    else:
        _transaction = get_object_or_404(Transaction, id=uuid)
        _invoices = Invoice.objects.filter(transaction=_transaction)
    return render(
        request,
        "main/transaction.html",
        {
            "transaction": _transaction,
            "form": TransactionForm(instance=_transaction),
            "invoices": _invoices,
            "invoice_form": InvoiceForm(instance=Invoice(transaction=_transaction)),
        },
    )


@login_required
def update_transaction(request, uuid):
    _transaction, _ = Transaction.objects.get_or_create(id=uuid)
    _form = TransactionForm(request.POST, instance=_transaction)
    _form.save()
    return redirect(transaction, uuid=uuid)


@login_required
def del_transaction(request, uuid):
    _transaction = get_object_or_404(Transaction, id=uuid)
    _transaction.delete()
    return redirect(index)


@login_required
def invoice(request, uuid):
    _invoice = get_object_or_404(Invoice, id=uuid)
    with _invoice.file.open() as _file:
        return HttpResponse(_file.read(), content_type="application/pdf")


@login_required
def add_invoice(request, uuid):
    _transaction = get_object_or_404(Transaction, id=uuid)
    _invoice = Invoice(transaction=_transaction)
    _form = InvoiceForm(request.POST, request.FILES, instance=_invoice)
    _form.save()
    return redirect(transaction, uuid=uuid)


@login_required
def del_invoice(request, uuid, invoice_id):
    _invoice = get_object_or_404(Invoice, id=invoice_id)
    _invoice.delete()
    return redirect(transaction, uuid=uuid)


@login_required
def category(request, uuid=None):
    if uuid is None:
        _category = Category()
        _transactions = None
    else:
        _category = get_object_or_404(Category, id=uuid)
        _transactions = Transaction.objects.filter(category=_category)
    return render(
        request,
        "main/category.html",
        {
            "category": _category,
            "form": CategoryForm(instance=_category),
            "transactions": _transactions,
        },
    )


@login_required
def update_category(request, uuid):
    _category, _ = Category.objects.get_or_create(id=uuid)
    _form = CategoryForm(request.POST, instance=_category)
    _form.save()
    return redirect(category, uuid=uuid)


@login_required
def del_category(request, uuid):
    _category = get_object_or_404(Category, id=uuid)
    _category.delete()
    return redirect(index)


@login_required
def snapshot(request, date=None):
    if date is None:
        _snapshot = Snapshot()
    else:
        _snapshot = get_object_or_404(Snapshot, date=date)
    return render(
        request,
        "main/snapshot.html",
        {
            "snapshot": _snapshot,
            "form": SnapshotForm(instance=_snapshot),
        },
    )


@login_required
def update_snapshot(request, uuid):
    try:
        _snapshot = Snapshot.objects.get(id=uuid)
    except Snapshot.DoesNotExist:
        _snapshot = Snapshot(id=uuid)
    _form = SnapshotForm(request.POST, instance=_snapshot)
    if _form.is_valid():
        _form.save()
        return redirect(snapshot, date=_snapshot.date)


@login_required
def del_snapshot(request, date):
    _snapshot = get_object_or_404(Snapshot, date=date)
    _snapshot.delete()
    return redirect(index)