polyapi-python 0.2.0.dev8__py3-none-any.whl → 0.2.1__py3-none-any.whl

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/webhook.py

@@ -4,7 +4,10 @@ from typing import Any, Dict, List, Tuple
 from polyapi.typedefs import PropertySpecification
 
 WEBHOOK_TEMPLATE = """
-async def {function_name}(callback, options=None):
+import asyncio
+
+
+def {function_name}(callback, options=None):
     \"""{description}
 
     Function ID: {function_id}
@@ -14,56 +17,60 @@ async def {function_name}(callback, options=None):
     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 _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 closeEventHandler():
-        nonlocal socket
-        if not socket:
-            return
+        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")
 
-        await socket.emit('unregisterWebhookEventHandler', {{
+        data = {{
             "clientID": eventsClientId,
             "webhookHandleID": function_id,
-            "apiKey": api_key
-        }}, namespace="/events")
+            "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()
 
-    await socket.wait()
+        return closeEventHandler
 
-    return closeEventHandler
+    return asyncio.run(_inner())
 """
 
 

{polyapi_python-0.2.0.dev8.dist-info → polyapi_python-0.2.1.dist-info}/METADATA

@@ -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.dist-info → polyapi_python-0.2.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 polyapi/__init__.py,sha256=VRAaN5WZPDmj7AnprtJ3szHXWlqWInKXBfEmfpyXTK8,434
 polyapi/__main__.py,sha256=V4zhAh_YGxno5f_KSrlkELxcuDh9bR3WSd0n-2r-qQQ,93
 polyapi/api.py,sha256=Pq_OT8egmtlzMjovN5GGZXWnF5oWMkrgR0rmrzJ6ifM,1861
-polyapi/auth.py,sha256=TVqOxF7A6kycRrYCIINgwX5hBAjZNcbeGj__2hlo8qk,4824
+polyapi/auth.py,sha256=p2KSLt6q52t9gnqPmgXTOQ2_lmdFilZkIoGmQrRTPLQ,5330
 polyapi/cli.py,sha256=wMv4RpNDEz2qIQ5NjlRTop-ImfeeqCSkx22ZIrzwA50,2086
 polyapi/config.py,sha256=S8TU10upy5OW1_vX-CqQTJD-ZOB6329aMjiUCmukfUI,2292
 polyapi/constants.py,sha256=NGjso6K5rGnE8TGdrXmdEfvvr-HI3DTVGwOYiWO68LM,511
@@ -15,9 +15,9 @@ polyapi/server.py,sha256=mcbH8pG16fFMiFL52wdii0217ZLsznQnElAFEvXZ7IA,2211
 polyapi/typedefs.py,sha256=RZ3I6sgJm_5MuuORG1QjUE-UJy_z2WRXNdiWjEdLvQg,1371
 polyapi/utils.py,sha256=wZQrwSOst5wyoYYLoMk1DnAEbVsJ0-2YkmmDQ9gI3FY,5399
 polyapi/variables.py,sha256=d36-trnfTL_8m2NkorMiImb4O3UrJbiFV38CHxV5i0A,4200
-polyapi/webhook.py,sha256=d-qCnZs5d8m4x4YX-QSsvwAwWWD4ZIyhVcMFlXFolCk,2690
-polyapi_python-0.2.0.dev8.dist-info/LICENSE,sha256=Hi0kDr56Dsy0uYIwNt4r9G7tI8x8miXRTlyvbeplCP8,1068
-polyapi_python-0.2.0.dev8.dist-info/METADATA,sha256=c7IBiIXI5Rz_g6HEXDd4DdEM_8b73-BfImDGYar4uMQ,4823
-polyapi_python-0.2.0.dev8.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-polyapi_python-0.2.0.dev8.dist-info/top_level.txt,sha256=CEFllOnzowci_50RYJac-M54KD2IdAptFsayVVF_f04,8
-polyapi_python-0.2.0.dev8.dist-info/RECORD,,
+polyapi/webhook.py,sha256=A89eNnYcVpVe9doJPDLfscIhF-C7Q2AI3vu-SzGxMBg,2923
+polyapi_python-0.2.1.dist-info/LICENSE,sha256=Hi0kDr56Dsy0uYIwNt4r9G7tI8x8miXRTlyvbeplCP8,1068
+polyapi_python-0.2.1.dist-info/METADATA,sha256=jn5GF02l2EVgP9NW5D1axfkqnCKnJWi5VnHjngfx8ts,4818
+polyapi_python-0.2.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+polyapi_python-0.2.1.dist-info/top_level.txt,sha256=CEFllOnzowci_50RYJac-M54KD2IdAptFsayVVF_f04,8
+polyapi_python-0.2.1.dist-info/RECORD,,