from django.shortcuts import render, redirect, get_object_or_404
from django.contrib.auth.decorators import login_required
from django.contrib import messages
from .models import Document
from credit_scoring.models import CreditApplication

@login_required
def upload_documents(request, app_id):
    application = get_object_or_404(CreditApplication, id=app_id, applicant=request.user)
    
    if request.method == 'POST':
        document_type = request.POST.get('document_type')
        uploaded_file = request.FILES.get('file')
        
        if uploaded_file and document_type:
            Document.objects.create(
                credit_application=application,
                document_type=document_type,
                file=uploaded_file,
                uploaded_by=request.user
            )
            messages.success(request, f'Document uploaded successfully!')
            return redirect('upload_documents', app_id=app_id)
    
    # Get existing documents
    documents = Document.objects.filter(credit_application=application)
    
    return render(request, 'documents/upload.html', {
        'application': application,
        'documents': documents,
        'document_types': Document.DOCUMENT_TYPES
    })