polyapi-python 0.3.7.dev0__py3-none-any.whl → 0.3.7.dev2__py3-none-any.whl

polyapi/__init__.py

@@ -1,7 +1,8 @@
 import os
 import sys
 import truststore
-from typing import Dict, Any
+from typing import Any, Optional, overload, Literal
+from typing_extensions import TypedDict
 truststore.inject_into_ssl()
 from .cli import CLI_COMMANDS
 
@@ -15,9 +16,77 @@ if len(sys.argv) > 1 and sys.argv[1] not in CLI_COMMANDS:
         sys.exit(1)
 
 
-polyCustom: Dict[str, Any] = {
-    "executionId": None,
-    "executionApiKey": None,
-    "responseStatusCode": 200,
-    "responseContentType": None,
-}
+class PolyCustomDict(TypedDict, total=False):
+    """Type definition for polyCustom dictionary."""
+    executionId: Optional[str]  # Read-only
+    executionApiKey: Optional[str]
+    responseStatusCode: int
+    responseContentType: Optional[str]
+
+
+class _PolyCustom:
+    def __init__(self):
+        self._internal_store = {
+            "executionId": None,
+            "executionApiKey": None,
+            "responseStatusCode": 200,
+            "responseContentType": None,
+        }
+        self._execution_id_locked = False
+
+    def set_once(self, key: str, value: Any) -> None:
+        if key == "executionId" and self._execution_id_locked:
+            # Silently ignore attempts to overwrite locked executionId
+            return
+        self._internal_store[key] = value
+        if key == "executionId":
+            # Lock executionId after setting it
+            self.lock_execution_id()
+
+    def get(self, key: str, default: Any = None) -> Any:
+        return self._internal_store.get(key, default)
+
+    def lock_execution_id(self) -> None:
+        self._execution_id_locked = True
+
+    def unlock_execution_id(self) -> None:
+        self._execution_id_locked = False
+
+    @overload
+    def __getitem__(self, key: Literal["executionId"]) -> Optional[str]: ...
+    
+    @overload
+    def __getitem__(self, key: Literal["executionApiKey"]) -> Optional[str]: ...
+    
+    @overload
+    def __getitem__(self, key: Literal["responseStatusCode"]) -> int: ...
+    
+    @overload
+    def __getitem__(self, key: Literal["responseContentType"]) -> Optional[str]: ...
+    
+    def __getitem__(self, key: str) -> Any:
+        return self.get(key)
+
+    @overload
+    def __setitem__(self, key: Literal["executionApiKey"], value: Optional[str]) -> None: ...
+    
+    @overload
+    def __setitem__(self, key: Literal["responseStatusCode"], value: int) -> None: ...
+    
+    @overload
+    def __setitem__(self, key: Literal["responseContentType"], value: Optional[str]) -> None: ...
+    
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.set_once(key, value)
+
+    def __repr__(self) -> str:
+        return f"PolyCustom({self._internal_store})"
+
+    def copy(self) -> '_PolyCustom':
+        new = _PolyCustom()
+        new._internal_store = self._internal_store.copy()
+        new._execution_id_locked = self._execution_id_locked
+        return new
+
+
+polyCustom: PolyCustomDict = _PolyCustom()

polyapi/api.py

@@ -23,8 +23,16 @@ def {function_name}(
 
     Function ID: {function_id}
     \"""
-    resp = execute("{function_type}", "{function_id}", {data})
-    return {api_response_type}(resp.json())  # type: ignore
+    if get_direct_execute_config():
+        resp = direct_execute("{function_type}", "{function_id}", {data})
+        return {api_response_type}({{
+            "status": resp.status_code,
+            "headers": dict(resp.headers),
+            "data": resp.json()
+        }})  # type: ignore
+    else:
+        resp = execute("{function_type}", "{function_id}", {data})
+        return {api_response_type}(resp.json())  # type: ignore
 
 
 """

polyapi/config.py

@@ -8,6 +8,10 @@ from polyapi.utils import is_valid_polyapi_url, print_green, print_yellow
 # cached values
 API_KEY = None
 API_URL = None
+API_FUNCTION_DIRECT_EXECUTE = None
+MTLS_CERT_PATH = None
+MTLS_KEY_PATH = None
+MTLS_CA_PATH = None
 
 
 def get_config_file_path() -> str:
@@ -45,6 +49,13 @@ def get_api_key_and_url() -> Tuple[str | None, str | None]:
         API_KEY = key
         API_URL = url
 
+        # Read and cache MTLS and direct execute settings
+        global API_FUNCTION_DIRECT_EXECUTE, MTLS_CERT_PATH, MTLS_KEY_PATH, MTLS_CA_PATH
+        API_FUNCTION_DIRECT_EXECUTE = config.get("polyapi", "api_function_direct_execute", fallback="false").lower() == "true"
+        MTLS_CERT_PATH = config.get("polyapi", "mtls_cert_path", fallback=None)
+        MTLS_KEY_PATH = config.get("polyapi", "mtls_key_path", fallback=None)
+        MTLS_CA_PATH = config.get("polyapi", "mtls_ca_path", fallback=None)
+
     return key, url
 
 
@@ -104,4 +115,22 @@ def clear_config():
 
     path = get_config_file_path()
     if os.path.exists(path):
-        os.remove(path)
+        os.remove(path)
+
+
+def get_mtls_config() -> Tuple[bool, str | None, str | None, str | None]:
+    """Return MTLS configuration settings"""
+    global MTLS_CERT_PATH, MTLS_KEY_PATH, MTLS_CA_PATH
+    if MTLS_CERT_PATH is None or MTLS_KEY_PATH is None or MTLS_CA_PATH is None:
+        # Force a config read if values aren't cached
+        get_api_key_and_url()
+    return bool(MTLS_CERT_PATH and MTLS_KEY_PATH and MTLS_CA_PATH), MTLS_CERT_PATH, MTLS_KEY_PATH, MTLS_CA_PATH
+
+
+def get_direct_execute_config() -> bool:
+    """Return whether direct execute is enabled"""
+    global API_FUNCTION_DIRECT_EXECUTE
+    if API_FUNCTION_DIRECT_EXECUTE is None:
+        # Force a config read if value isn't cached
+        get_api_key_and_url()
+    return bool(API_FUNCTION_DIRECT_EXECUTE)

polyapi/execute.py

@@ -1,22 +1,68 @@
-from typing import Dict
+from typing import Dict, Optional
 import requests
 from requests import Response
-from polyapi.config import get_api_key_and_url
+from polyapi.config import get_api_key_and_url, get_mtls_config
 from polyapi.exceptions import PolyApiException
 
+def direct_execute(function_type, function_id, data) -> Response:
+    """ execute a specific function id/type
+    """
+    api_key, api_url = get_api_key_and_url()
+    headers = {"Authorization": f"Bearer {api_key}"}
+    url = f"{api_url}/functions/{function_type}/{function_id}/direct-execute"
+    
+    endpoint_info = requests.post(url, json=data, headers=headers)
+    if endpoint_info.status_code < 200 or endpoint_info.status_code >= 300:
+        raise PolyApiException(f"{endpoint_info.status_code}: {endpoint_info.content.decode('utf-8', errors='ignore')}")
+    
+    endpoint_info_data = endpoint_info.json()
+    request_params = endpoint_info_data.copy()
+    request_params.pop("url", None)
+    
+    if "maxRedirects" in request_params:
+        request_params["allow_redirects"] = request_params.pop("maxRedirects") > 0
+    
+    has_mtls, cert_path, key_path, ca_path = get_mtls_config()
+    
+    if has_mtls:
+        resp = requests.request(
+            url=endpoint_info_data["url"],
+            cert=(cert_path, key_path),
+            verify=ca_path,
+            **request_params
+        )
+    else:
+        resp = requests.request(
+            url=endpoint_info_data["url"],
+            verify=False,
+            **request_params
+        )
+
+    if resp.status_code < 200 or resp.status_code >= 300:
+        error_content = resp.content.decode("utf-8", errors="ignore")
+        raise PolyApiException(f"{resp.status_code}: {error_content}")
+    
+    return resp
 
 def execute(function_type, function_id, data) -> Response:
     """ execute a specific function id/type
     """
     api_key, api_url = get_api_key_and_url()
     headers = {"Authorization": f"Bearer {api_key}"}
+
     url = f"{api_url}/functions/{function_type}/{function_id}/execute"
-    resp = requests.post(url, json=data, headers=headers)
-    # print(resp.status_code)
-    # print(resp.headers["content-type"])
+    
+    # Make the request
+    resp = requests.post(
+        url, 
+        json=data, 
+        headers=headers,
+    )
+
     if resp.status_code < 200 or resp.status_code >= 300:
         error_content = resp.content.decode("utf-8", errors="ignore")
         raise PolyApiException(f"{resp.status_code}: {error_content}")
+
     return resp
 
 

polyapi/generate.py

@@ -14,7 +14,7 @@ from .api import render_api_function
 from .server import render_server_function
 from .utils import add_import_to_init, get_auth_headers, init_the_init, print_green, to_func_namespace
 from .variables import generate_variables
-from .config import get_api_key_and_url
+from .config import get_api_key_and_url, get_direct_execute_config
 
 SUPPORTED_FUNCTION_TYPES = {
     "apiFunction",
@@ -46,6 +46,10 @@ def get_specs(contexts=Optional[List[str]], no_types: bool = False) -> List:
     if contexts:
         params["contexts"] = contexts
 
+    # Add apiFunctionDirectExecute parameter if direct execute is enabled
+    if get_direct_execute_config():
+        params["apiFunctionDirectExecute"] = "true"
+
     resp = requests.get(url, headers=headers, params=params)
     if resp.status_code == 200:
         return resp.json()

polyapi/utils.py

@@ -16,7 +16,7 @@ from polyapi.schema import (
 
 # this string should be in every __init__ file.
 # it contains all the imports needed for the function or variable code to run
-CODE_IMPORTS = "from typing import List, Dict, Any, Optional, Callable\nfrom typing_extensions import TypedDict, NotRequired\nimport logging\nimport requests\nimport socketio  # type: ignore\nfrom polyapi.config import get_api_key_and_url\nfrom polyapi.execute import execute, execute_post, variable_get, variable_update\n\n"
+CODE_IMPORTS = "from typing import List, Dict, Any, Optional, Callable\nfrom typing_extensions import TypedDict, NotRequired\nimport logging\nimport requests\nimport socketio  # type: ignore\nfrom polyapi.config import get_api_key_and_url, get_direct_execute_config\nfrom polyapi.execute import execute, execute_post, variable_get, variable_update, direct_execute\n\n"
 
 
 def init_the_init(full_path: str, code_imports="") -> None:

{polyapi_python-0.3.7.dev0.dist-info → polyapi_python-0.3.7.dev2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polyapi-python
-Version: 0.3.7.dev0
+Version: 0.3.7.dev2
 Summary: The Python Client for PolyAPI, the IPaaS by Developers for Developers
 Author-email: Dan Fellin <dan@polyapi.io>
 License: MIT License

{polyapi_python-0.3.7.dev0.dist-info → polyapi_python-0.3.7.dev2.dist-info}/RECORD

@@ -1,17 +1,17 @@
-polyapi/__init__.py,sha256=a1Poy1kaTncYnUg6nWRcTjVm-R1CUQk12UX7VYQ9d5k,616
+polyapi/__init__.py,sha256=Au5snCFb05a87mDMK2Cwoft7nntR35iw4lL0Ztgivoo,2931
 polyapi/__main__.py,sha256=V4zhAh_YGxno5f_KSrlkELxcuDh9bR3WSd0n-2r-qQQ,93
-polyapi/api.py,sha256=f1037HFJF7DtQSSypM4PE5AmmxWxjd0JiW6ARZqrgac,1879
+polyapi/api.py,sha256=2nds6ZdNe9OHvCba4IjOPga0CAYIsib2SbhEyDDCmd8,2188
 polyapi/auth.py,sha256=zrIGatjba5GwUTNjKj1GHQWTEDP9B-HrSzCKbLFoqvc,5336
 polyapi/cli.py,sha256=jtKXARbT9AOgYTR6nf3OiwiPvsyUlLVbyynEA84PDzw,8924
 polyapi/client.py,sha256=CoFDYvyKsqL4wPQbUDIr0Qb8Q5eD92xN4OEEcJEVuGQ,1296
-polyapi/config.py,sha256=IGPIA3-1iSejLLly3xW7ctLzTSM_TdGoqeqlQiU5C9Q,2954
+polyapi/config.py,sha256=QQxRZ9nMUykItUMAdw97dad0DPEV1luRwkeqOyrEKf8,4316
 polyapi/constants.py,sha256=sc-FnS0SngBLvSu1ZWMs0UCf9EYD1u1Yhfr-sZXGLns,607
 polyapi/deployables.py,sha256=WVcNNB6W5ZW_-ukf_kK3moRcnwIkC-O4te6vLepjcco,11936
 polyapi/error_handler.py,sha256=I_e0iz6VM23FLVQWJljxs2NGcl_OODbi43OcbnqBlp8,2398
 polyapi/exceptions.py,sha256=Zh7i7eCUhDuXEdUYjatkLFTeZkrx1BJ1P5ePgbJ9eIY,89
-polyapi/execute.py,sha256=akfiJMIlRCPNtDfRvGd11baGC7F8gjwD3Uv8w9RVND4,1967
+polyapi/execute.py,sha256=sjI6BMBYPSCD6UngV9DzpJIRSU6p02aShNaTXhDExtY,3457
 polyapi/function_cli.py,sha256=wDbgWGSQjMpzmZoPAcU57ZwD8EUTKE8sF9dgZLUInMk,4118
-polyapi/generate.py,sha256=KdkhvkefHbSWr-XqStLrDNtE61-beIbdIClq9_y5XcI,11899
+polyapi/generate.py,sha256=ZTLtiMPYUcRygJ-s--2IzLY2FQ1W7idLJnbwh2UngKU,12089
 polyapi/parser.py,sha256=mdoh4pNq8pyiHE0-i6Coqj8frEXfBLRk6itpAXMrrgI,20373
 polyapi/poly_schemas.py,sha256=T4kfZyfgVLiqLD28GmYNiHnrNx77J_HO4uzk8LUAhlo,3137
 polyapi/prepare.py,sha256=Q8CWV4kmZ2dbXYVsud34AgJkj5ymcQ_IcYhLuikc9yk,6659
@@ -21,11 +21,11 @@ polyapi/schema.py,sha256=ZSzeUjpqigLvE4tFKB7y4AaZG-W5N5Z9wMH-F-vjMBU,4616
 polyapi/server.py,sha256=YXWxhYBx-hluwDQ8Jvfpy2s8ogz0GsNTMcZVNcP5ca8,2147
 polyapi/sync.py,sha256=PGdC0feBBjEVrF3d9EluW_OAxbWuzSrfh84czma8kWg,6476
 polyapi/typedefs.py,sha256=MGDwWaijLNqokXF9UCHGAP-yKixOzztrH4Lsj800AJs,2328
-polyapi/utils.py,sha256=vq2seXLhbp9pOvemhS71DDwi1h4YofI8TXiRDjt5Eoo,10308
+polyapi/utils.py,sha256=hW_H207-aHB22UFFYtUjxZvW13TUXTR786-3LaL4OLc,10351
 polyapi/variables.py,sha256=j7WWrGLr2O5SkWGxnsusnnfl25kVL3b6SQYcVGEoC8c,4277
 polyapi/webhook.py,sha256=LWv28c2MLz_OKBI_Nn7WR4C-gs1SWgbdXsoxIIf-9UI,4886
-polyapi_python-0.3.7.dev0.dist-info/licenses/LICENSE,sha256=6b_I7aPVp8JXhqQwdw7_B84Ca0S4JGjHj0sr_1VOdB4,1068
-polyapi_python-0.3.7.dev0.dist-info/METADATA,sha256=z7EvzD8snhdr6Br_luCmLSNGGpnwHIg-BC_SZTVpTik,5782
-polyapi_python-0.3.7.dev0.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-polyapi_python-0.3.7.dev0.dist-info/top_level.txt,sha256=CEFllOnzowci_50RYJac-M54KD2IdAptFsayVVF_f04,8
-polyapi_python-0.3.7.dev0.dist-info/RECORD,,
+polyapi_python-0.3.7.dev2.dist-info/licenses/LICENSE,sha256=6b_I7aPVp8JXhqQwdw7_B84Ca0S4JGjHj0sr_1VOdB4,1068
+polyapi_python-0.3.7.dev2.dist-info/METADATA,sha256=rx9IKDlG3qPk8IGDAn3m7Z2BCL5gS43uTtQEYnUHPm8,5782
+polyapi_python-0.3.7.dev2.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+polyapi_python-0.3.7.dev2.dist-info/top_level.txt,sha256=CEFllOnzowci_50RYJac-M54KD2IdAptFsayVVF_f04,8
+polyapi_python-0.3.7.dev2.dist-info/RECORD,,

{polyapi_python-0.3.7.dev0.dist-info → polyapi_python-0.3.7.dev2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.7.1)
+Generator: setuptools (80.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any