webtap-tool 0.5.2__py3-none-any.whl → 0.7.0__py3-none-any.whl

webtap/app.py

@@ -89,6 +89,7 @@ else:
     from webtap.commands import inspect  # noqa: E402, F401
     from webtap.commands import fetch  # noqa: E402, F401
     from webtap.commands import body  # noqa: E402, F401
+    from webtap.commands import to_model  # noqa: E402, F401
     from webtap.commands import selections  # noqa: E402, F401
     from webtap.commands import server  # noqa: E402, F401
     from webtap.commands import setup  # noqa: E402, F401

webtap/commands/TIPS.md

@@ -26,12 +26,31 @@ body(123, "msgpack.unpackb(body)")         # Binary formats
 ```
 
 #### Tips
+- **Generate models:** `to_model({id}, "models/model.py", "Model")` - create Pydantic models from JSON
 - **Chain requests:** `body({id}, "httpx.get(json.loads(body)['next_url']).text[:100]")`
 - **Parse XML:** `body({id}, "ElementTree.fromstring(body).find('.//title').text")`
 - **Extract forms:** `body({id}, "[f['action'] for f in bs4(body, 'html.parser').find_all('form')]")`
 - **Decode protobuf:** `body({id}, "protobuf_json.Parse(body, YourMessage())")`
 - **Find related:** `events({'requestId': request_id})` - related events
 
+### to_model
+Generate Pydantic v2 models from JSON response bodies for reverse engineering APIs.
+
+#### Examples
+```python
+to_model(123, "models/product.py", "Product")                              # Generate from full response
+to_model(123, "models/customers/group.py", "CustomerGroup", "Data[0]")     # Extract nested + domain structure
+to_model(123, "/tmp/item.py", "Item", "items[0]")                          # Extract array items
+```
+
+#### Tips
+- **Check structure first:** `body({id})` - preview JSON before generating
+- **Domain organization:** Use paths like `"models/customers/group.py"` for structure
+- **Extract nested data:** Use `json_path="Data[0]"` to extract specific objects
+- **Array items:** Extract first item with `json_path="items[0]"` for model generation
+- **Auto-cleanup:** Generated models use snake_case fields and modern type hints (list, dict, | None)
+- **Edit after:** Add `Field(alias="...")` manually for API field mapping
+
 ### inspect
 Inspect CDP events with full Python debugging.
 
@@ -62,6 +81,7 @@ Show network requests with full data.
 
 #### Tips
 - **Analyze responses:** `body({id})` - fetch response body
+- **Generate models:** `to_model({id}, "models/model.py", "Model")` - create Pydantic models from JSON
 - **Parse HTML:** `body({id}, "bs4(body, 'html.parser').find('title').text")`
 - **Extract JSON:** `body({id}, "json.loads(body)['data']")`
 - **Find patterns:** `body({id}, "re.findall(r'/api/\\w+', body)")`

webtap/commands/to_model.py

@@ -0,0 +1,136 @@
+"""Generate Pydantic models from HTTP response bodies."""
+
+import json
+from pathlib import Path
+from datamodel_code_generator import generate, InputFileType, DataModelType
+from webtap.app import app
+from webtap.commands._builders import check_connection, success_response, error_response
+from webtap.commands._tips import get_mcp_description
+
+
+mcp_desc = get_mcp_description("to_model")
+
+
+@app.command(display="markdown", fastmcp={"type": "tool", "description": mcp_desc} if mcp_desc else {"type": "tool"})
+def to_model(state, response: int, output: str, model_name: str, json_path: str = None) -> dict:  # pyright: ignore[reportArgumentType]
+    """Generate Pydantic model from response body using datamodel-codegen.
+
+    Args:
+        response: Response row ID from network() table
+        output: Output file path for generated model (e.g., "models/customers/group.py")
+        model_name: Class name for generated model (e.g., "CustomerGroup")
+        json_path: Optional JSON path to extract nested data (e.g., "Data[0]")
+
+    Returns:
+        Success message with generation details
+    """
+    if error := check_connection(state):
+        return error
+
+    # Get response body from service
+    body_service = state.service.body
+    result = body_service.get_response_body(response, use_cache=True)
+
+    if "error" in result:
+        return error_response(result["error"])
+
+    body_content = result.get("body", "")
+    is_base64 = result.get("base64Encoded", False)
+
+    # Decode if needed
+    if is_base64:
+        decoded = body_service.decode_body(body_content, is_base64)
+        if isinstance(decoded, bytes):
+            return error_response(
+                "Response body is binary",
+                suggestions=["Only JSON responses can be converted to models", "Try a different response"],
+            )
+        body_content = decoded
+
+    # Parse JSON
+    try:
+        data = json.loads(body_content)
+    except json.JSONDecodeError as e:
+        return error_response(
+            f"Invalid JSON: {e}",
+            suggestions=[
+                "Response must be valid JSON",
+                "Check the response with body() first",
+                "Try a different response",
+            ],
+        )
+
+    # Extract JSON path if specified
+    if json_path:
+        try:
+            # Support simple bracket notation like "Data[0]"
+            parts = json_path.replace("[", ".").replace("]", "").split(".")
+            for part in parts:
+                if part:
+                    if part.isdigit():
+                        data = data[int(part)]
+                    else:
+                        data = data[part]
+        except (KeyError, IndexError, TypeError) as e:
+            return error_response(
+                f"JSON path extraction failed: {e}",
+                suggestions=[
+                    f"Path '{json_path}' not found in response",
+                    "Check the response structure with body()",
+                    'Try a simpler path like "Data" or "Data[0]"',
+                ],
+            )
+
+    # Ensure data is dict or list for model generation
+    if not isinstance(data, (dict, list)):
+        return error_response(
+            f"Extracted data is {type(data).__name__}, not dict or list",
+            suggestions=[
+                "Model generation requires dict or list structure",
+                "Adjust json_path to extract a complex object",
+            ],
+        )
+
+    # Create output directory if needed
+    output_path = Path(output).expanduser().resolve()
+    output_path.parent.mkdir(parents=True, exist_ok=True)
+
+    # Generate model using datamodel-codegen Python API
+    try:
+        generate(
+            json.dumps(data),
+            input_file_type=InputFileType.Json,
+            input_filename="response.json",
+            output=output_path,
+            output_model_type=DataModelType.PydanticV2BaseModel,
+            class_name=model_name,  # Set generated class name
+            snake_case_field=True,  # Convert to snake_case
+            use_standard_collections=True,  # Use list instead of List
+            use_union_operator=True,  # Use | instead of Union
+        )
+    except Exception as e:
+        return error_response(
+            f"Model generation failed: {e}",
+            suggestions=[
+                "Check that the JSON structure is valid",
+                "Try simplifying the JSON path",
+                "Ensure output directory is writable",
+            ],
+        )
+
+    # Count fields in generated model
+    try:
+        model_content = output_path.read_text()
+        field_count = model_content.count(": ")  # Count field definitions
+    except Exception:
+        field_count = "unknown"
+
+    return success_response(
+        "Model generated successfully",
+        details={
+            "Class": model_name,
+            "Output": str(output_path),
+            "Fields": field_count,
+            "Size": f"{output_path.stat().st_size} bytes",
+        },
+    )

{webtap_tool-0.5.2.dist-info → webtap_tool-0.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: webtap-tool
-Version: 0.5.2
+Version: 0.7.0
 Summary: Terminal-based web page inspector for AI debugging sessions
 Author-email: Fredrik Angelsen <fredrikangelsen@gmail.com>
 Classifier: Development Status :: 3 - Alpha
@@ -12,6 +12,7 @@ Classifier: Topic :: Software Development :: Debuggers
 Requires-Python: >=3.12
 Requires-Dist: beautifulsoup4>=4.13.5
 Requires-Dist: cryptography>=45.0.6
+Requires-Dist: datamodel-code-generator>=0.35.0
 Requires-Dist: duckdb>=1.3.2
 Requires-Dist: fastapi>=0.116.1
 Requires-Dist: httpx>=0.28.1

{webtap_tool-0.5.2.dist-info → webtap_tool-0.7.0.dist-info}/RECORD

@@ -1,7 +1,7 @@
 webtap/VISION.md,sha256=kfoJfEPVc4chOrD9tNMDmYBY9rX9KB-286oZj70ALCE,7681
 webtap/__init__.py,sha256=DFWJGmqZfX8_h4csLA5pKPR4SkaHBMlUgU-WQIE96Gw,1092
 webtap/api.py,sha256=twDO_aA861yjjvrKquZ0phm2SIL07Wq37DmFjtVbTw4,17990
-webtap/app.py,sha256=UipzpRGS6riZxU7CgyW_eeWcB-6817P3egEO0pIdcp4,3790
+webtap/app.py,sha256=75x_MQ2sjt6-HJi-YYRdFoKc3ss6SQfgFVArE1QKs3M,3851
 webtap/filters.py,sha256=kRCicGMSV3R_zSvwzqZqksnry6jxJNdXRcgWvpoBLfc,13323
 webtap/cdp/README.md,sha256=0TS0V_dRgRAzBqhddpXWD4S0YVi5wI4JgFJSll_KUBE,5660
 webtap/cdp/__init__.py,sha256=c6NFG0XJnAa5GTe9MLr9mDZcLZqoTQN7A1cvvOfLcgY,453
@@ -11,7 +11,7 @@ webtap/cdp/schema/README.md,sha256=hnWCzbXYcYtWaZb_SgjVaFBiG81S9b9Y3x-euQFwQDo,1
 webtap/cdp/schema/cdp_protocol.json,sha256=dp9_OLYLuVsQb1oV5r6MZfMzURscBLyAXUckdaPWyv4,1488452
 webtap/cdp/schema/cdp_version.json,sha256=OhGy1qpfQjSe3Z7OqL6KynBFlDFBXxKGPZCY-ZN_lVU,399
 webtap/commands/DEVELOPER_GUIDE.md,sha256=LYOhycZ3k5EHx5nREfkjvLz7vOs8pXCRLlcDm-keWao,11973
-webtap/commands/TIPS.md,sha256=LnzNHb6fhoeqEyNIhx5IOKd05nY-xCiDLCrJ5zBOmNY,8131
+webtap/commands/TIPS.md,sha256=XwnPKY5AgLsxNw3q0aF4Amr0P891Pt4QIkw0_3AF58g,9293
 webtap/commands/__init__.py,sha256=rr3xM_bY0BgxkDOjsnsI8UBhjlz7nqiYlgJ8fjiJ1jQ,270
 webtap/commands/_builders.py,sha256=SYacZmZTdkolQ7OOf3rFtFPCjkukY8z020WFA-i_O_A,7902
 webtap/commands/_tips.py,sha256=SleMpwdghrHNqdzR60Cu8T0NZqJfWfcfrgIcyWI6GIQ,4793
@@ -30,6 +30,7 @@ webtap/commands/network.py,sha256=gEOg_u7VF9A5aKv5myzLCuvfAUkF1OPxsuj4UAgbS44,31
 webtap/commands/selections.py,sha256=M001d_Gc51aSTuVeXGa19LDh2ZGR_qBJEjVGKpcGGFM,4895
 webtap/commands/server.py,sha256=DOcIgYuKp0ydwrK9EA3hGwqOwfwM9DABhdPu3hk_jjo,6948
 webtap/commands/setup.py,sha256=dov1LaN50nAEMNIuBLSK7mcnwhfn9rtqdTopBm1-PhA,9648
+webtap/commands/to_model.py,sha256=jOb93t616m5weT75VyF506J6nydDXUWENF7cpscbe9Q,4962
 webtap/services/README.md,sha256=rala_jtnNgSiQ1lFLM7x_UQ4SJZDceAm7dpkQMRTYaI,2346
 webtap/services/__init__.py,sha256=IjFqu0Ak6D-r18aokcQMtenDV3fbelvfjTCejGv6CZ0,570
 webtap/services/body.py,sha256=XQPa19y5eUc3XJ2TuwVK6kffO1VQoKqNs33MBBz7hzU,3913
@@ -44,7 +45,7 @@ webtap/services/setup/desktop.py,sha256=fXwQa201W-s2mengm_dJZ9BigJopVrO9YFUQcW_T
 webtap/services/setup/extension.py,sha256=iJY43JlQO6Vicgd9Mz6Mw0LQfbBNUGhnwI8n-LnvHBY,3602
 webtap/services/setup/filters.py,sha256=lAPSLMH_KZQO-7bRkmURwzforx7C3SDrKEw2ZogN-Lo,3220
 webtap/services/setup/platform.py,sha256=7yn-7LQFffgerWzWRtOG-yNEsR36ICThYUAu_N2FAso,4532
-webtap_tool-0.5.2.dist-info/METADATA,sha256=Tcq1QufQPMu8KhNGAOnV0UIc6_y1jiXag6Oe0yuipgE,17588
-webtap_tool-0.5.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-webtap_tool-0.5.2.dist-info/entry_points.txt,sha256=iFe575I0CIb1MbfPt0oX2VYyY5gSU_dA551PKVR83TU,39
-webtap_tool-0.5.2.dist-info/RECORD,,
+webtap_tool-0.7.0.dist-info/METADATA,sha256=JAfV9YTu10f5ss15yatHtYFx8W8HJewrPthMBS0TwN4,17636
+webtap_tool-0.7.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+webtap_tool-0.7.0.dist-info/entry_points.txt,sha256=iFe575I0CIb1MbfPt0oX2VYyY5gSU_dA551PKVR83TU,39
+webtap_tool-0.7.0.dist-info/RECORD,,