Update app.py

app.py

@@ -74,10 +74,21 @@ def get_required_fields(sf):
 
 def create_contract_document(sf, name, status='Draft', document_url='https://example.com/contract.pdf'):
     """
-    Create a Contract_Document__c record with the given name, status, and document URL.
-    Includes required fields Document_URL__c and Upload_Date__c.
+    Create a Contract_Document__c record if it doesn't exist.
+    Checks for existing records by Name to prevent duplicates.
     """
-    logger.info("Creating Contract_Document__c record")
+    logger.info("Checking for existing Contract_Document__c record")
+    try:
+        query = f"SELECT Id, Name, Status__c, Document_URL__c, Upload_Date__c FROM Contract_Document__c WHERE Name = '{name}' LIMIT 1"
+        result = sf.query(query)
+        if result['totalSize'] > 0:
+            logger.info(f"Record with Name '{name}' already exists: {result['records'][0]['Id']}")
+            return result['records'][0]
+    except SalesforceError as e:
+        logger.error(f"Error checking for existing record: {e}")
+        return None
+    
+    logger.info("Creating new Contract_Document__c record")
     valid_statuses = get_valid_status_values(sf)
     
     if status not in valid_statuses:
@@ -151,7 +162,7 @@ def main():
             sys.exit(1)
         
         logger.info("Application completed successfully")
-        sys.exit(0)  # Explicitly exit with success code
+        sys.exit(0)
         
     except SalesforceError as e:
         logger.error(f"Error initializing Salesforce connection: {e}")
@@ -162,4 +173,5 @@ def main():
 
 if __name__ == "__main__":
     logger.info("Application entry point")
-    main()
+    main()
+