sfq 0.0.29__tar.gz → 0.0.30__tar.gz

{sfq-0.0.29 → sfq-0.0.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sfq
-Version: 0.0.29
+Version: 0.0.30
 Summary: Python wrapper for the Salesforce's Query API.
 Author-email: David Moruzzi <sfq.pypi@dmoruzi.com>
 Keywords: salesforce,salesforce query

{sfq-0.0.29 → sfq-0.0.30}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sfq"
-version = "0.0.29"
+version = "0.0.30"
 description = "Python wrapper for the Salesforce's Query API."
 readme = "README.md"
 authors = [{ name = "David Moruzzi", email = "sfq.pypi@dmoruzi.com" }]

{sfq-0.0.29 → sfq-0.0.30}/src/sfq/__init__.py

@@ -99,7 +99,7 @@ class SFAuth:
         access_token: Optional[str] = None,
         token_expiration_time: Optional[float] = None,
         token_lifetime: int = 15 * 60,
-        user_agent: str = "sfq/0.0.29",
+        user_agent: str = "sfq/0.0.30",
         sforce_client: str = "_auto",
         proxy: str = "_auto",
     ) -> None:
@@ -1099,6 +1099,7 @@ class SFAuth:
         insert_list: List[Dict[str, Any]],
         batch_size: int = 200,
         max_workers: int = None,
+        api_type: Literal["enterprise", "tooling", "metadata"] = "enterprise",
     ) -> Optional[Dict[str, Any]]:
         """
         Execute the Insert API to insert multiple records via SOAP calls.
@@ -1110,7 +1111,19 @@ class SFAuth:
         :return: JSON response from the insert request or None on failure.
         """
 
-        endpoint = f"/services/Soap/c/{self.api_version}"
+        endpoint = "/services/Soap/"
+        if api_type == "enterprise":
+            endpoint += f"c/{self.api_version}"
+        elif api_type == "tooling":
+            endpoint += f"T/{self.api_version}"
+        elif api_type == "metadata":
+            endpoint += f"m/{self.api_version}"
+        else:
+            logger.error(
+                "Invalid API type: %s. Must be one of: 'enterprise', 'tooling', 'metadata'.",
+                api_type,
+            )
+            return None
 
         if isinstance(insert_list, dict):
             insert_list = [insert_list]
@@ -1165,3 +1178,23 @@ class SFAuth:
         ]
 
         return combined_response or None
+
+    def _debug_cleanup_apex_logs(self):
+        """
+        This function performs cleanup operations for Apex debug logs.
+        """
+        apex_logs = self.query("SELECT Id FROM ApexLog ORDER BY LogLength DESC")
+        if apex_logs and apex_logs.get("records"):
+            log_ids = [log["Id"] for log in apex_logs["records"]]
+            if log_ids:
+                delete_response = self.cdelete(log_ids)
+                logger.debug("Deleted Apex logs: %s", delete_response)
+        else:
+            logger.debug("No Apex logs found to delete.")
+
+    def debug_cleanup(self, apex_logs: bool = True) -> None:
+        """
+        Perform cleanup operations for Apex debug logs.
+        """
+        if apex_logs:
+            self._debug_cleanup_apex_logs()

sfq-0.0.30/tests/test_debug_cleanup.py

@@ -0,0 +1,67 @@
+import os
+import sys
+from pathlib import Path
+from datetime import datetime, timedelta
+
+import pytest
+
+# --- Setup local import path ---
+project_root = Path(__file__).resolve().parents[1]
+src_path = project_root / "src"
+sys.path.insert(0, str(src_path))
+from sfq import SFAuth  # noqa: E402
+
+
+@pytest.fixture(scope="module")
+def sf_instance():
+    required_env_vars = [
+        "SF_INSTANCE_URL",
+        "SF_CLIENT_ID",
+        "SF_CLIENT_SECRET",
+        "SF_REFRESH_TOKEN",
+    ]
+
+    missing_vars = [var for var in required_env_vars if not os.getenv(var)]
+    if missing_vars:
+        pytest.fail(f"Missing required env vars: {', '.join(missing_vars)}")
+
+    sf = SFAuth(
+        instance_url=os.getenv("SF_INSTANCE_URL"),
+        client_id=os.getenv("SF_CLIENT_ID"),
+        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        refresh_token=os.getenv("SF_REFRESH_TOKEN"),
+    )
+    return sf
+
+def test_debug_cleanup(sf_instance, already_executed: bool = False):
+    """
+    Test the debug_cleanup method of SFAuth.
+    This test will check if the method can be called without errors.
+    """
+    apex_logs = sf_instance.query("SELECT Id FROM ApexLog LIMIT 1")
+    apex_log_count = len(apex_logs.get("records", []))
+    if apex_log_count == 0:
+        # OK, so we need to create an Apex log to test cleanup
+        # To do this, let's execute a simple Apex anonymous block
+        import http.client
+        conn = http.client.HTTPSConnection(sf_instance.instance_url.replace("https://", ""))
+        conn.request(
+            "GET",
+            '/services/data/v64.0/tooling/executeAnonymous/?anonymousBody=Long%20currentUnixTime%20%3D%20DateTime.now().getTime()%20%2F%201000%3B',
+            headers={
+                "Authorization": f"Bearer {sf_instance.access_token}",
+                "Content-Type": "application/json"
+            }
+        )
+        response = conn.getresponse()
+        if response.status != http.client.OK:
+            pytest.fail(f"Failed to create Apex logs: {response.reason}")
+        if already_executed:
+            pytest.fail("ApexLog creation failed, cannot evaluate Apex log test.")
+        return test_debug_cleanup(sf_instance, already_executed=True)
+
+    sf_instance.debug_cleanup(apex_logs=True)
+
+    apex_logs = sf_instance.query("SELECT Id FROM ApexLog LIMIT 1")
+    apex_log_count = len(apex_logs.get("records", []))
+    assert apex_log_count == 0, "Apex logs were not cleaned up successfully."

{sfq-0.0.29 → sfq-0.0.30}/uv.lock

@@ -3,5 +3,5 @@ requires-python = ">=3.9"
 
 [[package]]
 name = "sfq"
-version = "0.0.29"
+version = "0.0.30"
 source = { editable = "." }