polyapi-python 0.2.0.dev8__tar.gz → 0.2.1__tar.gz

{polyapi-python-0.2.0.dev8 → polyapi-python-0.2.1}/PKG-INFO

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

{polyapi-python-0.2.0.dev8 → polyapi-python-0.2.1}/polyapi/auth.py

@@ -15,6 +15,12 @@ GET_TOKEN_TEMPLATE = """
 import asyncio
 
 
+class AuthFunctionResponse(TypedDict):
+  status: int
+  data: Any
+  headers: Dict[str, str]
+
+
 async def getToken(clientId: str, clientSecret: str, scopes: List[str], callback, options: Optional[Dict[str, Any]] = None):
     \"""{description}
 
@@ -102,27 +108,40 @@ async def getToken(clientId: str, clientSecret: str, scopes: List[str], callback
     return {{"close": closeEventHandler}}
 """
 
+INTROSPECT_TOKEN_TEMPLATE = """
+def introspectToken(token: str) -> AuthFunctionResponse:
+    \"""{description}
+
+    Function ID: {function_id}
+    \"""
+    url = "/auth-providers/{function_id}/introspect"
+    resp = execute_post(url, {{"token": token}})
+    return resp.json()
+"""
+
 REFRESH_TOKEN_TEMPLATE = """
-def refreshToken(token: str) -> str:
+def refreshToken(token: str) -> AuthFunctionResponse:
     \"""{description}
 
     Function ID: {function_id}
     \"""
     url = "/auth-providers/{function_id}/refresh"
     resp = execute_post(url, {{"token": token}})
-    assert resp.status_code == 201, (resp.status_code, resp.content)
-    return resp.text
+    return resp.json()
 """
 
 REVOKE_TOKEN_TEMPLATE = """
-def revokeToken(token: str) -> None:
+def revokeToken(token: str) -> Optional[AuthFunctionResponse]:
     \"""{description}
 
     Function ID: {function_id}
     \"""
     url = "/auth-providers/{function_id}/revoke"
     resp = execute_post(url, {{"token": token}})
-    assert resp.status_code == 201, (resp.status_code, resp.content)
+    try:
+        return resp.json()
+    except:
+        return None
 """
 
 
@@ -147,6 +166,8 @@ def render_auth_function(
 
     if function_name == "getToken":
         func_str = GET_TOKEN_TEMPLATE.format(function_id=function_id, description=function_description, client_id=uuid.uuid4().hex)
+    elif function_name == "introspectToken":
+        func_str = INTROSPECT_TOKEN_TEMPLATE.format(function_id=function_id, description=function_description)
     elif function_name == "refreshToken":
         func_str = REFRESH_TOKEN_TEMPLATE.format(function_id=function_id, description=function_description)
     elif function_name == "revokeToken":

polyapi-python-0.2.1/polyapi/webhook.py

@@ -0,0 +1,92 @@
+import uuid
+from typing import Any, Dict, List, Tuple
+
+from polyapi.typedefs import PropertySpecification
+
+WEBHOOK_TEMPLATE = """
+import asyncio
+
+
+def {function_name}(callback, options=None):
+    \"""{description}
+
+    Function ID: {function_id}
+    \"""
+    options = options or {{}}
+    eventsClientId = "{client_id}"
+    function_id = "{function_id}"
+
+    api_key, base_url = get_api_key_and_url()
+
+    async def _inner():
+        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 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
+
+    return asyncio.run(_inner())
+"""
+
+
+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.2.0.dev8 → polyapi-python-0.2.1}/polyapi_python.egg-info/PKG-INFO

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

{polyapi-python-0.2.0.dev8 → polyapi-python-0.2.1}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["setuptools>=61.2", "wheel"]
 
 [project]
 name = "polyapi-python"
-version = "0.2.0.dev8"
+version = "0.2.1"
 description = "The PolyAPI Python Client"
 authors = [{ name = "Dan Fellin", email = "dan@polyapi.io" }]
 dependencies = [

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

@@ -1,85 +0,0 @@
-import uuid
-from typing import Any, Dict, List, Tuple
-
-from polyapi.typedefs import PropertySpecification
-
-WEBHOOK_TEMPLATE = """
-async def {function_name}(callback, options=None):
-    \"""{description}
-
-    Function ID: {function_id}
-    \"""
-    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, ""