Update app.py

app.py

@@ -1,51 +1,59 @@
-
-import os
+from simple_salesforce import Salesforce
 import requests
-import json
-from salesforce import Salesforce
-from jwt import encode
-import base64
-from cryptography.hazmat.primitives import serialization
-from cryptography.hazmat.backends import default_backend
-
-# Salesforce credentials
-CONSUMER_KEY = os.environ['SALESFORCE_CONSUMER_KEY']
-CONSUMER_SECRET = os.environ['SALESFORCE_CONSUMER_SECRET']
-USERNAME = os.environ['SALESFORCE_USERNAME']
-PASSWORD = os.environ['SALESFORCE_PASSWORD']
-SECURITY_TOKEN = os.environ['SALESFORCE_SECURITY_TOKEN']
-
-# Custom object and field
-OBJECT_NAME = 'Guest__c'
-FIELD_NAME = 'Adhar_Card_Photo__c'
-
-# Authenticate and get access token
-def get_access_token():
-    payload = {
-        'grant_type': 'password',
-        'client_id': CONSUMER_KEY,
-        'client_secret': CONSUMER_SECRET,
-        'username': USERNAME,
-        'password': PASSWORD + SECURITY_TOKEN
-    }
-    response = requests.post('(link unavailable)', data=payload)
-    return response.json()['access_token']
-
-# Upload image to Salesforce
-def upload_image(image_path, image_name):
-    access_token = get_access_token()
-    sf = Salesforce(instance_url='(link unavailable)', 
-                    session_id=access_token)
-
-    # Read image content
-    with open(image_path, 'rb') as image_file:
-        image_content = image_file.read()
-        image_base64 = base64.b64encode(image_content).decode('utf-8')
-
-    # Create or update record
-    record = sf.query(f"SELECT Id FROM {OBJECT_NAME} WHERE Name = '{image_name}'")
-    if record['records']:
-        record_id = record['records'][0]['Id']
-        sf.update(OBJECT_NAME, {'Id': record_id, FIELD_NAME: image_base64})
+import os
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
+
+# Salesforce login credentials
+sf_username = os.getenv('SF_USERNAME')
+sf_password = os.getenv('SF_PASSWORD')
+sf_security_token = os.getenv('SF_SECURITY_TOKEN')
+
+# Ensure environment variables are set
+if not all([sf_username, sf_password, sf_security_token]):
+    raise ValueError("Please set SF_USERNAME, SF_PASSWORD, and SF_SECURITY_TOKEN environment variables.")
+
+# Folder where the files will be saved
+save_folder = r'D:\salesforce-usecase\Testimages'
+
+# Ensure the folder exists
+if not os.path.exists(save_folder):
+    os.makedirs(save_folder)
+
+# Authenticate to Salesforce (with production domain)
+sf = Salesforce(username=sf_username, password=sf_password, security_token=sf_security_token, domain='login')
+
+# Retrieve CustomerData__c records with non-null picture__c fields
+query = """
+SELECT Id, Name, picture__c
+FROM CustomerData__c
+WHERE picture__c != null
+"""
+results = sf.query_all(query)
+
+# Process each record and check the URL
+for record in results['records']:
+    picture_url = record['picture__c']
+    print(f"Picture URL for {record['Name']}: {picture_url}")
+
+    # Verify if the URL redirects to an image
+    response = requests.head(picture_url, allow_redirects=True)
+    if response.status_code == 200:
+        content_type = response.headers.get('Content-Type')
+        if content_type and 'image' in content_type:
+            print(f"Redirects to an image: {picture_url}")
+            # Download the image
+            response = requests.get(picture_url)
+            file_name = f"{record['Name']}.jpg"  # Assuming images are JPEG
+            file_path = os.path.join(save_folder, file_name)
+            with open(file_path, 'wb') as file:
+                file.write(response.content)
+                print(f"Downloaded {file_name} to {file_path}")
+        else:
+            print(f"URL does not lead to an image: {picture_url}")
     else:
-        sf.create(OBJECT_NAME, {FIELD_NAME: image_base64, 'Name': image_name})
+        print(f"Failed to access URL: {picture_url}, Status code: {response.status_code}")
+
+print("Process complete!")