Create app.py

app.py

@@ -0,0 +1,62 @@
+# app.py
+import os
+import requests
+import json
+from simple_salesforce import Salesforce
+from google.oauth2 import service_account
+from googleapiclient.discovery import build
+
+# Google Drive Authentication Setup
+SCOPES = ['https://www.googleapis.com/auth/drive.readonly']
+SERVICE_ACCOUNT_FILE = 'service_account.json'
+credentials = service_account.Credentials.from_service_account_file(
+    SERVICE_ACCOUNT_FILE, scopes=SCOPES)
+drive_service = build('drive', 'v3', credentials=credentials)
+
+# Salesforce Authentication
+sf = Salesforce(
+    username='YOUR_SALESFORCE_USERNAME',
+    password='YOUR_SALESFORCE_PASSWORD',
+    security_token='YOUR_SALESFORCE_SECURITY_TOKEN',
+    domain='login'
+)
+
+# Local folder to save retrieved files
+LOCAL_FOLDER = 'retrieved_files'
+if not os.path.exists(LOCAL_FOLDER):
+    os.makedirs(LOCAL_FOLDER)
+
+def get_salesforce_files(custom_object_id):
+    query = f"SELECT Id, Title, ContentDocumentId FROM ContentDocumentLink WHERE LinkedEntityId = '{custom_object_id}'"
+    results = sf.query(query)
+    return results['records']
+
+def download_google_drive_file(file_id, file_name):
+    request = drive_service.files().get_media(fileId=file_id)
+    file_path = os.path.join(LOCAL_FOLDER, file_name)
+    with open(file_path, 'wb') as f:
+        downloader = MediaIoBaseDownload(f, request)
+        done = False
+        while done is False:
+            status, done = downloader.next_chunk()
+            print(f"Download progress: {int(status.progress() * 100)}%")
+    return file_path
+
+def save_salesforce_files(custom_object_id):
+    files = get_salesforce_files(custom_object_id)
+    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
+        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}")
+
+if __name__ == "__main__":
+    # Replace 'YOUR_CUSTOM_OBJECT_ID' with the Salesforce custom object Id
+    custom_object_id = 'YOUR_CUSTOM_OBJECT_ID'
+    save_salesforce_files(custom_object_id)