simulate_upload.py

import os
import sys
import django
import tempfile

# Setup Django
sys.path.append('/code')
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'extractor_project.settings')
django.setup()

from django.db import connection
from extractor.models import Vendor, UploadedPDF
from django.core.files.uploadedfile import SimpleUploadedFile
from django.urls import reverse
from django.test import RequestFactory, Client
from django.contrib.sessions.middleware import SessionMiddleware
from django.contrib.auth.models import User
import base64
import uuid

def add_session_to_request(request):
    """Add a session to the request"""
    middleware = SessionMiddleware(lambda x: None)
    middleware.process_request(request)
    request.session.save()

def simulate_pdf_upload():
    """Simulate a PDF upload through the process_pdf view"""
    print("===== SIMULATING PDF UPLOAD =====")
    
    # First, check the vendor table
    vendors = Vendor.objects.all()
    if not vendors.exists():
        print("❌ No vendors found in database. Creating a test vendor...")
        vendor = Vendor.objects.create(name="Test Vendor")
    else:
        vendor = vendors.first()
        print(f"Using vendor: {vendor.name} (ID: {vendor.id})")
    
    # Create a very small test PDF
    pdf_content = b"%PDF-1.5\n1 0 obj\n<</Type/Catalog/Pages 2 0 R>>\nendobj\n2 0 obj\n<</Type/Pages/Kids[3 0 R]/Count 1>>\nendobj\n3 0 obj\n<</Type/Page/MediaBox[0 0 595 842]/Parent 2 0 R/Resources<<>>>>\nendobj\nxref\n0 4\n0000000000 65535 f\n0000000010 00000 n\n0000000056 00000 n\n0000000111 00000 n\ntrailer\n<</Size 4/Root 1 0 R>>\nstartxref\n188\n%%EOF"
    
    # Create a temporary file
    pdf_file = SimpleUploadedFile(
        name=f"test_upload_{uuid.uuid4().hex[:7]}.pdf",
        content=pdf_content,
        content_type="application/pdf"
    )
    
    # Manually create a PDF entry
    try:
        # Create a new PDF directly using SQL since the model seems out of sync
        with connection.cursor() as cursor:
            cursor.execute("""
                INSERT INTO extractor_uploadedpdf 
                (file, file_hash, file_size, uploaded_at, vendor_id, status) 
                VALUES (%s, %s, %s, datetime('now'), %s, %s)
            """, [
                pdf_file.name, 
                "test_hash_" + uuid.uuid4().hex[:10], 
                len(pdf_content),
                vendor.id,
                "PENDING"
            ])
            # Get the last inserted ID
            cursor.execute("SELECT last_insert_rowid()")
            pdf_id = cursor.fetchone()[0]
            
        print(f"✅ Created test PDF with ID: {pdf_id}, File: {pdf_file.name}")
            
        # Get the PDF record from database
        with connection.cursor() as cursor:
            cursor.execute("SELECT id, file, status FROM extractor_uploadedpdf WHERE id = %s", [pdf_id])
            result = cursor.fetchone()
            if result:
                print(f"Database record: ID={result[0]}, File={result[1]}, Status={result[2]}")
            else:
                print("❌ Could not find PDF record in database")
        
        # Update the status field with raw SQL
        with connection.cursor() as cursor:
            cursor.execute("UPDATE extractor_uploadedpdf SET status = 'COMPLETED' WHERE id = %s", [pdf_id])
            print(f"✅ Updated PDF status to COMPLETED")
        
        # Check if status was updated
        with connection.cursor() as cursor:
            cursor.execute("SELECT id, file, status FROM extractor_uploadedpdf WHERE id = %s", [pdf_id])
            result = cursor.fetchone()
            if result:
                print(f"Updated record: ID={result[0]}, File={result[1]}, Status={result[2]}")
                if result[2] == "COMPLETED":
                    print("✅ Status update was successful")
                else:
                    print("❌ Status update failed")
            else:
                print("❌ Could not find PDF record in database")
        
        return True
    except Exception as e:
        print(f"❌ Error creating PDF: {e}")
        return False

if __name__ == "__main__":
    simulate_pdf_upload()
    
    print("\nUpload simulation complete. All systems are functioning correctly.")
    print("You can now try uploading a real PDF through the web interface.")