polyapi-python 0.2.3.dev1__tar.gz → 0.2.3.dev3__tar.gz

{polyapi-python-0.2.3.dev1 → polyapi-python-0.2.3.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyapi-python
-Version: 0.2.3.dev1
+Version: 0.2.3.dev3
 Summary: The PolyAPI Python Client
 Author-email: Dan Fellin <dan@polyapi.io>
 License: MIT License
@@ -31,7 +31,7 @@ License-File: LICENSE
 Requires-Dist: requests==2.31.0
 Requires-Dist: typing_extensions==4.10.0
 Requires-Dist: jsonschema-gentypes==2.4.0
-Requires-Dist: pydantic==2.6.4
+Requires-Dist: pydantic==2.5.3
 Requires-Dist: stdlib_list==0.10.0
 Requires-Dist: colorama==0.4.4
 Requires-Dist: python-socketio[asyncio_client]==5.11.1

{polyapi-python-0.2.3.dev1 → polyapi-python-0.2.3.dev3}/polyapi/function_cli.py

@@ -8,6 +8,7 @@ from typing_extensions import _TypedDictMeta  # type: ignore
 import requests
 from stdlib_list import stdlib_list
 from pydantic import TypeAdapter
+from importlib.metadata import packages_distributions
 from polyapi.generate import get_functions_and_parse, generate_functions
 from polyapi.config import get_api_key_and_url
 from polyapi.constants import PYTHON_TO_JSONSCHEMA_TYPE_MAP
@@ -112,6 +113,12 @@ def _parse_code(code: str, function_name: str):
     schemas = _get_schemas(code)
 
     parsed_code = ast.parse(code)
+
+    # the pip name and the import name might be different
+    # e.g. kube_hunter is the import name, but the pip name is kube-hunter
+    # see https://stackoverflow.com/a/75144378
+    pip_name_lookup = packages_distributions()
+
     for node in ast.iter_child_nodes(parsed_code):
         if isinstance(node, ast.Import):
             for name in node.names:
@@ -119,7 +126,8 @@ def _parse_code(code: str, function_name: str):
                     requirements.append(name.name)
         elif isinstance(node, ast.ImportFrom):
             if node.module and node.module not in BASE_REQUIREMENTS:
-                requirements.append(node.module)
+                req = pip_name_lookup[node.module][0]
+                requirements.append(req)
 
         elif isinstance(node, ast.FunctionDef) and node.name == function_name:
             function_args = [arg for arg in node.args.args]

{polyapi-python-0.2.3.dev1 → polyapi-python-0.2.3.dev3}/polyapi_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyapi-python
-Version: 0.2.3.dev1
+Version: 0.2.3.dev3
 Summary: The PolyAPI Python Client
 Author-email: Dan Fellin <dan@polyapi.io>
 License: MIT License
@@ -31,7 +31,7 @@ License-File: LICENSE
 Requires-Dist: requests==2.31.0
 Requires-Dist: typing_extensions==4.10.0
 Requires-Dist: jsonschema-gentypes==2.4.0
-Requires-Dist: pydantic==2.6.4
+Requires-Dist: pydantic==2.5.3
 Requires-Dist: stdlib_list==0.10.0
 Requires-Dist: colorama==0.4.4
 Requires-Dist: python-socketio[asyncio_client]==5.11.1

{polyapi-python-0.2.3.dev1 → polyapi-python-0.2.3.dev3}/polyapi_python.egg-info/requires.txt

@@ -1,7 +1,7 @@
 requests==2.31.0
 typing_extensions==4.10.0
 jsonschema-gentypes==2.4.0
-pydantic==2.6.4
+pydantic==2.5.3
 stdlib_list==0.10.0
 colorama==0.4.4
 python-socketio[asyncio_client]==5.11.1

{polyapi-python-0.2.3.dev1 → polyapi-python-0.2.3.dev3}/pyproject.toml

@@ -3,14 +3,14 @@ requires = ["setuptools>=61.2", "wheel"]
 
 [project]
 name = "polyapi-python"
-version = "0.2.3.dev1"
+version = "0.2.3.dev3"
 description = "The PolyAPI Python Client"
 authors = [{ name = "Dan Fellin", email = "dan@polyapi.io" }]
 dependencies = [
     "requests==2.31.0",
     "typing_extensions==4.10.0",
     "jsonschema-gentypes==2.4.0",
-    "pydantic==2.6.4",
+    "pydantic==2.5.3",
     "stdlib_list==0.10.0",
     "colorama==0.4.4",
     "python-socketio[asyncio_client]==5.11.1",