Update app.py

app.py

@@ -1,25 +1,33 @@
 from simple_salesforce import Salesforce
 import requests
-domain='login'
+import os
 
-# Salesforce login
-sf = Salesforce(username='alwaysajay411@gmail.com', password='Anusha@310819', security_token='qTNjQknSUHtbDAjkgGKR02SR')
+# Salesforce credentials
+sf = Salesforce(username='alwaysajay411@gmail.com', 
+                password='Anusha@310819', 
+                security_token='Anusha@310819')
 
-# URL of the image you want to download
-url = 'https://ddl00000ao4u7uaj-dev-ed.develop.my.salesforce.com/sfc/dist/version/renditionDownload?rendition=ORIGINAL_Jpg&versionId=068dL000005Qu25&operationContext=DELIVERY&contentId=05TdL0000060xRF&page=0&d=/a/dL0000002yZZ/L_lC3MGQhWHNFIOXncHjmz6jw6D4fdTgCBu6'
+# Query Salesforce custom object for image file details
+query = "SELECT Id, Name, FileExtension, ContentDocumentId FROM CustomObjectName__c WHERE Name = 'RecordName'"
+response = sf.query(query)
+records = response['records']
 
-# Adding session ID as authorization header to request the image
-headers = {
-    'Authorization': 'Bearer ' + sf.session_id
-}
+# Create a folder to save images locally if it doesn't exist
+local_folder = 'salesforce_images'
+if not os.path.exists(local_folder):
+    os.makedirs(local_folder)
 
-# Sending a request to fetch the image
-response = requests.get(url, headers=headers)
-
-# Save the image if request is successful
-if response.status_code == 200:
-    with open('john_image.jpg', 'wb') as f:
-        f.write(response.content)
-        print('Image downloaded successfully')
-else:
-    print('Failed to download the image. Status code:', response.status_code)
+# Iterate through records and download images
+for record in records:
+    file_id = record['ContentDocumentId']
+    file_details = sf.ContentDocument.get(file_id)
+    file_name = file_details['Title'] + '.' + file_details['FileExtension']
+    file_url = f'/services/data/v50.0/sobjects/ContentVersion/{file_id}/VersionData'
+    
+    # Download file content
+    file_response = requests.get(sf.base_url + file_url, headers={'Authorization': 'Bearer ' + sf.session_id})
+    
+    # Save the image file locally
+    with open(os.path.join(local_folder, file_name), 'wb') as f:
+        f.write(file_response.content)
+        print(f'Downloaded {file_name}')