sfq 0.0.16__py3-none-any.whl → 0.0.17__py3-none-any.whl

sfq/__init__.py

@@ -84,9 +84,9 @@ class SFAuth:
         access_token: Optional[str] = None,
         token_expiration_time: Optional[float] = None,
         token_lifetime: int = 15 * 60,
-        user_agent: str = "sfq/0.0.16",
+        user_agent: str = "sfq/0.0.17",
         sforce_client: str = "_auto",
-        proxy: str = "auto",
+        proxy: str = "_auto",
     ) -> None:
         """
         Initializes the SFAuth with necessary parameters.
@@ -100,9 +100,9 @@ class SFAuth:
         :param access_token: The access token for the current session (default is None).
         :param token_expiration_time: The expiration time of the access token (default is None).
         :param token_lifetime: The lifetime of the access token in seconds (default is 15 minutes).
-        :param user_agent: Custom User-Agent string (default is "sfq/0.0.16").
+        :param user_agent: Custom User-Agent string (default is "sfq/0.0.17").
         :param sforce_client: Custom Application Identifier (default is user_agent).
-        :param proxy: The proxy configuration, "auto" to use environment (default is "auto").
+        :param proxy: The proxy configuration, "_auto" to use environment (default is "_auto").
         """
         self.instance_url = self._format_instance_url(instance_url)
         self.client_id = client_id
@@ -114,12 +114,12 @@ class SFAuth:
         self.token_expiration_time = token_expiration_time
         self.token_lifetime = token_lifetime
         self.user_agent = user_agent
-        self.sforce_client = quote(str(sforce_client), safe="")
+        self.sforce_client = str(sforce_client).replace(",", "")
         self._auto_configure_proxy(proxy)
         self._high_api_usage_threshold = 80
 
         if sforce_client == "_auto":
-            self.sforce_client = quote(str(user_agent), safe="")
+            self.sforce_client = user_agent
 
         if self.client_secret == "_deprecation_warning":
             warnings.warn(
@@ -134,6 +134,13 @@ class SFAuth:
             )
 
     def _format_instance_url(self, instance_url) -> str:
+        """
+        HTTPS is mandatory with Spring '21 release,
+        This method ensures that the instance URL is formatted correctly.
+
+        :param instance_url: The Salesforce instance URL.
+        :return: The formatted instance URL.
+        """
         if instance_url.startswith("https://"):
             return instance_url
         if instance_url.startswith("http://"):
@@ -144,8 +151,8 @@ class SFAuth:
         """
         Automatically configure the proxy based on the environment or provided value.
         """
-        if proxy == "auto":
-            self.proxy = os.environ.get("https_proxy")
+        if proxy == "_auto":
+            self.proxy = os.environ.get("https_proxy")  # HTTPs is mandatory
             if self.proxy:
                 logger.debug("Auto-configured proxy: %s", self.proxy)
         else:
@@ -798,3 +805,74 @@ class SFAuth:
             if isinstance(result, (dict, list))
         ]
         return combined_response or None
+
+    def _cupdate(self, update_dict: Dict[str, Any], batch_size: int = 25, max_workers: int = None) -> Optional[Dict[str, Any]]:
+        """
+        Execute the Composite Update API to update multiple records.
+
+        :param update_dict: A dictionary of keys of records to be updated, and a dictionary of field-value pairs to be updated, with a special key '_' overriding the sObject type which is otherwise inferred from the key. Example:
+            {'001aj00000C8kJhAAJ': {'Subject': 'Easily updated via SFQ'}, '00aaj000006wtdcAAA': {'_': 'CaseComment', 'IsPublished': False}, '001aj0000002yJRCAY': {'_': 'IdeaComment', 'CommentBody': 'Hello World!'}} 
+        :param batch_size: The number of records to update in each batch (default is 25).
+        :return: JSON response from the update request or None on failure.
+        """
+        allOrNone = False
+        endpoint = f"/services/data/{self.api_version}/composite"
+
+        compositeRequest_payload = []
+        sobject_prefixes = {}
+
+        for key, record in update_dict.items():
+            sobject = record.copy().pop("_", None)
+            if not sobject and not sobject_prefixes:
+                sobject_prefixes = self.get_sobject_prefixes()
+            
+            sobject = str(sobject) or str(sobject_prefixes.get(str(key[:3]), None))
+            
+            compositeRequest_payload.append(
+                {
+                    'method': 'PATCH',
+                    'url': f"/services/data/{self.api_version}/sobjects/{sobject}/{key}",
+                    'referenceId': key,
+                    'body': record,
+                }
+            )
+
+        chunks = [compositeRequest_payload[i:i+batch_size] for i in range(0, len(compositeRequest_payload), batch_size)]
+
+        def update_chunk(chunk: List[Dict[str, Any]]) -> Optional[Dict[str, Any]]:
+            payload = {
+                "allOrNone": bool(allOrNone),
+                "compositeRequest": chunk
+            }
+
+            status_code, resp_data = self._send_request(
+                method="POST",
+                endpoint=endpoint,
+                headers=self._get_common_headers(),
+                body=json.dumps(payload),
+            )
+
+            if status_code == 200:
+                logger.debug("Composite update API response without errors.")
+                return json.loads(resp_data)
+            else:
+                logger.error("Composite update API request failed: %s", status_code)
+                logger.debug("Response body: %s", resp_data)
+                return None
+
+        results = []
+        with ThreadPoolExecutor(max_workers=max_workers) as executor:
+            futures = [executor.submit(update_chunk, chunk) for chunk in chunks]
+            for future in as_completed(futures):
+                result = future.result()
+                if result:
+                    results.append(result)
+
+        combined_response = [
+            item
+            for result in results
+            for item in (result if isinstance(result, list) else [result])
+            if isinstance(result, (dict, list))
+        ]
+        
+        return combined_response or None

{sfq-0.0.16.dist-info → sfq-0.0.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sfq
-Version: 0.0.16
+Version: 0.0.17
 Summary: Python wrapper for the Salesforce's Query API.
 Author-email: David Moruzzi <sfq.pypi@dmoruzi.com>
 Keywords: salesforce,salesforce query
@@ -56,18 +56,6 @@ print(sf.query("SELECT Id FROM Account LIMIT 5"))
 print(sf.tooling_query("SELECT Id, FullName, Metadata FROM SandboxSettings LIMIT 5"))
 ```
 
-### sObject Key Prefixes
-
-```python
-# Key prefix via IDs
-print(sf.get_sobject_prefixes())
->>> {'0Pp': 'AIApplication', '6S9': 'AIApplicationConfig', '9qd': 'AIInsightAction', '9bq': 'AIInsightFeedback', '0T2': 'AIInsightReason', '9qc': 'AIInsightValue', ...}
-
-# Key prefix via names
-print(sf.get_sobject_prefixes(key_type="name"))
->>> {'AIApplication': '0Pp', 'AIApplicationConfig': '6S9', 'AIInsightAction': '9qd', 'AIInsightFeedback': '9bq', 'AIInsightReason': '0T2', 'AIInsightValue': '9qc', ...}
-```
-
 ### Composite Batch Queries
 
 ```python
@@ -94,7 +82,7 @@ for subrequest_identifer, subrequest_response in batched_response.items():
 
 ```python
 response = sf.cdelete(['07La0000000bYgj', '07La0000000bYgk', '07La0000000bYgl'])
->>> [{'id': '500aj000006wtdZAAQ', 'success': True, 'errors': []}, {'id': '500aj000006wtdaAAA', 'success': True, 'errors': []}, {'id': '500aj000006wtdbAAA', 'success': True, 'errors': []}]
+>>> [{'id': '07La0000000bYgj', 'success': True, 'errors': []}, {'id': '07La0000000bYgk', 'success': True, 'errors': []}, {'id': '07La0000000bYgl', 'success': True, 'errors': []}]
 ```
 
 ### Static Resources
@@ -110,6 +98,18 @@ print(f'Updated resource: {page}')
 sf.update_static_resource_id('081aj000009jUMXAA2', '<h1>It works!</h1>')
 ```
 
+### sObject Key Prefixes
+
+```python
+# Key prefix via IDs
+print(sf.get_sobject_prefixes())
+>>> {'0Pp': 'AIApplication', '6S9': 'AIApplicationConfig', '9qd': 'AIInsightAction', '9bq': 'AIInsightFeedback', '0T2': 'AIInsightReason', '9qc': 'AIInsightValue', ...}
+
+# Key prefix via names
+print(sf.get_sobject_prefixes(key_type="name"))
+>>> {'AIApplication': '0Pp', 'AIApplicationConfig': '6S9', 'AIInsightAction': '9qd', 'AIInsightFeedback': '9bq', 'AIInsightReason': '0T2', 'AIInsightValue': '9qc', ...}
+```
+
 ## How to Obtain Salesforce Tokens
 
 To use the `sfq` library, you'll need a **client ID** and **refresh token**. The easiest way to obtain these is by using the Salesforce CLI:

sfq-0.0.17.dist-info/RECORD

@@ -0,0 +1,6 @@
+sfq/__init__.py,sha256=qtXrbjzTQL1nU2RaMbxWUDRFt4CAUKAkwwp-_9lqavE,35361
+sfq/_cometd.py,sha256=XimQEubmJwUmbWe85TxH_cuhGvWVuiHHrVr41tguuiI,10508
+sfq/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sfq-0.0.17.dist-info/METADATA,sha256=XgKcuglHsZ1dN1rqNSgaWb-A-wf8-bStAMt2Gmj4GlE,6899
+sfq-0.0.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sfq-0.0.17.dist-info/RECORD,,

sfq-0.0.16.dist-info/RECORD

@@ -1,6 +0,0 @@
-sfq/__init__.py,sha256=8OErEOUxTDKOoj4sFyovhBXi0m918s_G1gaHerqTwzE,31967
-sfq/_cometd.py,sha256=XimQEubmJwUmbWe85TxH_cuhGvWVuiHHrVr41tguuiI,10508
-sfq/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sfq-0.0.16.dist-info/METADATA,sha256=_rW37RIboasxVhI0wJ0GkSK_IYtd7ld5MJtZLqc_Vpg,6908
-sfq-0.0.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sfq-0.0.16.dist-info/RECORD,,