wbapi-codegen 0.1.0__tar.gz → 0.1.2__tar.gz

wbapi_codegen-0.1.2/PKG-INFO

@@ -0,0 +1,21 @@
+Metadata-Version: 2.4
+Name: wbapi-codegen
+Version: 0.1.2
+Summary: Code generator for wbapi-async — generates types, methods and API client from Wildberries OpenAPI specs
+Project-URL: Repository, https://github.com/serdukow/wbapi-codegen
+Author-email: Andrei Serdiukov <asyncdf@gmail.com>
+Maintainer-email: Andrei Serdiukov <asyncdf@gmail.com>
+License: MIT
+Keywords: api,codegen,openapi,wb,wildberries
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Software Development :: Code Generators
+Requires-Python: >=3.10
+Requires-Dist: pyyaml>=6.0
+Description-Content-Type: text/markdown
+
+# wbapi-codegen
+
+Code generator for [wbapi-async](https://github.com/serdukow/wbapi-async).

wbapi_codegen-0.1.2/README.md

@@ -0,0 +1,3 @@
+# wbapi-codegen
+
+Code generator for [wbapi-async](https://github.com/serdukow/wbapi-async).

{wbapi_codegen-0.1.0 → wbapi_codegen-0.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "wbapi-codegen"
-version = "0.1.0"
+version = "0.1.2"
 description = "Code generator for wbapi-async — generates types, methods and API client from Wildberries OpenAPI specs"
 authors = [
     { name = "Andrei Serdiukov", email = "asyncdf@gmail.com" },

{wbapi_codegen-0.1.0 → wbapi_codegen-0.1.2}/src/wbapi_codegen/codegen.py

@@ -12,8 +12,7 @@ Parses OpenAPI YAML files and generates:
 from __future__ import annotations
 
 import re
-import sys
-from dataclasses import dataclass, field
+from dataclasses import dataclass
 from keyword import iskeyword
 from pathlib import Path
 from typing import Any
@@ -177,6 +176,7 @@ class MethodDef:
     description: str = ""
     source_url: str = ""
     type_def: TypeDef | None = None
+    empty_response: bool = False  # True for 204 No Content or empty schema
 
 
 # ---------------------------------------------------------------------------
@@ -254,6 +254,27 @@ def _find_response_array(schema: dict, spec: dict) -> tuple[list[FieldDef], str
     return fields, dot_path
 
 
+_DOC_SLUG_MAP: dict[str, str] = {
+    "01-general": "api-information",
+    "02-products": "work-with-products",
+    "03-orders-fbs": "orders-fbs",
+    "04-orders-dbw": "orders-dbw",
+    "05-orders-dbs": "orders-dbs",
+    "06-in-store-pickup": "in-store-pickup",
+    "07-orders-fbw": "orders-fbw",
+    "08-promotion": "promotion",
+    "09-communications": "communications",
+    "10-tariffs": "tariffs",
+    "11-analytics": "analytics",
+    "12-reports": "reports",
+    "13-finances": "financial-reports-and-accounting",
+}
+
+
+def _yaml_stem_to_doc_slug(stem: str) -> str:
+    return _DOC_SLUG_MAP.get(stem, stem)
+
+
 def parse_yaml(yaml_path: Path) -> list[MethodDef]:
     with yaml_path.open() as f:
         spec = yaml.safe_load(f)
@@ -338,14 +359,16 @@ def parse_yaml(yaml_path: Path) -> list[MethodDef]:
 
             # response type
             responses = operation.get("responses", {})
+            # prefer 200, then 201, then 204
             ok_response = _resolve_schema(
-                responses.get("200", responses.get("201", {})), spec
+                responses.get("200", responses.get("201", responses.get("204", {}))), spec
             )
             ok_content = ok_response.get("content", {})
             ok_schema = _resolve_schema(
                 ok_content.get("application/json", {}).get("schema", {}), spec
             )
             item_fields, data_key = _find_response_array(ok_schema, spec)
+            empty_response = not ok_content or not ok_schema
 
             # build names from summary
             if summary:
@@ -373,10 +396,15 @@ def parse_yaml(yaml_path: Path) -> list[MethodDef]:
 
             tag = (operation.get("tags") or [""])[0]
             tag_slug = re.sub(r"[^a-zA-Z0-9]", "-", tag).strip("-")
-            path_encoded = path_str.replace("/", "~1")
+            path_encoded = (
+                path_str.replace("/", "~1")
+                        .replace("{", "%7B")
+                        .replace("}", "%7D")
+            )
             source_url = (
-                f"https://dev.wildberries.ru/en/openapi/"
-                f"{yaml_path.stem}#tag/{tag_slug}/paths/{path_encoded}/{http_verb}"
+                f"https://dev.wildberries.ru/en/docs/openapi/"
+                f"{_yaml_stem_to_doc_slug(yaml_path.stem)}"
+                f"#tag/{tag_slug}/paths/{path_encoded}/{http_verb}"
             )
 
             results.append(MethodDef(
@@ -393,6 +421,7 @@ def parse_yaml(yaml_path: Path) -> list[MethodDef]:
                 params=param_fields,
                 description=summary,
                 source_url=source_url,
+                empty_response=empty_response,
                 type_def=TypeDef(
                     class_name=type_name,
                     fields=item_fields or [],
@@ -472,6 +501,8 @@ def generate_method_file(md: MethodDef) -> str:
         lines += ['    """'] + [f"    {dl}" if dl else "" for dl in doc_lines] + ['    """', ""]
 
     lines.append(f"    __return__ = {md.return_type}")
+    if md.empty_response:
+        lines.append("    __empty_response__ = True")
     lines.append(f'    __api__ = "{md.api_host}"')
     if md.path_template:
         lines.append('    __method__ = ""')
@@ -562,7 +593,7 @@ def _api_method_wrapper(md: MethodDef) -> str:
         sig_parts.append(f"{p.py_name}: {p.py_type} = {default}")
 
     sig = ", ".join(sig_parts)
-    ret = f"list[{md.return_type}]"
+    ret = "None" if md.empty_response else f"list[{md.return_type}]"
 
     call_args = ", ".join(
         f"{p.py_name}={p.py_name}" for p in path_params + non_path_params
@@ -719,24 +750,28 @@ def generate_test_file(md: MethodDef) -> str:
         f"{p.py_name}={_mock_value(p.py_type, p.py_name)}" for p in required_params
     )
 
-    mock_response = _build_mock_response(
-        md.type_def.fields if md.type_def else [], md.data_key
-    )
-
-    # pick a field to assert on (first non-None field)
-    assert_lines: list[str] = []
-    if md.type_def and md.type_def.fields:
-        for f in md.type_def.fields[:3]:
-            val = _mock_value(f.py_type, f.alias)
-            if val != "None":
-                assert_lines.append(
-                    f"        assert result[0].{f.py_name} == {val}"
-                )
+    if md.empty_response:
+        mock_response = "None"
+        assert_lines = ["        assert result is None"]
+    else:
+        mock_response = _build_mock_response(
+            md.type_def.fields if md.type_def else [], md.data_key
+        )
+        assert_lines = [
+            f"        assert isinstance(result, list)",
+            f"        assert len(result) == 1",
+            f"        assert isinstance(result[0], {md.return_type})",
+        ]
+        if md.type_def and md.type_def.fields:
+            for f in md.type_def.fields[:3]:
+                val = _mock_value(f.py_type, f.alias)
+                if val != "None":
+                    assert_lines.append(f"        assert result[0].{f.py_name} == {val}")
 
     lines = [
         "import pytest",
         "",
-        f"from wbapi_async.types.{md.return_type_file} import {md.return_type}",
+        *([f"from wbapi_async.types.{md.return_type_file} import {md.return_type}"] if not md.empty_response else []),
         "from tests.mocked_api import MockedAPI",
         "",
         "",
@@ -750,9 +785,6 @@ def generate_test_file(md: MethodDef) -> str:
         "",
         f"        result = await api.{md.method_name}({call_args})",
         "",
-        f"        assert isinstance(result, list)",
-        f"        assert len(result) == 1",
-        f"        assert isinstance(result[0], {md.return_type})",
         *assert_lines,
         "",
     ]
@@ -855,6 +887,7 @@ def run(yaml_files: list[Path], target: Path) -> None:
     ))
     print(f"[init] {(methods_dir / '__init__.py').relative_to(target)}")
 
+    all_method_defs.sort(key=lambda m: m.method_name)
     api_path.write_text(generate_api_file(all_method_defs, unofficial_block))
     print(f"[api]  {api_path.relative_to(target)}")
 

wbapi_codegen-0.1.0/PKG-INFO

@@ -1,59 +0,0 @@
-Metadata-Version: 2.4
-Name: wbapi-codegen
-Version: 0.1.0
-Summary: Code generator for wbapi-async — generates types, methods and API client from Wildberries OpenAPI specs
-Project-URL: Repository, https://github.com/serdukow/wbapi-codegen
-Author-email: Andrei Serdiukov <asyncdf@gmail.com>
-Maintainer-email: Andrei Serdiukov <asyncdf@gmail.com>
-License: MIT
-Keywords: api,codegen,openapi,wb,wildberries
-Classifier: Development Status :: 3 - Alpha
-Classifier: Intended Audience :: Developers
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Classifier: Topic :: Software Development :: Code Generators
-Requires-Python: >=3.10
-Requires-Dist: pyyaml>=6.0
-Description-Content-Type: text/markdown
-
-# wbapi-codegen
-
-Code generator for [wbapi-async](https://github.com/serdukow/wbapi-async).
-
-Parses Wildberries OpenAPI YAML specs and generates:
-- `src/wbapi_async/types/` — Pydantic response models
-- `src/wbapi_async/methods/` — `WbMethod` request classes
-- `src/wbapi_async/client/api.py` — `WbAPI` wrapper methods
-
-Files marked `__unofficial__ = True` are never overwritten.
-
-## Install
-
-```bash
-pip install wbapi-codegen
-```
-
-## Usage
-
-```bash
-# process all YAMLs from .wb/swagger/ into current directory
-wbapi-codegen --target /path/to/wbapi-async
-
-# custom swagger directory
-wbapi-codegen --target /path/to/wbapi-async --swagger-dir /path/to/yaml/files
-
-# single file
-wbapi-codegen --target /path/to/wbapi-async /path/to/02-products.yaml
-```
-
-## How it works
-
-1. Downloads OpenAPI YAMLs from Wildberries developer portal
-2. Parses paths, parameters, request bodies and response schemas
-3. Generates typed Python files following the `wbapi-async` conventions
-4. Creates a PR in the `wbapi-async` repo via GitHub Actions (runs daily at 04:00 UTC)
-
-## Setup for GitHub Actions
-
-In the `wbapi-codegen` repo, add a secret `WBAPI_PAT` — a GitHub Personal Access Token
-with `repo` scope on the `wbapi-async` repository.

wbapi_codegen-0.1.0/README.md

@@ -1,41 +0,0 @@
-# wbapi-codegen
-
-Code generator for [wbapi-async](https://github.com/serdukow/wbapi-async).
-
-Parses Wildberries OpenAPI YAML specs and generates:
-- `src/wbapi_async/types/` — Pydantic response models
-- `src/wbapi_async/methods/` — `WbMethod` request classes
-- `src/wbapi_async/client/api.py` — `WbAPI` wrapper methods
-
-Files marked `__unofficial__ = True` are never overwritten.
-
-## Install
-
-```bash
-pip install wbapi-codegen
-```
-
-## Usage
-
-```bash
-# process all YAMLs from .wb/swagger/ into current directory
-wbapi-codegen --target /path/to/wbapi-async
-
-# custom swagger directory
-wbapi-codegen --target /path/to/wbapi-async --swagger-dir /path/to/yaml/files
-
-# single file
-wbapi-codegen --target /path/to/wbapi-async /path/to/02-products.yaml
-```
-
-## How it works
-
-1. Downloads OpenAPI YAMLs from Wildberries developer portal
-2. Parses paths, parameters, request bodies and response schemas
-3. Generates typed Python files following the `wbapi-async` conventions
-4. Creates a PR in the `wbapi-async` repo via GitHub Actions (runs daily at 04:00 UTC)
-
-## Setup for GitHub Actions
-
-In the `wbapi-codegen` repo, add a secret `WBAPI_PAT` — a GitHub Personal Access Token
-with `repo` scope on the `wbapi-async` repository.