misfinanzas/finanzas/views.py

from django.shortcuts import render, redirect,  get_object_or_404

from accounts import models
from .models import Movimiento, Cuenta
from .forms import MovimientoForm, FiltroForm, CuentaForm
from django.contrib.auth.decorators import login_required

@login_required
def dashboard(request):
    movimientos = Movimiento.objects.filter(usuario=request.user).order_by('-fecha')

    favoritas = Cuenta.objects.filter(favorita='True')

    return render(request, 'dashboard.html', {
        'movimientos': movimientos,
        'user': request.user,
        'favoritas': favoritas,
    })

@login_required
def agregar_movimiento(request):
    if request.method == 'POST':
        form = MovimientoForm(request.POST)
        form.fields['cuenta_origen'].queryset = Cuenta.objects.filter(usuario=request.user)
        form.fields['cuenta_destino'].queryset = Cuenta.objects.filter(usuario=request.user)
        if form.is_valid():
            movimiento = form.save(commit=False)
            movimiento.usuario = request.user
            movimiento.save()
            # Ajustar saldos de cuentas
            movimiento.cuenta_origen.saldo -= movimiento.monto
            movimiento.cuenta_origen.save()
            movimiento.cuenta_destino.saldo += movimiento.monto
            movimiento.cuenta_destino.save()
            return redirect('dashboard')
    else:
        form = MovimientoForm()
        form.fields['cuenta_origen'].queryset = Cuenta.objects.filter(usuario=request.user)
        form.fields['cuenta_destino'].queryset = Cuenta.objects.filter(usuario=request.user)
    return render(request, 'agregar_movimiento.html', {'form': form})


@login_required
def agregar_cuenta(request):
    if request.method == 'POST':
        form = CuentaForm(request.POST)
        if form.is_valid():
            cuenta = form.save(commit=False)
            cuenta.usuario = request.user
            cuenta.save()
            return redirect('dashboard')
    else:
        form = CuentaForm()
    return render(request, 'agregar_cuenta.html', {'form': form})


@login_required
def listar_cuentas(request):
    cuentas = Cuenta.objects.filter(usuario=request.user)
    return render(request, 'listar_cuentas.html', {'cuentas': cuentas})


@login_required
def editar_cuenta(request, cuenta_id):
    cuenta = get_object_or_404(Cuenta, id=cuenta_id, usuario=request.user)
    if request.method == 'POST':
        form = CuentaForm(request.POST, instance=cuenta)
        if form.is_valid():
            form.save()
            return redirect('listar_cuentas')
    else:
        form = CuentaForm(instance=cuenta)
    return render(request, 'editar_cuenta.html', {'form': form})


@login_required
def eliminar_cuenta(request, cuenta_id):
    cuenta = get_object_or_404(Cuenta, id=cuenta_id, usuario=request.user)
    tiene_movimientos = cuenta.movimientos_salida.exists() or cuenta.movimientos_entrada.exists()

    if request.method == 'POST' and not tiene_movimientos:
        cuenta.delete()
        return redirect('listar_cuentas')

    return render(request, 'eliminar_cuenta.html', {
        'cuenta': cuenta,
        'tiene_movimientos': tiene_movimientos,
    })