diff --git a/preprocess_schemas.py b/preprocess_schemas.py index 090b0f7..9294143 100644 --- a/preprocess_schemas.py +++ b/preprocess_schemas.py @@ -461,33 +461,59 @@ def rewrite_refs_to_variants(root, op, file_path, variant_needs): ) -def _create_single_variant( - schema, op, stem, file_path, global_variant_requirements +def _apply_request_rules_to_object( + object_schema, op, file_path, global_variant_requirements ): - """Creates a modified copy of the schema tailored for a specific operation.""" - variant = copy.deepcopy(schema) - update_variant_identity(variant, op, stem) + """Filters an object schema's properties for a request operation.""" + properties = object_schema.get("properties", {}) + if not isinstance(properties, dict): + return new_props = {} new_required = [] - base_req = schema.get("required", []) + base_required = object_schema.get("required", []) - for name, data in schema.get("properties", {}).items(): - include, required = eval_prop_inclusion(name, data, op, base_req) + for name, data in properties.items(): + include, required = eval_prop_inclusion(name, data, op, base_required) if include: - prop_data = copy.deepcopy(data) - if isinstance(prop_data, dict): - prop_data.pop("ucp_request", None) + if isinstance(data, dict): + data.pop("ucp_request", None) rewrite_refs_to_variants( - prop_data, op, file_path, global_variant_requirements + data, op, file_path, global_variant_requirements ) - new_props[name] = prop_data + new_props[name] = data if required: new_required.append(name) - variant["properties"] = new_props - variant["required"] = new_required + object_schema["properties"] = new_props + object_schema["required"] = new_required + + +def _create_single_variant( + schema, op, stem, file_path, global_variant_requirements +): + """Creates a modified copy of the schema tailored for a specific operation.""" + variant = copy.deepcopy(schema) + update_variant_identity(variant, op, stem) + + if variant.get("type") == "array" and isinstance( + variant.get("items"), dict + ): + object_nodes = [ + node + for node in iter_nodes(variant["items"]) + if isinstance(node, dict) and "properties" in node + ] + for node in object_nodes: + _apply_request_rules_to_object( + node, op, file_path, global_variant_requirements + ) + elif "properties" in variant or variant.get("type") == "object": + _apply_request_rules_to_object( + variant, op, file_path, global_variant_requirements + ) + return variant diff --git a/src/ucp_sdk/models/schemas/shopping/types/totals_create_request.py b/src/ucp_sdk/models/schemas/shopping/types/totals_create_request.py index 6530e8c..c201fdb 100644 --- a/src/ucp_sdk/models/schemas/shopping/types/totals_create_request.py +++ b/src/ucp_sdk/models/schemas/shopping/types/totals_create_request.py @@ -20,54 +20,14 @@ from typing import Annotated -from pydantic import BaseModel, ConfigDict, Field +from pydantic import Field from typing_extensions import TypeAliasType -from . import signed_amount -from .total import Total - - -class Line(BaseModel): - """ - Sub-line entry. Additional metadata MAY be included. - """ - - model_config = ConfigDict( - extra="allow", - ) - display_text: str - """ - Human-readable label for this sub-line. - """ - amount: signed_amount.SignedAmount - - -class TotalsCreateRequestItem(Total): - model_config = ConfigDict( - extra="allow", - ) - lines: list[Line] | None = None - """ - Optional itemized breakdown. The parent entry is always rendered; lines are supplementary. Sum of line amounts MUST equal the parent entry amount. - """ - - -class TotalsCreateRequest1(BaseModel): - """ - Pricing breakdown provided by the business. MUST contain exactly one subtotal and one total entry. Detail types (tax, fee, discount, fulfillment) may appear multiple times for itemization. Platforms MUST render all entries in order using display_text and amount. - """ - - model_config = ConfigDict( - extra="allow", - ) - +from . import total TotalsCreateRequest = TypeAliasType( "TotalsCreateRequest", - Annotated[ - list[TotalsCreateRequestItem] | TotalsCreateRequest1, - Field(..., title="Totals Create Request"), - ], + Annotated[list[total.Total], Field(..., title="Totals Create Request")], ) """ Pricing breakdown provided by the business. MUST contain exactly one subtotal and one total entry. Detail types (tax, fee, discount, fulfillment) may appear multiple times for itemization. Platforms MUST render all entries in order using display_text and amount. diff --git a/src/ucp_sdk/models/schemas/shopping/types/totals_update_request.py b/src/ucp_sdk/models/schemas/shopping/types/totals_update_request.py index 58251b2..1bb995d 100644 --- a/src/ucp_sdk/models/schemas/shopping/types/totals_update_request.py +++ b/src/ucp_sdk/models/schemas/shopping/types/totals_update_request.py @@ -20,54 +20,14 @@ from typing import Annotated -from pydantic import BaseModel, ConfigDict, Field +from pydantic import Field from typing_extensions import TypeAliasType -from . import signed_amount -from .total import Total - - -class Line(BaseModel): - """ - Sub-line entry. Additional metadata MAY be included. - """ - - model_config = ConfigDict( - extra="allow", - ) - display_text: str - """ - Human-readable label for this sub-line. - """ - amount: signed_amount.SignedAmount - - -class TotalsUpdateRequestItem(Total): - model_config = ConfigDict( - extra="allow", - ) - lines: list[Line] | None = None - """ - Optional itemized breakdown. The parent entry is always rendered; lines are supplementary. Sum of line amounts MUST equal the parent entry amount. - """ - - -class TotalsUpdateRequest1(BaseModel): - """ - Pricing breakdown provided by the business. MUST contain exactly one subtotal and one total entry. Detail types (tax, fee, discount, fulfillment) may appear multiple times for itemization. Platforms MUST render all entries in order using display_text and amount. - """ - - model_config = ConfigDict( - extra="allow", - ) - +from . import total TotalsUpdateRequest = TypeAliasType( "TotalsUpdateRequest", - Annotated[ - list[TotalsUpdateRequestItem] | TotalsUpdateRequest1, - Field(..., title="Totals Update Request"), - ], + Annotated[list[total.Total], Field(..., title="Totals Update Request")], ) """ Pricing breakdown provided by the business. MUST contain exactly one subtotal and one total entry. Detail types (tax, fee, discount, fulfillment) may appear multiple times for itemization. Platforms MUST render all entries in order using display_text and amount.