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'
@@ -35,7 +35,7 @@ def get_all_salesforce_files(custom_object_api_name):
     
     all_files = []
 
-    # Step 2: Loop through each record ID and get linked files
+    # Step 2: Loop through each record ID and get linked files from the Files object
     for record_id in record_ids:
         query_files = f"SELECT Id, ContentDocumentId FROM ContentDocumentLink WHERE LinkedEntityId = '{record_id}'"
         results = sf.query(query_files)
@@ -64,7 +64,7 @@ def save_salesforce_files(custom_object_api_name):
     for file in files:
         content_doc_id = file['ContentDocumentId']
         
-        # Query ContentDocument to get the Title
+        # Query ContentDocument to get the Title and FileExtension
         content_doc_query = f"SELECT Title, FileExtension FROM ContentDocument WHERE Id = '{content_doc_id}'"
         content_doc_result = sf.query(content_doc_query)
 
@@ -73,17 +73,20 @@ def save_salesforce_files(custom_object_api_name):
             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
+        file_name = content_doc['Title'] + '.' + content_doc['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):
-            print(f"File {google_drive_file_name} saved successfully to {file_path}")
-        else:
-            print(f"Failed to save file {google_drive_file_name}")
+        # Download the file from Google Drive using the ContentDocument ID (assuming the file is stored in Google Drive)
+        try:
+            print(f"Downloading file {file_name} from Google Drive...")
+            file_path = download_google_drive_file(content_doc_id, file_name)
+            if os.path.exists(file_path):
+                print(f"File {file_name} saved successfully to {file_path}")
+            else:
+                print(f"Failed to save file {file_name}")
+        except Exception as e:
+            print(f"Error downloading file {file_name}: {e}")
 
 if __name__ == "__main__":
     # Replace 'YOUR_CUSTOM_OBJECT_API_NAME' with the Salesforce custom object API name
-    custom_object_api_name = 'CustomerInfo__c'
+    custom_object_api_name = 'YOUR_CUSTOM_OBJECT_API_NAME'
     save_salesforce_files(custom_object_api_name)