polyapi-python 0.2.3.dev4__py3-none-any.whl → 0.2.3.dev6__py3-none-any.whl

polyapi/client.py

@@ -0,0 +1,25 @@
+from typing import Any, Dict, List, Tuple
+
+from polyapi.typedefs import PropertySpecification
+from polyapi.utils import camelCase, add_type_import_path, parse_arguments, get_type_and_def
+
+DEFS_TEMPLATE = """
+from typing import List, Dict, Any, TypedDict
+{args_def}
+{return_type_def}
+"""
+
+
+def render_client_function(
+    function_name: str,
+    code: str,
+    arguments: List[PropertySpecification],
+    return_type: Dict[str, Any],
+) -> Tuple[str, str]:
+    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 = DEFS_TEMPLATE.format(
+        args_def=args_def,
+        return_type_def=return_type_def,
+    )
+    return code, func_type_defs

polyapi/function_cli.py

@@ -3,7 +3,7 @@ import argparse
 import json
 import types
 import sys
-from typing import Dict, List, Tuple
+from typing import Dict, List, Mapping, Optional, Tuple
 from typing_extensions import _TypedDictMeta  # type: ignore
 import requests
 from stdlib_list import stdlib_list
@@ -18,7 +18,7 @@ import importlib
 
 # these libraries are already installed in the base docker image
 # and shouldnt be included in additional requirements
-BASE_REQUIREMENTS = {"polyapi", "requests", "typing_extensions", "jsonschema-gentypes", "pydantic"}
+BASE_REQUIREMENTS = {"polyapi", "requests", "typing_extensions", "jsonschema-gentypes", "pydantic", "cloudevents"}
 all_stdlib_symbols = stdlib_list('.'.join([str(v) for v in sys.version_info[0:2]]))
 BASE_REQUIREMENTS.update(all_stdlib_symbols)  # dont need to pip install stuff in the python standard library
 
@@ -104,6 +104,19 @@ def _get_type(expr: ast.expr | None, schemas: List[Dict]) -> Tuple[str, Dict | N
     return json_type, _get_type_schema(json_type, python_type, schemas)
 
 
+def _get_req_name_if_not_in_base(n: Optional[str], pip_name_lookup: Mapping[str, List[str]]) -> Optional[str]:
+    if not n:
+        return None
+
+    if "." in n:
+        n = n.split(".")[0]
+
+    if n in BASE_REQUIREMENTS:
+        return None
+    else:
+        return pip_name_lookup[n][0]
+
+
 def _parse_code(code: str, function_name: str):
     parsed_args = []
     return_type = None
@@ -121,13 +134,16 @@ def _parse_code(code: str, function_name: str):
 
     for node in ast.iter_child_nodes(parsed_code):
         if isinstance(node, ast.Import):
+            # TODO maybe handle `import foo.bar` case?
             for name in node.names:
-                if name.name not in BASE_REQUIREMENTS:
-                    requirements.append(name.name)
+                req = _get_req_name_if_not_in_base(name.name, pip_name_lookup)
+                if req:
+                    requirements.append(req)
         elif isinstance(node, ast.ImportFrom):
-            if node.module and node.module not in BASE_REQUIREMENTS:
-                req = pip_name_lookup[node.module][0]
-                requirements.append(req)
+            if node.module:
+                req = _get_req_name_if_not_in_base(node.module, pip_name_lookup)
+                if req:
+                    requirements.append(req)
 
         elif isinstance(node, ast.FunctionDef) and node.name == function_name:
             function_args = [arg for arg in node.args.args]
@@ -168,7 +184,8 @@ def function_add_or_update(
     args = parser.parse_args(subcommands)
 
     verb = "Updating" if _func_already_exists(context, args.function_name) else "Adding"
-    print(f"{verb} custom server side function...", end="")
+    ftype = "server" if server else "client"
+    print(f"{verb} custom {ftype} function...", end="")
 
     with open(args.filename, "r") as f:
         code = f.read()
@@ -186,9 +203,6 @@ def function_add_or_update(
         print(f"Function {args.function_name} not found as top-level function in {args.filename}")
         sys.exit(1)
 
-    if requirements:
-        print_yellow('\nPlease note that deploying your functions will take a few minutes because it makes use of libraries other than polyapi.')
-
     data = {
         "context": context,
         "name": args.function_name,
@@ -197,18 +211,20 @@ def function_add_or_update(
         "language": "python",
         "returnType": return_type,
         "returnTypeSchema": return_type_schema,
-        "requirements": requirements,
         "arguments": arguments,
         "logsEnabled": logs_enabled,
     }
 
+    if server and requirements:
+        print_yellow('\nPlease note that deploying your functions will take a few minutes because it makes use of libraries other than polyapi.')
+        data["requirements"] = requirements
+
     api_key, api_url = get_api_key_and_url()
     assert api_key
     if server:
         url = f"{api_url}/functions/server"
     else:
-        raise NotImplementedError("Client functions not yet implemented.")
-        # url = f"{base_url}/functions/client"
+        url = f"{api_url}/functions/client"
 
     headers = get_auth_headers(api_key)
     resp = requests.post(url, headers=headers, json=data)

polyapi/generate.py

@@ -2,9 +2,10 @@ import json
 import requests
 import os
 import shutil
-from typing import Any, Dict, List, Tuple
+from typing import List
 
 from polyapi.auth import render_auth_function
+from polyapi.client import render_client_function
 from polyapi.execute import execute_post
 from polyapi.webhook import render_webhook_handle
 
@@ -18,6 +19,7 @@ from .config import get_api_key_and_url, initialize_config
 SUPPORTED_FUNCTION_TYPES = {
     "apiFunction",
     "authFunction",
+    "customFunction",
     "serverFunction",
     "webhookHandle",
 }
@@ -38,32 +40,29 @@ def get_specs() -> List:
 
 
 def parse_function_specs(
-    specs: List, limit_ids: List[str] | None  # optional list of ids to limit to
-) -> List[Tuple[str, str, str, str, List[PropertySpecification], Dict[str, Any]]]:
+    specs: List[SpecificationDto],
+    limit_ids: List[str] | None,  # optional list of ids to limit to
+) -> List[SpecificationDto]:
     functions = []
     for spec in specs:
+        if not spec or "function" not in spec:
+            continue
+
+        if not spec["function"]:
+            continue
+
         if limit_ids and spec["id"] not in limit_ids:
             continue
 
         if spec["type"] not in SUPPORTED_FUNCTION_TYPES:
             continue
 
-        function_type = spec["type"]
-        function_name = f"poly.{spec['context']}.{spec['name']}"
-        function_id = spec["id"]
-        arguments: List[PropertySpecification] = [
-            arg for arg in spec["function"]["arguments"]
-        ]
-        functions.append(
-            (
-                function_type,
-                function_name,
-                function_id,
-                spec["description"],
-                arguments,
-                spec["function"]["returnType"],
-            )
-        )
+        if spec["type"] == "customFunction" and spec["language"] != "python":
+            # poly libraries only support client functions of same language
+            continue
+
+        functions.append(spec)
+
     return functions
 
 
@@ -90,11 +89,10 @@ def read_cached_specs() -> List[SpecificationDto]:
         return json.loads(f.read())
 
 
-def get_functions_and_parse(limit_ids: List[str] | None = None):
+def get_functions_and_parse(limit_ids: List[str] | None = None) -> List[SpecificationDto]:
     specs = get_specs()
     cache_specs(specs)
-    functions = parse_function_specs(specs, limit_ids=limit_ids)
-    return functions
+    return parse_function_specs(specs, limit_ids=limit_ids)
 
 
 def get_variables() -> List[VariableSpecDto]:
@@ -162,14 +160,7 @@ def save_rendered_specs() -> None:
     api_specs = [spec for spec in specs if spec["type"] == "apiFunction"]
     for spec in api_specs:
         assert spec["function"]
-        func_str, type_defs = render_spec(
-            spec["type"],
-            spec["name"],
-            spec["id"],
-            spec["description"],
-            spec["function"]["arguments"],
-            spec["function"]["returnType"],
-        )
+        func_str, type_defs = render_spec(spec)
         data = {
             "language": "python",
             "apiFunctionId": spec["id"],
@@ -181,14 +172,20 @@ def save_rendered_specs() -> None:
         assert resp.status_code == 201, (resp.text, resp.status_code)
 
 
-def render_spec(
-    function_type: str,
-    function_name: str,
-    function_id: str,
-    function_description: str,
-    arguments: List[PropertySpecification],
-    return_type: Dict[str, Any],
-):
+def render_spec(spec: SpecificationDto):
+    function_type = spec["type"]
+    function_description = spec["description"]
+    function_name = spec["name"]
+    function_id = spec["id"]
+
+    arguments: List[PropertySpecification] = []
+    return_type = {}
+    if spec["function"]:
+        arguments = [
+            arg for arg in spec["function"]["arguments"]
+        ]
+        return_type = spec["function"]["returnType"]
+
     if function_type == "apiFunction":
         func_str, func_type_defs = render_api_function(
             function_type,
@@ -198,6 +195,13 @@ def render_spec(
             arguments,
             return_type,
         )
+    elif function_type == "customFunction":
+        func_str, func_type_defs = render_client_function(
+            function_name,
+            spec["code"],
+            arguments,
+            return_type,
+        )
     elif function_type == "serverFunction":
         func_str, func_type_defs = render_server_function(
             function_type,
@@ -229,25 +233,14 @@ def render_spec(
 
 
 def add_function_file(
-    function_type: str,
     full_path: str,
     function_name: str,
-    function_id: str,
-    function_description: str,
-    arguments: List[PropertySpecification],
-    return_type: Dict[str, Any],
+    spec: SpecificationDto,
 ):
     # first lets add the import to the __init__
     init_the_init(full_path)
 
-    func_str, func_type_defs = render_spec(
-        function_type,
-        function_name,
-        function_id,
-        function_description,
-        arguments,
-        return_type,
-    )
+    func_str, func_type_defs = render_spec(spec)
 
     if func_str:
         # add function to init
@@ -262,27 +255,17 @@ def add_function_file(
 
 
 def create_function(
-    function_type: str,
-    path: str,
-    function_id: str,
-    function_description: str,
-    arguments: List[PropertySpecification],
-    return_type: Dict[str, Any],
+    spec: SpecificationDto
 ) -> None:
     full_path = os.path.dirname(os.path.abspath(__file__))
-
-    folders = path.split(".")
+    folders = f"poly.{spec['context']}.{spec['name']}".split(".")
     for idx, folder in enumerate(folders):
         if idx + 1 == len(folders):
             # special handling for final level
             add_function_file(
-                function_type,
                 full_path,
                 folder,
-                function_id,
-                function_description,
-                arguments,
-                return_type,
+                spec,
             )
         else:
             full_path = os.path.join(full_path, folder)
@@ -296,9 +279,6 @@ 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:
+def generate_functions(functions: List[SpecificationDto]) -> None:
     for func in functions:
-        create_function(*func)
+        create_function(func)

polyapi/typedefs.py

@@ -32,9 +32,11 @@ class SpecificationDto(TypedDict):
     context: str
     name: str
     description: str
-    # function is none if this is actually VariableSpecDto
-    function: FunctionSpecification | None
+    # function is none (or function key not present) if this is actually VariableSpecDto
+    function: NotRequired[FunctionSpecification | None]
     type: Literal['apiFunction', 'customFunction', 'serverFunction', 'authFunction', 'webhookHandle', 'serverVariable']
+    code: NotRequired[str]
+    language: str
 
 
 class VariableSpecification(TypedDict):

{polyapi_python-0.2.3.dev4.dist-info → polyapi_python-0.2.3.dev6.dist-info}/METADATA

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

{polyapi_python-0.2.3.dev4.dist-info → polyapi_python-0.2.3.dev6.dist-info}/RECORD

@@ -3,22 +3,23 @@ polyapi/__main__.py,sha256=V4zhAh_YGxno5f_KSrlkELxcuDh9bR3WSd0n-2r-qQQ,93
 polyapi/api.py,sha256=Pq_OT8egmtlzMjovN5GGZXWnF5oWMkrgR0rmrzJ6ifM,1861
 polyapi/auth.py,sha256=p2KSLt6q52t9gnqPmgXTOQ2_lmdFilZkIoGmQrRTPLQ,5330
 polyapi/cli.py,sha256=wMv4RpNDEz2qIQ5NjlRTop-ImfeeqCSkx22ZIrzwA50,2086
+polyapi/client.py,sha256=JIHZaun5aebPA_Fv8-wTsOys_Q5RN_tZp-lZG52Ureg,752
 polyapi/config.py,sha256=S8TU10upy5OW1_vX-CqQTJD-ZOB6329aMjiUCmukfUI,2292
 polyapi/constants.py,sha256=NGjso6K5rGnE8TGdrXmdEfvvr-HI3DTVGwOYiWO68LM,511
 polyapi/error_handler.py,sha256=vl6ZBtsHmC3eu13IMpmZEXBTDJbrPrzmViBCCrEspd4,1621
 polyapi/exceptions.py,sha256=Zh7i7eCUhDuXEdUYjatkLFTeZkrx1BJ1P5ePgbJ9eIY,89
 polyapi/execute.py,sha256=06XWTxGJqtsDiLY10RjRebIMFRfhtAIMmBRbuWu3e8A,1873
-polyapi/function_cli.py,sha256=x1nAnBQVpjhCDUd_eazQ7HTLgpsNFcY9p6YHy-8iv5E,7660
-polyapi/generate.py,sha256=ZZVkA3uL5yPWisnFhIBjoFAYfr96h9m6I0XOCbEpiyk,8970
+polyapi/function_cli.py,sha256=DPaxCYFausf01solU4pHYVp4ePO_ecg10jP-XW_zCmc,8077
+polyapi/generate.py,sha256=O1lb7rLsce7tItIrMx1EbWzRlAKKOp-7upNM80cMmMw,8499
 polyapi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 polyapi/schema.py,sha256=1a7nIO867Xy3bagmPUNdYMxtS5OoCAv8oKIbYgj55dk,2957
 polyapi/server.py,sha256=mcbH8pG16fFMiFL52wdii0217ZLsznQnElAFEvXZ7IA,2211
-polyapi/typedefs.py,sha256=RZ3I6sgJm_5MuuORG1QjUE-UJy_z2WRXNdiWjEdLvQg,1371
+polyapi/typedefs.py,sha256=a5WfHaAvjeql3y1iA3_SkpUztTbKvS5bPqkVxkCvr9E,1459
 polyapi/utils.py,sha256=wZQrwSOst5wyoYYLoMk1DnAEbVsJ0-2YkmmDQ9gI3FY,5399
 polyapi/variables.py,sha256=d36-trnfTL_8m2NkorMiImb4O3UrJbiFV38CHxV5i0A,4200
 polyapi/webhook.py,sha256=A89eNnYcVpVe9doJPDLfscIhF-C7Q2AI3vu-SzGxMBg,2923
-polyapi_python-0.2.3.dev4.dist-info/LICENSE,sha256=Hi0kDr56Dsy0uYIwNt4r9G7tI8x8miXRTlyvbeplCP8,1068
-polyapi_python-0.2.3.dev4.dist-info/METADATA,sha256=hD2Xr1MamfbMNVt43IcGERa9LfV1sIOTT1SXVhK5rao,4823
-polyapi_python-0.2.3.dev4.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-polyapi_python-0.2.3.dev4.dist-info/top_level.txt,sha256=CEFllOnzowci_50RYJac-M54KD2IdAptFsayVVF_f04,8
-polyapi_python-0.2.3.dev4.dist-info/RECORD,,
+polyapi_python-0.2.3.dev6.dist-info/LICENSE,sha256=Hi0kDr56Dsy0uYIwNt4r9G7tI8x8miXRTlyvbeplCP8,1068
+polyapi_python-0.2.3.dev6.dist-info/METADATA,sha256=6sys65C7Lq9DT1EVLph-fVGdmjBOvm6FzmXEkvfzJqY,4823
+polyapi_python-0.2.3.dev6.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+polyapi_python-0.2.3.dev6.dist-info/top_level.txt,sha256=CEFllOnzowci_50RYJac-M54KD2IdAptFsayVVF_f04,8
+polyapi_python-0.2.3.dev6.dist-info/RECORD,,