polyapi-python 0.1.2__tar.gz → 0.2.0.dev1__tar.gz

{polyapi-python-0.1.2/polyapi_python.egg-info → polyapi-python-0.2.0.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyapi-python
-Version: 0.1.2
+Version: 0.2.0.dev1
 Summary: The PolyAPI Python Client
 Author-email: Dan Fellin <dan@polyapi.io>
 License: MIT License

{polyapi-python-0.1.2 → polyapi-python-0.2.0.dev1}/polyapi/auth.py

@@ -13,7 +13,6 @@ from typing import List, Dict, Any, TypedDict, Optional
 
 GET_TOKEN_TEMPLATE = """
 import asyncio
-import socketio  # type: ignore
 from polyapi.config import get_api_key_and_url
 
 

{polyapi-python-0.1.2 → polyapi-python-0.2.0.dev1}/polyapi/generate.py

@@ -5,6 +5,7 @@ import shutil
 from typing import Any, Dict, List, Tuple
 
 from polyapi.auth import render_auth_function
+from polyapi.webhook import render_webhook_handle
 
 from .typedefs import PropertySpecification, SpecificationDto, VariableSpecDto
 from .api import render_api_function
@@ -17,6 +18,7 @@ SUPPORTED_FUNCTION_TYPES = {
     "apiFunction",
     "authFunction",
     "serverFunction",
+    "webhookHandle",
 }
 
 SUPPORTED_TYPES = SUPPORTED_FUNCTION_TYPES | {"serverVariable"}
@@ -187,6 +189,15 @@ def add_function_file(
             arguments,
             return_type,
         )
+    elif function_type == "webhookHandle":
+        func_str, func_type_defs = render_webhook_handle(
+            function_type,
+            function_name,
+            function_id,
+            function_description,
+            arguments,
+            return_type,
+        )
 
     if func_str:
         # add function to init

{polyapi-python-0.1.2 → polyapi-python-0.2.0.dev1}/polyapi/utils.py

@@ -9,7 +9,7 @@ from polyapi.schema import generate_schema_types, clean_title, map_primitive_typ
 
 # 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, TypedDict, Optional\nimport requests\nfrom polyapi.execute import execute, execute_post, variable_get, variable_update\n\n"
+CODE_IMPORTS = "from typing import List, Dict, Any, TypedDict, Optional\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"
 
 
 def init_the_init(full_path: str) -> None:

polyapi-python-0.2.0.dev1/polyapi/webhook.py

@@ -0,0 +1,82 @@
+import uuid
+from typing import Any, Dict, List, Tuple
+
+from polyapi.typedefs import PropertySpecification
+
+WEBHOOK_TEMPLATE = """
+async def {function_name}(callback, options=None):
+    "{description}"
+    options = options or {{}}
+    eventsClientId = "{client_id}"
+    function_id = "{function_id}"
+
+    api_key, base_url = get_api_key_and_url()
+    socket = socketio.AsyncClient()
+    await socket.connect(base_url, transports=['websocket'], namespaces=['/events'])
+
+    def registerCallback(registered: bool):
+        nonlocal socket
+        if registered:
+            socket.on('handleWebhookEvent:{function_id}', handleEvent, namespace="/events")
+        else:
+            print("Could not set register webhook event handler for {function_id}")
+
+    async def handleEvent(data):
+        nonlocal api_key
+        nonlocal options
+        polyCustom = {{}}
+        resp = await callback(data.get("body"), data.get("headers"), data.get("params"), polyCustom)
+        if resp and options.get("waitForResponse"):
+            await socket.emit('setWebhookListenerResponse', {{
+                "webhookHandleID": function_id,
+                "apiKey": api_key,
+                "clientID": eventsClientId,
+                "executionId": data.get("executionId"),
+                "response": {{
+                    "data": resp,
+                    "statusCode": polyCustom.get("responseStatusCode", 200),
+                    "contentType": polyCustom.get("responseContentType", None),
+                }},
+            }}, namespace="/events")
+
+    data = {{
+        "clientID": eventsClientId,
+        "webhookHandleID": function_id,
+        "apiKey": api_key,
+        "waitForResponse": options.get("waitForResponse"),
+    }}
+    await socket.emit('registerWebhookEventHandler', data, namespace="/events", callback=registerCallback)
+
+    async def closeEventHandler():
+        nonlocal socket
+        if not socket:
+            return
+
+        await socket.emit('unregisterWebhookEventHandler', {{
+            "clientID": eventsClientId,
+            "webhookHandleID": function_id,
+            "apiKey": api_key
+        }}, namespace="/events")
+
+    await socket.wait()
+
+    return closeEventHandler
+"""
+
+
+def render_webhook_handle(
+    function_type: str,
+    function_name: str,
+    function_id: str,
+    function_description: str,
+    arguments: List[PropertySpecification],
+    return_type: Dict[str, Any],
+) -> Tuple[str, str]:
+    func_str = WEBHOOK_TEMPLATE.format(
+        description=function_description,
+        client_id=uuid.uuid4().hex,
+        function_id=function_id,
+        function_name=function_name,
+    )
+
+    return func_str, ""

{polyapi-python-0.1.2 → polyapi-python-0.2.0.dev1/polyapi_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyapi-python
-Version: 0.1.2
+Version: 0.2.0.dev1
 Summary: The PolyAPI Python Client
 Author-email: Dan Fellin <dan@polyapi.io>
 License: MIT License

{polyapi-python-0.1.2 → polyapi-python-0.2.0.dev1}/polyapi_python.egg-info/SOURCES.txt

@@ -18,6 +18,7 @@ polyapi/server.py
 polyapi/typedefs.py
 polyapi/utils.py
 polyapi/variables.py
+polyapi/webhook.py
 polyapi_python.egg-info/PKG-INFO
 polyapi_python.egg-info/SOURCES.txt
 polyapi_python.egg-info/dependency_links.txt

{polyapi-python-0.1.2 → polyapi-python-0.2.0.dev1}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["setuptools>=61.2", "wheel"]
 
 [project]
 name = "polyapi-python"
-version = "0.1.2"
+version = "0.2.0.dev1"
 description = "The PolyAPI Python Client"
 authors = [{ name = "Dan Fellin", email = "dan@polyapi.io" }]
 dependencies = ["requests", "typing_extensions", "jsonschema-gentypes", "pydantic>=2.5.3", "stdlib_list", "colorama", "python-socketio[asyncio_client]", "truststore"]