Update app.py

app.py

@@ -41,14 +41,14 @@ def calculate_scores(vendor_log):
     """Calculate scores based on Vendor Log data."""
     try:
         # Timeliness Score: Based on Delay Days
-        delay_days = vendor_log['Delay_Days__c']
+        delay_days = vendor_log['Delay_Days__c'] if vendor_log['Delay_Days__c'] is not None else 0
         timeliness_score = 100.0 if delay_days <= 0 else 80.0 if delay_days <= 3 else 60.0 if delay_days <= 7 else 40.0
 
         # Quality Score: Direct mapping from Quality Percentage
-        quality_score = float(vendor_log['Quality_Percentage__c'])
+        quality_score = float(vendor_log['Quality_Percentage__c']) if vendor_log['Quality_Percentage__c'] is not None else 0.0
 
         # Safety Score: Based on Incident Severity
-        incident_severity = vendor_log['Incident_Severity__c']
+        incident_severity = vendor_log['Incident_Severity__c'] if vendor_log['Incident_Severity__c'] is not None else 'None'
         safety_score = {
             'None': 100.0,
             'Low': 80.0,
@@ -58,7 +58,7 @@ def calculate_scores(vendor_log):
         }.get(incident_severity, 100.0)
 
         # Communication Score: Based on Communication Count (scale 0-100)
-        comm_count = vendor_log['Communication_Count__c']
+        comm_count = vendor_log['Communication_Count__c'] if vendor_log['Communication_Count__c'] is not None else 0
         communication_score = min(100.0, comm_count * 10.0)  # 10 communications = 100%
 
         # Round scores
@@ -70,14 +70,14 @@ def calculate_scores(vendor_log):
         }
         return scores
     except Exception as e:
-        logger.error(f"Error calculating scores for Vendor Log {vendor_log['Vendor_Log_ID__c']}: {str(e)}")
+        logger.error(f"Error calculating scores for Vendor Log {vendor_log.get('Vendor_Log_ID__c', 'Unknown')}: {str(e)}")
         raise
 
 def generate_pdf(vendor_log, scores):
     """Generate a PDF report for the vendor."""
     try:
-        vendor_id = vendor_log['Vendor__r']['Name']
-        vendor_name = vendor_log['Vendor__r']['Vendor_Name__c']
+        vendor_id = vendor_log['Vendor__r']['Name'] if vendor_log['Vendor__r'] and vendor_log['Vendor__r']['Name'] else 'Unknown'
+        vendor_name = vendor_log['Vendor__r']['Vendor_Name__c'] if vendor_log['Vendor__r'] and vendor_log['Vendor__r']['Vendor_Name__c'] else 'Unknown'
         filename = f'report_{vendor_id}_{datetime.now().strftime("%Y%m%d_%H%M%S")}.pdf'
         c = canvas.Canvas(filename, pagesize=letter)
         c.setFont('Helvetica', 12)
@@ -102,7 +102,7 @@ def generate_pdf(vendor_log, scores):
 
         return pdf_base64
     except Exception as e:
-        logger.error(f"Error generating PDF for Vendor Log {vendor_log['Vendor_Log_ID__c']}: {str(e)}")
+        logger.error(f"Error generating PDF for Vendor Log {vendor_log.get('Vendor_Log_ID__c', 'Unknown')}: {str(e)}")
         raise
 
 def fetch_and_process_logs():
@@ -127,6 +127,7 @@ def fetch_and_process_logs():
         query = query.replace(':currentMonth', str(current_month)).replace(':currentYear', str(current_year))
         result = sf.query(query)
         vendor_logs = result['records']
+        logger.info(f"Fetched {len(vendor_logs)} Vendor Log records for {current_month}-{current_year}")
 
         subcontractor_scores = []
         for log in vendor_logs:
@@ -168,7 +169,7 @@ def fetch_and_process_logs():
                 'scores': scores,
                 'pdf_link': pdf_link
             })
-        logger.info(f"Processed {len(vendor_logs)} vendor logs.")
+        logger.info(f"Processed {len(vendor_logs)} vendor logs and created {len(subcontractor_scores)} score records.")
     except Exception as e:
         logger.error(f"Error processing vendor logs: {str(e)}")
         raise