fix

tools/verify_automation_fix.py

@@ -0,0 +1,136 @@
+#!/usr/bin/env python3
+"""
+Quick verification script for automation progress bar fix
+Tests that the AutomationService methods return correct data structures
+"""
+import sys
+import os
+
+# Add backend to path
+sys.path.insert(0, '/data/app/igny8/backend')
+os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'igny8_core.settings')
+
+import django
+django.setup()
+
+from igny8_core.business.automation.models import AutomationRun
+from igny8_core.business.automation.services import AutomationService
+
+def test_processing_state():
+    """Test that get_current_processing_state returns correct structure"""
+    print("=" * 80)
+    print("AUTOMATION PROGRESS BAR FIX - VERIFICATION TEST")
+    print("=" * 80)
+    
+    # Find a recent running or paused run
+    runs = AutomationRun.objects.filter(status__in=['running', 'paused']).order_by('-started_at')[:5]
+    
+    if not runs.exists():
+        print("\n❌ No running or paused automation runs found")
+        print("   To test: Start an automation run from the UI")
+        return
+    
+    print(f"\n✓ Found {runs.count()} active run(s)")
+    
+    for run in runs:
+        print(f"\n{'='*80}")
+        print(f"Run ID: {run.run_id}")
+        print(f"Status: {run.status}")
+        print(f"Current Stage: {run.current_stage}")
+        print(f"Started: {run.started_at}")
+        
+        try:
+            service = AutomationService.from_run_id(run.run_id)
+            state = service.get_current_processing_state()
+            
+            if state is None:
+                print("❌ get_current_processing_state() returned None")
+                print(f"   This should not happen for status='{run.status}'")
+                continue
+            
+            # Verify required fields
+            required_fields = [
+                'stage_number', 'stage_name', 'stage_type',
+                'total_items', 'processed_items', 'percentage',
+                'currently_processing', 'up_next', 'remaining_count'
+            ]
+            
+            print("\n✓ State object returned successfully")
+            print("\nField values:")
+            
+            missing_fields = []
+            for field in required_fields:
+                if field in state:
+                    value = state[field]
+                    if isinstance(value, list):
+                        print(f"  • {field}: [{len(value)} items]")
+                    else:
+                        print(f"  • {field}: {value}")
+                else:
+                    missing_fields.append(field)
+                    print(f"  ❌ {field}: MISSING")
+            
+            if missing_fields:
+                print(f"\n❌ Missing fields: {', '.join(missing_fields)}")
+            else:
+                print("\n✓ All required fields present")
+            
+            # Verify progress calculation
+            if state['total_items'] > 0:
+                expected_pct = round((state['processed_items'] / state['total_items']) * 100)
+                if state['percentage'] == expected_pct:
+                    print(f"✓ Progress calculation correct: {state['processed_items']}/{state['total_items']} = {state['percentage']}%")
+                else:
+                    print(f"❌ Progress mismatch: expected {expected_pct}%, got {state['percentage']}%")
+            
+            # Check if paused state works
+            if run.status == 'paused':
+                print("\n✓ PAUSED RUN FIX VERIFIED: State returned for paused run!")
+                print("  (Previously this would have returned None and caused blank card)")
+            
+        except Exception as e:
+            print(f"❌ Error getting state: {e}")
+            import traceback
+            traceback.print_exc()
+    
+    print("\n" + "="*80)
+    print("VERIFICATION COMPLETE")
+    print("="*80)
+    
+    # Check for trace files
+    print("\nChecking for JSONL trace files...")
+    import glob
+    trace_files = glob.glob('/data/app/logs/automation/*/*/run_*/run_trace.jsonl')
+    if trace_files:
+        print(f"✓ Found {len(trace_files)} trace file(s)")
+        latest = sorted(trace_files, key=os.path.getmtime, reverse=True)[:3]
+        print("\nMost recent trace files:")
+        for f in latest:
+            size = os.path.getsize(f)
+            print(f"  • {f} ({size} bytes)")
+            
+            # Check for stage_item_processed events
+            try:
+                with open(f, 'r') as tf:
+                    content = tf.read()
+                    if 'stage_item_processed' in content:
+                        count = content.count('stage_item_processed')
+                        print(f"    ✓ Contains {count} stage_item_processed event(s)")
+                    else:
+                        print(f"    ℹ No stage_item_processed events (may be older run)")
+            except Exception:
+                pass
+    else:
+        print("ℹ No trace files found yet (will appear for new runs)")
+    
+    print("\n" + "="*80)
+    print("NEXT STEPS:")
+    print("1. Start a new automation run from the UI")
+    print("2. Watch the progress bar - it should animate smoothly")
+    print("3. Try pausing - card should stay visible with yellow theme")
+    print("4. Check logs in: /data/app/logs/automation/<account>/<site>/<run_id>/")
+    print("5. Verify run_trace.jsonl contains 'stage_item_processed' events")
+    print("="*80)
+
+if __name__ == '__main__':
+    test_processing_state()