misfinanzas/finanzas/views.py

from django.shortcuts import render, redirect,  get_object_or_404
from decimal import Decimal, InvalidOperation
import logging
import re
from django.db import models as db_models

from accounts import models
from .models import Movimiento, Cuenta, MovimientoFrecuente
from .forms import MovimientoForm, FiltroForm, CuentaForm
from django.contrib.auth.decorators import login_required
from django.contrib import messages
from django.utils.timezone import now

logger = logging.getLogger(__name__)

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

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

    frecuentes = MovimientoFrecuente.objects.filter(usuario=request.user)

    form = FiltroForm(request.GET or None)
    
    if form.is_valid():
#        if form.cleaned_data['fecha_inicio']:
#            movimientos = movimientos.filter(fecha__gte=form.cleaned_data['fecha_inicio'])
#        if form.cleaned_data['fecha_fin']:
#            movimientos = movimientos.filter(fecha__lte=form.cleaned_data['fecha_fin'])
#        if form.cleaned_data['cuenta']:
#            cuenta = form.cleaned_data['cuenta']
#            movimientos = movimientos.filter(models.Q(cuenta_origen=cuenta) | models.Q(cuenta_destino=cuenta))
        busqueda = form.cleaned_data.get('buscar')
        if busqueda:
            movimientos = movimientos.filter(
                db_models.Q(descripcion__icontains=busqueda) |
                db_models.Q(cuenta_origen__nombre__icontains=busqueda) |
                db_models.Q(cuenta_destino__nombre__icontains=busqueda)
            )
    logger.warning(f"Frecuentes: {frecuentes}")
    logger.warning(f"Monto: {frecuentes[0].monto}")
    # Aplicar filtros si se han proporcionado

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

# Movimientos

@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()
            messages.success(request, 'Movimiento registrado con éxito.')

            return redirect('dashboard')
    else:
        datos_iniciales = {}
        origen = request.GET.get('origen')
        destino = request.GET.get('destino')
        monto_raw = request.GET.get('monto')

        try:
            if origen:
                datos_iniciales['cuenta_origen'] = Cuenta.objects.get(nombre=request.GET['origen'], usuario=request.user)
            if destino:
                datos_iniciales['cuenta_destino'] = Cuenta.objects.get(nombre=request.GET['destino'], usuario=request.user)
            if monto_raw:
                # Reemplaza coma por punto, elimina símbolos y espacios
                monto_normalizado = re.sub(r'[^\d,.-]', '', monto_raw).replace(',', '.')
                try:
                    datos_iniciales['monto'] = str(Decimal(monto_normalizado).quantize(Decimal('0.01')))
                except InvalidOperation:
                    pass  # Ignorar monto malformado
                    logger.warning(f"monto: {datos_iniciales['monto']}")
        except (Cuenta.DoesNotExist, InvalidOperation):
            pass

        logger.warning(f"datos iniciales: {datos_iniciales}")

        form = MovimientoForm(initial=datos_iniciales)
        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 editar_movimiento(request, movimiento_id):
    movimiento = get_object_or_404(Movimiento, id=movimiento_id, usuario=request.user)
    saldo_original_origen = movimiento.cuenta_origen
    saldo_original_destino = movimiento.cuenta_destino
    monto_original = movimiento.monto

    if request.method == 'POST':
        form = MovimientoForm(request.POST, instance=movimiento)
        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_nuevo = form.save(commit=False)

            # Revertir efecto anterior
            saldo_original_origen.saldo += monto_original
            saldo_original_origen.save()
            saldo_original_destino.saldo -= monto_original
            saldo_original_destino.save()

            # Aplicar nuevo efecto
            movimiento_nuevo.usuario = request.user
            movimiento_nuevo.save()
            movimiento_nuevo.cuenta_origen.saldo -= movimiento_nuevo.monto
            movimiento_nuevo.cuenta_origen.save()
            movimiento_nuevo.cuenta_destino.saldo += movimiento_nuevo.monto
            movimiento_nuevo.cuenta_destino.save()

            return redirect('dashboard')
    else:
        form = MovimientoForm(instance=movimiento)
        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, 'editar_movimiento.html', {'form': form})


@login_required
def eliminar_movimiento(request, movimiento_id):
    movimiento = get_object_or_404(Movimiento, id=movimiento_id, usuario=request.user)
    if request.method == 'POST':
        # Revertir efecto del movimiento antes de borrarlo
        movimiento.cuenta_origen.saldo += movimiento.monto
        movimiento.cuenta_origen.save()
        movimiento.cuenta_destino.saldo -= movimiento.monto
        movimiento.cuenta_destino.save()
        movimiento.delete()
        return redirect('dashboard')

    return render(request, 'eliminar_movimiento.html', {'movimiento': movimiento})


@login_required
def duplicar_movimiento(request, movimiento_id):
    original = get_object_or_404(Movimiento, id=movimiento_id, usuario=request.user)

    # Crear una copia del movimiento sin guardar
    movimiento_nuevo = Movimiento(
        cuenta_origen=original.cuenta_origen,
        cuenta_destino=original.cuenta_destino,
        monto=original.monto,
        fecha=now(),
        descripcion=original.descripcion,
        usuario=request.user,
    )

    if request.method == 'POST':
        form = MovimientoForm(request.POST, instance=movimiento_nuevo)
        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():
            nuevo = form.save(commit=False)
            nuevo.usuario = request.user
            nuevo.save()
            # Ajustar saldos
            nuevo.cuenta_origen.saldo -= nuevo.monto
            nuevo.cuenta_origen.save()
            nuevo.cuenta_destino.saldo += nuevo.monto
            nuevo.cuenta_destino.save()
            messages.success(request, 'Movimiento duplicado con éxito.')
            return redirect('dashboard')
    else:
        form = MovimientoForm(instance=movimiento_nuevo)
        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, 'duplicar_movimiento.html', {'form': form})

# Cuentas

@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,
    })