dub 0.34.1__py3-none-any.whl → 0.35.0__py3-none-any.whl

dub/models/operations/listfolders.py

@@ -1,9 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from dub.types import BaseModel
+from dub.types import BaseModel, UNSET_SENTINEL
 from dub.utils import FieldMetadata, QueryParamMetadata
 import pydantic
+from pydantic import model_serializer
 from typing import Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -36,3 +37,19 @@ class ListFoldersRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = 50
     r"""The number of items per page."""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["search", "page", "pageSize"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m

dub/models/operations/listpartners.py

@@ -2,12 +2,12 @@
 
 from __future__ import annotations
 from dub.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from dub.utils import FieldMetadata, QueryParamMetadata
+from dub.utils import FieldMetadata, QueryParamMetadata, get_discriminator
 from enum import Enum
 import pydantic
-from pydantic import model_serializer
-from typing import List, Optional
-from typing_extensions import Annotated, NotRequired, TypedDict
+from pydantic import Discriminator, Tag, model_serializer
+from typing import List, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class ListPartnersQueryParamStatus(str, Enum):
@@ -128,6 +128,34 @@ class ListPartnersRequest(BaseModel):
     ] = 100
     r"""The number of items per page."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(
+            [
+                "status",
+                "country",
+                "sortBy",
+                "sortOrder",
+                "email",
+                "tenantId",
+                "search",
+                "page",
+                "pageSize",
+            ]
+        )
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 class ListPartnersStatus(str, Enum):
     r"""The status of the partner's enrollment in the program."""
@@ -196,6 +224,22 @@ class ListPartnersLinks(BaseModel):
     sale_amount: Annotated[Optional[float], pydantic.Field(alias="saleAmount")] = 0
     r"""The total dollar value of sales (in cents) generated by the short link."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["clicks", "leads", "conversions", "sales", "saleAmount"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 class ListPartnersBannedReason(str, Enum):
     r"""If the partner was banned from the program, this is the reason for the ban."""
@@ -208,6 +252,381 @@ class ListPartnersBannedReason(str, Enum):
     BRAND_ABUSE = "brand_abuse"
 
 
+class ListPartnersFieldsPartnersResponse200ApplicationJSONResponseBodyReferralFormData8Type(
+    str, Enum
+):
+    PHONE = "phone"
+
+
+class Fields8TypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: ListPartnersFieldsPartnersResponse200ApplicationJSONResponseBodyReferralFormData8Type
+
+
+class Fields8(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: ListPartnersFieldsPartnersResponse200ApplicationJSONResponseBodyReferralFormData8Type
+
+
+class ListPartnersFieldsPartnersResponse200ApplicationJSONResponseBodyReferralFormDataType(
+    str, Enum
+):
+    NUMBER = "number"
+
+
+class Fields7TypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: ListPartnersFieldsPartnersResponse200ApplicationJSONResponseBodyReferralFormDataType
+
+
+class Fields7(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: ListPartnersFieldsPartnersResponse200ApplicationJSONResponseBodyReferralFormDataType
+
+
+class ListPartnersFieldsPartnersResponse200ApplicationJSONResponseBodyType(str, Enum):
+    MULTI_SELECT = "multiSelect"
+
+
+class ListPartnersFieldsPartnersOptionsTypedDict(TypedDict):
+    label: str
+    value: str
+
+
+class ListPartnersFieldsPartnersOptions(BaseModel):
+    label: str
+
+    value: str
+
+
+class Fields6TypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: ListPartnersFieldsPartnersResponse200ApplicationJSONResponseBodyType
+    options: List[ListPartnersFieldsPartnersOptionsTypedDict]
+
+
+class Fields6(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: ListPartnersFieldsPartnersResponse200ApplicationJSONResponseBodyType
+
+    options: List[ListPartnersFieldsPartnersOptions]
+
+
+class ListPartnersFieldsPartnersResponse200ApplicationJSONType(str, Enum):
+    DATE = "date"
+
+
+class Fields5TypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: ListPartnersFieldsPartnersResponse200ApplicationJSONType
+
+
+class Fields5(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: ListPartnersFieldsPartnersResponse200ApplicationJSONType
+
+
+class ListPartnersFieldsPartnersResponse200Type(str, Enum):
+    COUNTRY = "country"
+
+
+class ListPartnersFields4TypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: ListPartnersFieldsPartnersResponse200Type
+
+
+class ListPartnersFields4(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: ListPartnersFieldsPartnersResponse200Type
+
+
+class ListPartnersFieldsPartnersResponseType(str, Enum):
+    SELECT = "select"
+
+
+class ListPartnersFieldsOptionsTypedDict(TypedDict):
+    label: str
+    value: str
+
+
+class ListPartnersFieldsOptions(BaseModel):
+    label: str
+
+    value: str
+
+
+class ListPartnersFields3TypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: ListPartnersFieldsPartnersResponseType
+    options: List[ListPartnersFieldsOptionsTypedDict]
+
+
+class ListPartnersFields3(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: ListPartnersFieldsPartnersResponseType
+
+    options: List[ListPartnersFieldsOptions]
+
+
+class ListPartnersFieldsPartnersType(str, Enum):
+    TEXTAREA = "textarea"
+
+
+class ListPartnersFieldsPartnersConstraintsTypedDict(TypedDict):
+    max_length: NotRequired[int]
+
+
+class ListPartnersFieldsPartnersConstraints(BaseModel):
+    max_length: Annotated[Optional[int], pydantic.Field(alias="maxLength")] = None
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["maxLength"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
+
+class ListPartnersFields2TypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: ListPartnersFieldsPartnersType
+    constraints: NotRequired[ListPartnersFieldsPartnersConstraintsTypedDict]
+
+
+class ListPartnersFields2(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: ListPartnersFieldsPartnersType
+
+    constraints: Optional[ListPartnersFieldsPartnersConstraints] = None
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["constraints"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
+
+class ListPartnersFieldsType(str, Enum):
+    TEXT = "text"
+
+
+class ListPartnersFieldsConstraintsTypedDict(TypedDict):
+    max_length: NotRequired[int]
+    pattern: NotRequired[str]
+
+
+class ListPartnersFieldsConstraints(BaseModel):
+    max_length: Annotated[Optional[int], pydantic.Field(alias="maxLength")] = None
+
+    pattern: Optional[str] = None
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["maxLength", "pattern"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
+
+class ListPartnersFields1TypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: ListPartnersFieldsType
+    constraints: NotRequired[ListPartnersFieldsConstraintsTypedDict]
+
+
+class ListPartnersFields1(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: ListPartnersFieldsType
+
+    constraints: Optional[ListPartnersFieldsConstraints] = None
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["constraints"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
+
+ListPartnersFieldsTypedDict = TypeAliasType(
+    "ListPartnersFieldsTypedDict",
+    Union[
+        ListPartnersFields4TypedDict,
+        Fields5TypedDict,
+        Fields7TypedDict,
+        Fields8TypedDict,
+        ListPartnersFields1TypedDict,
+        ListPartnersFields2TypedDict,
+        ListPartnersFields3TypedDict,
+        Fields6TypedDict,
+    ],
+)
+
+
+ListPartnersFields = Annotated[
+    Union[
+        Annotated[ListPartnersFields1, Tag("text")],
+        Annotated[ListPartnersFields2, Tag("textarea")],
+        Annotated[ListPartnersFields3, Tag("select")],
+        Annotated[ListPartnersFields4, Tag("country")],
+        Annotated[Fields5, Tag("date")],
+        Annotated[Fields6, Tag("multiSelect")],
+        Annotated[Fields7, Tag("number")],
+        Annotated[Fields8, Tag("phone")],
+    ],
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]
+
+
+class ListPartnersReferralFormDataTypedDict(TypedDict):
+    fields: List[ListPartnersFieldsTypedDict]
+
+
+class ListPartnersReferralFormData(BaseModel):
+    fields: List[ListPartnersFields]
+
+
 class ListPartnersResponseBodyTypedDict(TypedDict):
     id: str
     r"""The partner's unique ID on Dub."""
@@ -256,6 +675,7 @@ class ListPartnersResponseBodyTypedDict(TypedDict):
     r"""If the partner was banned from the program, this is the date of the ban."""
     banned_reason: NotRequired[Nullable[ListPartnersBannedReason]]
     r"""If the partner was banned from the program, this is the reason for the ban."""
+    referral_form_data: NotRequired[Nullable[ListPartnersReferralFormDataTypedDict]]
     total_clicks: NotRequired[float]
     r"""The total number of clicks on the partner's links"""
     total_leads: NotRequired[float]
@@ -386,6 +806,11 @@ class ListPartnersResponseBody(BaseModel):
     ] = UNSET
     r"""If the partner was banned from the program, this is the reason for the ban."""
 
+    referral_form_data: Annotated[
+        OptionalNullable[ListPartnersReferralFormData],
+        pydantic.Field(alias="referralFormData"),
+    ] = UNSET
+
     total_clicks: Annotated[Optional[float], pydantic.Field(alias="totalClicks")] = 0
     r"""The total number of clicks on the partner's links"""
 
@@ -458,91 +883,92 @@ class ListPartnersResponseBody(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = [
-            "description",
-            "groupId",
-            "totalCommissions",
-            "clickRewardId",
-            "leadRewardId",
-            "saleRewardId",
-            "discountId",
-            "applicationId",
-            "bannedAt",
-            "bannedReason",
-            "totalClicks",
-            "totalLeads",
-            "totalConversions",
-            "totalSales",
-            "totalSaleAmount",
-            "netRevenue",
-            "earningsPerClick",
-            "averageLifetimeValue",
-            "clickToLeadRate",
-            "clickToConversionRate",
-            "leadToConversionRate",
-            "returnOnAdSpend",
-            "website",
-            "youtube",
-            "twitter",
-            "linkedin",
-            "instagram",
-            "tiktok",
-        ]
-        nullable_fields = [
-            "companyName",
-            "email",
-            "image",
-            "description",
-            "country",
-            "paypalEmail",
-            "stripeConnectId",
-            "payoutsEnabledAt",
-            "trustedAt",
-            "groupId",
-            "tenantId",
-            "links",
-            "clickRewardId",
-            "leadRewardId",
-            "saleRewardId",
-            "discountId",
-            "applicationId",
-            "bannedAt",
-            "bannedReason",
-            "earningsPerClick",
-            "averageLifetimeValue",
-            "clickToLeadRate",
-            "clickToConversionRate",
-            "leadToConversionRate",
-            "returnOnAdSpend",
-            "website",
-            "youtube",
-            "twitter",
-            "linkedin",
-            "instagram",
-            "tiktok",
-        ]
-        null_default_fields = []
-
+        optional_fields = set(
+            [
+                "description",
+                "groupId",
+                "totalCommissions",
+                "clickRewardId",
+                "leadRewardId",
+                "saleRewardId",
+                "discountId",
+                "applicationId",
+                "bannedAt",
+                "bannedReason",
+                "referralFormData",
+                "totalClicks",
+                "totalLeads",
+                "totalConversions",
+                "totalSales",
+                "totalSaleAmount",
+                "netRevenue",
+                "earningsPerClick",
+                "averageLifetimeValue",
+                "clickToLeadRate",
+                "clickToConversionRate",
+                "leadToConversionRate",
+                "returnOnAdSpend",
+                "website",
+                "youtube",
+                "twitter",
+                "linkedin",
+                "instagram",
+                "tiktok",
+            ]
+        )
+        nullable_fields = set(
+            [
+                "companyName",
+                "email",
+                "image",
+                "description",
+                "country",
+                "paypalEmail",
+                "stripeConnectId",
+                "payoutsEnabledAt",
+                "trustedAt",
+                "groupId",
+                "tenantId",
+                "links",
+                "clickRewardId",
+                "leadRewardId",
+                "saleRewardId",
+                "discountId",
+                "applicationId",
+                "bannedAt",
+                "bannedReason",
+                "referralFormData",
+                "earningsPerClick",
+                "averageLifetimeValue",
+                "clickToLeadRate",
+                "clickToConversionRate",
+                "leadToConversionRate",
+                "returnOnAdSpend",
+                "website",
+                "youtube",
+                "twitter",
+                "linkedin",
+                "instagram",
+                "tiktok",
+            ]
+        )
         serialized = handler(self)
-
         m = {}
 
         for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
-            serialized.pop(k, None)
-
-            optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (
-                self.__pydantic_fields_set__.intersection({n})
-                or k in null_default_fields
-            )  # pylint: disable=no-member
-
-            if val is not None and val != UNSET_SENTINEL:
-                m[k] = val
-            elif val != UNSET_SENTINEL and (
-                not k in optional_fields or (optional_nullable and is_set)
-            ):
-                m[k] = val
+            is_nullable_and_explicitly_set = (
+                k in nullable_fields
+                and (self.__pydantic_fields_set__.intersection({n}))  # pylint: disable=no-member
+            )
+
+            if val != UNSET_SENTINEL:
+                if (
+                    val is not None
+                    or k not in optional_fields
+                    or is_nullable_and_explicitly_set
+                ):
+                    m[k] = val
 
         return m