Update app.py

app.py

@@ -17,7 +17,7 @@ drive_service = build('drive', 'v3', credentials=credentials)
 
 # Salesforce Authentication
 sf = Salesforce(
-    username='bharat@sathkrutha.sandbox',
+    username='bharatmali@sathkrutha.com',
     password='Mahadev@9',
     security_token='q67QF1yG0qBTSaiEfDrRrIEx',
     domain='login'
@@ -37,7 +37,7 @@ def get_all_salesforce_files(custom_object_api_name):
 
     # Step 2: Loop through each record ID and get linked files
     for record_id in record_ids:
-        query_files = f"SELECT Id, Title, ContentDocumentId FROM ContentDocumentLink WHERE LinkedEntityId = '{record_id}'"
+        query_files = f"SELECT Id, ContentDocumentId FROM ContentDocumentLink WHERE LinkedEntityId = '{record_id}'"
         results = sf.query(query_files)
         
         if results['records']:
@@ -63,9 +63,19 @@ def save_salesforce_files(custom_object_api_name):
     files = get_all_salesforce_files(custom_object_api_name)
     for file in files:
         content_doc_id = file['ContentDocumentId']
-        content_doc = sf.ContentDocument.get(content_doc_id)
-        google_drive_id = content_doc['Title']  # Assuming Title is the Google Drive file ID
-        google_drive_file_name = content_doc['Title'] + '.jpg'  # Replace '.jpg' with actual extension
+        
+        # Query ContentDocument to get the Title
+        content_doc_query = f"SELECT Title, FileExtension FROM ContentDocument WHERE Id = '{content_doc_id}'"
+        content_doc_result = sf.query(content_doc_query)
+
+        if not content_doc_result['records']:
+            print(f"No details found for ContentDocumentId: {content_doc_id}")
+            continue
+        
+        content_doc = content_doc_result['records'][0]
+        google_drive_id = content_doc_id  # Assuming ContentDocumentId is the Google Drive file ID
+        google_drive_file_name = content_doc['Title'] + '.' + content_doc['FileExtension']  # Use Title and FileExtension
+        
         print(f"Downloading file {google_drive_file_name} from Google Drive...")
         file_path = download_google_drive_file(google_drive_id, google_drive_file_name)
         if os.path.exists(file_path):
@@ -77,4 +87,3 @@ if __name__ == "__main__":
     # Replace 'YOUR_CUSTOM_OBJECT_API_NAME' with the Salesforce custom object API name
     custom_object_api_name = 'CustomerInfo__c'
     save_salesforce_files(custom_object_api_name)
-