Ajay98's picture
Update app.py
f4fded3 verified
raw
history blame
1.29 kB
from simple_salesforce import Salesforce
import requests
import os
# Salesforce credentials
sf = Salesforce(username='alwaysajay411@gmail.com',
password='Anusha@310819',
security_token='qTNjQknSUHtbDAjkgGKR02SR')
# 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']
# 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)
# 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}')