polyapi-python 0.2.3.dev1__py3-none-any.whl → 0.2.3.dev2__py3-none-any.whl

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.dist-info → polyapi_python-0.2.3.dev2.dist-info}/METADATA

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

{polyapi_python-0.2.3.dev1.dist-info → polyapi_python-0.2.3.dev2.dist-info}/RECORD

@@ -8,7 +8,7 @@ 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=5e6hJ3nitKxbQNSkp4jbuAzbJvykj3de1_aSsc1MBnY,7331
+polyapi/function_cli.py,sha256=x1nAnBQVpjhCDUd_eazQ7HTLgpsNFcY9p6YHy-8iv5E,7660
 polyapi/generate.py,sha256=ZZVkA3uL5yPWisnFhIBjoFAYfr96h9m6I0XOCbEpiyk,8970
 polyapi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 polyapi/schema.py,sha256=1a7nIO867Xy3bagmPUNdYMxtS5OoCAv8oKIbYgj55dk,2957
@@ -17,8 +17,8 @@ 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=A89eNnYcVpVe9doJPDLfscIhF-C7Q2AI3vu-SzGxMBg,2923
-polyapi_python-0.2.3.dev1.dist-info/LICENSE,sha256=Hi0kDr56Dsy0uYIwNt4r9G7tI8x8miXRTlyvbeplCP8,1068
-polyapi_python-0.2.3.dev1.dist-info/METADATA,sha256=9IbEwxs-o-u_OMvTeOW_MjVIMMCQr98eqKAhPRWSy3s,4823
-polyapi_python-0.2.3.dev1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-polyapi_python-0.2.3.dev1.dist-info/top_level.txt,sha256=CEFllOnzowci_50RYJac-M54KD2IdAptFsayVVF_f04,8
-polyapi_python-0.2.3.dev1.dist-info/RECORD,,
+polyapi_python-0.2.3.dev2.dist-info/LICENSE,sha256=Hi0kDr56Dsy0uYIwNt4r9G7tI8x8miXRTlyvbeplCP8,1068
+polyapi_python-0.2.3.dev2.dist-info/METADATA,sha256=-kX0lz42ns-Bzdl5XS8FKZF4qUkDndJH1JlIm_GfNSo,4823
+polyapi_python-0.2.3.dev2.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+polyapi_python-0.2.3.dev2.dist-info/top_level.txt,sha256=CEFllOnzowci_50RYJac-M54KD2IdAptFsayVVF_f04,8
+polyapi_python-0.2.3.dev2.dist-info/RECORD,,