|
2 | 2 | import requests |
3 | 3 | import os |
4 | 4 | import shutil |
5 | | -from typing import List, Optional, Tuple, cast |
| 5 | +from typing import Any, List, Optional, Tuple, cast |
6 | 6 |
|
7 | 7 | from .auth import render_auth_function |
8 | 8 | from .client import render_client_function |
@@ -41,7 +41,7 @@ def get_specs(contexts: Optional[List[str]] = None, names: Optional[List[str]] = |
41 | 41 | assert api_key |
42 | 42 | headers = get_auth_headers(api_key) |
43 | 43 | url = f"{api_url}/specs" |
44 | | - params = {"noTypes": str(no_types).lower()} |
| 44 | + params: Any = {"noTypes": str(no_types).lower()} |
45 | 45 |
|
46 | 46 | if contexts: |
47 | 47 | params["contexts"] = contexts |
@@ -155,7 +155,7 @@ def parse_function_specs( |
155 | 155 |
|
156 | 156 | # Functions with serverSideAsync True will always return a Dict with execution ID |
157 | 157 | if spec.get('serverSideAsync') and spec.get("function"): |
158 | | - spec['function']['returnType'] = {'kind': 'plain', 'value': 'object'} |
| 158 | + spec['function']['returnType'] = {'kind': 'plain', 'value': 'object'} # type: ignore |
159 | 159 |
|
160 | 160 | functions.append(spec) |
161 | 161 |
|
@@ -321,11 +321,9 @@ def generate(contexts: Optional[List[str]] = None, names: Optional[List[str]] = |
321 | 321 | ) |
322 | 322 | exit() |
323 | 323 |
|
324 | | - # Only process variables if no_types is False |
325 | | - if not no_types: |
326 | | - variables = get_variables() |
327 | | - if variables: |
328 | | - generate_variables(variables) |
| 324 | + variables = get_variables() |
| 325 | + if variables: |
| 326 | + generate_variables(variables) |
329 | 327 |
|
330 | 328 | # indicator to vscode extension that this is a polyapi-python project |
331 | 329 | file_path = os.path.join(os.getcwd(), ".polyapi-python") |
@@ -354,8 +352,9 @@ def render_spec(spec: SpecificationDto) -> Tuple[str, str]: |
354 | 352 | function_id = spec["id"] |
355 | 353 |
|
356 | 354 | arguments: List[PropertySpecification] = [] |
357 | | - return_type = {} |
| 355 | + return_type: Any = {} |
358 | 356 | if spec.get("function"): |
| 357 | + assert spec["function"] |
359 | 358 | # Handle cases where arguments might be missing or None |
360 | 359 | if spec["function"].get("arguments"): |
361 | 360 | arguments = [ |
|
0 commit comments