polyapi-python 0.1.0.dev14__tar.gz → 0.1.1__tar.gz

{polyapi-python-0.1.0.dev14/polyapi_python.egg-info → polyapi-python-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyapi-python
-Version: 0.1.0.dev14
+Version: 0.1.1
 Summary: The PolyAPI Python Client
 Author-email: Dan Fellin <dan@polyapi.io>
 License: MIT License
@@ -34,6 +34,8 @@ Requires-Dist: jsonschema-gentypes
 Requires-Dist: pydantic>=2.5.3
 Requires-Dist: stdlib_list
 Requires-Dist: colorama
+Requires-Dist: python-socketio[asyncio_client]
+Requires-Dist: truststore
 
 # PolyAPI Python Library
 

polyapi-python-0.1.1/polyapi/auth.py

@@ -0,0 +1,153 @@
+from typing import List, Dict, Any, Tuple
+import uuid
+
+from polyapi.typedefs import PropertySpecification
+from polyapi.utils import parse_arguments, get_type_and_def
+
+
+AUTH_DEFS_TEMPLATE = """
+from typing import List, Dict, Any, TypedDict, Optional
+{args_def}
+{return_type_def}
+"""
+
+GET_TOKEN_TEMPLATE = """
+import asyncio
+import socketio  # type: ignore
+import truststore
+truststore.inject_into_ssl()
+from polyapi.config import get_api_key_and_url
+
+
+async def getToken(clientId: str, clientSecret: str, scopes: List[str], callback, options: Optional[Dict[str, Any]] = None):
+    {description}
+    eventsClientId = "{client_id}"
+    function_id = "{function_id}"
+
+    options = options or {{}}
+    path = "/auth-providers/{function_id}/execute"
+    data = {{
+        "clientId": clientId,
+        "clientSecret": clientSecret,
+        "scopes": scopes,
+        "audience": options.get("audience"),
+        "callbackUrl": options.get("callbackUrl"),
+        "userId": options.get("userId"),
+    }}
+    resp = execute_post(path, data)
+    data = resp.json()
+    assert resp.status_code == 201, (resp.status_code, resp.content)
+
+    token = data.get("token")
+    url = data.get("url")
+    error = data.get("error")
+    if token:
+        return callback(token, url, error)
+    elif url and options.get("autoCloseOnUrl"):
+        return callback(token, url, error)
+
+    timeout = options.get("timeout", 120)
+
+    api_key, base_url = get_api_key_and_url()
+    socket = socketio.AsyncClient()
+    await socket.connect(base_url, transports=['websocket'], namespaces=['/events'])
+
+    async def closeEventHandler():
+        nonlocal socket
+        if not socket:
+            return
+
+        del socket.handlers['/events']['handleAuthFunctionEvent:{function_id}']
+        await socket.emit('unregisterAuthFunctionEventHandler', {{
+            "clientID": eventsClientId,
+            "functionId": function_id,
+            "apiKey": api_key
+        }}, namespace="/events")
+        await socket.disconnect()
+        socket = None
+
+
+    async def waitUntilTimeout(timeout):
+        await asyncio.sleep(timeout)
+        await closeEventHandler()
+
+
+    async def handleEvent(data):
+        nonlocal options
+        callback(data.get('token'), data.get('url'), data.get('error'))
+        if data.get('token') and options.get("autoCloseOnToken", True):
+            await closeEventHandler()
+
+
+    def registerCallback(registered: bool):
+        nonlocal socket
+        if registered:
+            socket.on('handleAuthFunctionEvent:{function_id}', handleEvent, namespace="/events")
+            callback(data.get('token'), data.get('url'), data.get('error'))
+
+    data2 = {{
+        "clientID": eventsClientId,
+        "functionId": function_id,
+        "apiKey": api_key
+    }}
+    await socket.emit('registerAuthFunctionEventHandler', data2, namespace="/events", callback=registerCallback)
+
+    # run timeout task in background
+    timeout = options.get("timeout", 120)
+    timeout_task = asyncio.create_task(waitUntilTimeout(timeout))
+
+    # cancel timeout task if socket.wait finishes before timeout up
+    await socket.wait()
+    timeout_task.cancel()
+
+    return {{"close": closeEventHandler}}
+"""
+
+REFRESH_TOKEN_TEMPLATE = """
+def refreshToken(token: str) -> str:
+    {description}
+    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
+"""
+
+REVOKE_TOKEN_TEMPLATE = """
+def revokeToken(token: str) -> None:
+    {description}
+    url = "/auth-providers/{function_id}/revoke"
+    resp = execute_post(url, {{"token": token}})
+    assert resp.status_code == 201, (resp.status_code, resp.content)
+"""
+
+
+def render_auth_function(
+    function_type: str,
+    function_name: str,
+    function_id: str,
+    function_description: str,
+    arguments: List[PropertySpecification],
+    return_type: Dict[str, Any],
+) -> Tuple[str, str]:
+    """ renders getToken, revokeToken, refreshToken as appropriate
+    """
+    args, args_def = parse_arguments(function_name, arguments)
+    return_type_name, return_type_def = get_type_and_def(return_type)  # type: ignore
+    func_type_defs = AUTH_DEFS_TEMPLATE.format(
+        args_def=args_def,
+        return_type_def=return_type_def,
+    )
+
+    func_str = ""
+
+    if function_description:
+        function_description = f'"""{function_description}"""'
+
+    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 == "refreshToken":
+        func_str = REFRESH_TOKEN_TEMPLATE.format(function_id=function_id, description=function_description)
+    elif function_name == "revokeToken":
+        func_str = REVOKE_TOKEN_TEMPLATE.format(function_id=function_id, description=function_description)
+
+    return func_str, func_type_defs

{polyapi-python-0.1.0.dev14 → polyapi-python-0.1.1}/polyapi/generate.py

@@ -235,6 +235,9 @@ def create_function(
                 add_import_to_init(full_path, next)
 
 
+# TODO create the socket and pass to create_function?
+
+
 def generate_functions(functions: List) -> None:
     for func in functions:
         create_function(*func)

{polyapi-python-0.1.0.dev14 → polyapi-python-0.1.1}/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\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\nfrom polyapi.execute import execute, execute_post, variable_get, variable_update\n\n"
 
 
 def init_the_init(full_path: str) -> None:

{polyapi-python-0.1.0.dev14 → polyapi-python-0.1.1/polyapi_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyapi-python
-Version: 0.1.0.dev14
+Version: 0.1.1
 Summary: The PolyAPI Python Client
 Author-email: Dan Fellin <dan@polyapi.io>
 License: MIT License
@@ -34,6 +34,8 @@ Requires-Dist: jsonschema-gentypes
 Requires-Dist: pydantic>=2.5.3
 Requires-Dist: stdlib_list
 Requires-Dist: colorama
+Requires-Dist: python-socketio[asyncio_client]
+Requires-Dist: truststore
 
 # PolyAPI Python Library
 

{polyapi-python-0.1.0.dev14 → polyapi-python-0.1.1}/polyapi_python.egg-info/requires.txt

@@ -4,3 +4,5 @@ jsonschema-gentypes
 pydantic>=2.5.3
 stdlib_list
 colorama
+python-socketio[asyncio_client]
+truststore

{polyapi-python-0.1.0.dev14 → polyapi-python-0.1.1}/pyproject.toml

@@ -3,10 +3,10 @@ requires = ["setuptools>=61.2", "wheel"]
 
 [project]
 name = "polyapi-python"
-version = "0.1.0.dev14"
+version = "0.1.1"
 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"]
+dependencies = ["requests", "typing_extensions", "jsonschema-gentypes", "pydantic>=2.5.3", "stdlib_list", "colorama", "python-socketio[asyncio_client]", "truststore"]
 readme = "README.md"
 license = { file = "LICENSE" }
 requires-python = ">=3.10"

polyapi-python-0.1.0.dev14/polyapi/auth.py

@@ -1,80 +0,0 @@
-from typing import List, Dict, Any, Tuple
-
-from polyapi.typedefs import PropertySpecification
-from polyapi.utils import parse_arguments, get_type_and_def
-
-
-AUTH_DEFS_TEMPLATE = """
-from typing import List, Dict, Any, TypedDict
-{args_def}
-{return_type_def}
-"""
-
-GET_TOKEN_TEMPLATE = """
-def getToken(clientId: str, clientSecret: str, scopes: List[str], callback, options: Dict[str, Any] = None):
-    {description}
-    # TODO timeout, autoCloseOnUrl, autoCloseOnToken
-    options = options or {{}}
-    url = "/auth-providers/{function_id}/execute"
-    data = {{
-        "clientId": clientId,
-        "clientSecret": clientSecret,
-        "scopes": scopes,
-        "audience": options.get("audience"),
-        "callbackUrl": options.get("callbackUrl"),
-        "userId": options.get("userId"),
-    }}
-    resp = execute_post(url, data)
-    data = resp.json()
-    assert resp.status_code == 201, (resp.status_code, resp.content)
-    return callback(data.get("token"), data.get("url"), data.get("error"))
-"""
-
-REFRESH_TOKEN_TEMPLATE = """
-def refreshToken(token: str) -> str:
-    {description}
-    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
-"""
-
-REVOKE_TOKEN_TEMPLATE = """
-def revokeToken(token: str) -> None:
-    {description}
-    url = "/auth-providers/{function_id}/revoke"
-    resp = execute_post(url, {{"token": token}})
-    assert resp.status_code == 201, (resp.status_code, resp.content)
-"""
-
-
-def render_auth_function(
-    function_type: str,
-    function_name: str,
-    function_id: str,
-    function_description: str,
-    arguments: List[PropertySpecification],
-    return_type: Dict[str, Any],
-) -> Tuple[str, str]:
-    """ renders getToken, revokeToken, refreshToken as appropriate
-    """
-    args, args_def = parse_arguments(function_name, arguments)
-    return_type_name, return_type_def = get_type_and_def(return_type)  # type: ignore
-    func_type_defs = AUTH_DEFS_TEMPLATE.format(
-        args_def=args_def,
-        return_type_def=return_type_def,
-    )
-
-    func_str = ""
-
-    if function_description:
-        function_description = f'"""{function_description}"""'
-
-    if function_name == "getToken":
-        func_str = GET_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":
-        func_str = REVOKE_TOKEN_TEMPLATE.format(function_id=function_id, description=function_description)
-
-    return func_str, func_type_defs