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

dub/models/components/partnerenrolledevent.py

@@ -2,11 +2,12 @@
 
 from __future__ import annotations
 from dub.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from dub.utils import 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 PartnerEnrolledEventEvent(str, Enum):
@@ -80,6 +81,22 @@ class Links(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 BannedReason(str, Enum):
     r"""If the partner was banned from the program, this is the reason for the ban."""
@@ -92,6 +109,377 @@ class BannedReason(str, Enum):
     BRAND_ABUSE = "brand_abuse"
 
 
+class PartnerEnrolledEventFieldsDataReferralFormData8Type(str, Enum):
+    PHONE = "phone"
+
+
+class EightTypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: PartnerEnrolledEventFieldsDataReferralFormData8Type
+
+
+class Eight(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: PartnerEnrolledEventFieldsDataReferralFormData8Type
+
+
+class PartnerEnrolledEventFieldsDataReferralFormData7Type(str, Enum):
+    NUMBER = "number"
+
+
+class SevenTypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: PartnerEnrolledEventFieldsDataReferralFormData7Type
+
+
+class Seven(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: PartnerEnrolledEventFieldsDataReferralFormData7Type
+
+
+class PartnerEnrolledEventFieldsDataReferralFormData6Type(str, Enum):
+    MULTI_SELECT = "multiSelect"
+
+
+class FieldsOptionsTypedDict(TypedDict):
+    label: str
+    value: str
+
+
+class FieldsOptions(BaseModel):
+    label: str
+
+    value: str
+
+
+class SixTypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: PartnerEnrolledEventFieldsDataReferralFormData6Type
+    options: List[FieldsOptionsTypedDict]
+
+
+class Six(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: PartnerEnrolledEventFieldsDataReferralFormData6Type
+
+    options: List[FieldsOptions]
+
+
+class PartnerEnrolledEventFieldsDataReferralFormData5Type(str, Enum):
+    DATE = "date"
+
+
+class FiveTypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: PartnerEnrolledEventFieldsDataReferralFormData5Type
+
+
+class Five(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: PartnerEnrolledEventFieldsDataReferralFormData5Type
+
+
+class PartnerEnrolledEventFieldsDataReferralFormDataType(str, Enum):
+    COUNTRY = "country"
+
+
+class FourTypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: PartnerEnrolledEventFieldsDataReferralFormDataType
+
+
+class Four(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: PartnerEnrolledEventFieldsDataReferralFormDataType
+
+
+class PartnerEnrolledEventFieldsDataType(str, Enum):
+    SELECT = "select"
+
+
+class OptionsTypedDict(TypedDict):
+    label: str
+    value: str
+
+
+class Options(BaseModel):
+    label: str
+
+    value: str
+
+
+class Fields3TypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: PartnerEnrolledEventFieldsDataType
+    options: List[OptionsTypedDict]
+
+
+class Fields3(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: PartnerEnrolledEventFieldsDataType
+
+    options: List[Options]
+
+
+class PartnerEnrolledEventFieldsType(str, Enum):
+    TEXTAREA = "textarea"
+
+
+class FieldsConstraintsTypedDict(TypedDict):
+    max_length: NotRequired[int]
+
+
+class FieldsConstraints(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 Fields2TypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: PartnerEnrolledEventFieldsType
+    constraints: NotRequired[FieldsConstraintsTypedDict]
+
+
+class Fields2(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: PartnerEnrolledEventFieldsType
+
+    constraints: Optional[FieldsConstraints] = 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 FieldsType(str, Enum):
+    TEXT = "text"
+
+
+class ConstraintsTypedDict(TypedDict):
+    max_length: NotRequired[int]
+    pattern: NotRequired[str]
+
+
+class Constraints(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 Fields1TypedDict(TypedDict):
+    key: str
+    label: str
+    required: bool
+    locked: bool
+    position: int
+    type: FieldsType
+    constraints: NotRequired[ConstraintsTypedDict]
+
+
+class Fields1(BaseModel):
+    key: str
+
+    label: str
+
+    required: bool
+
+    locked: bool
+
+    position: int
+
+    type: FieldsType
+
+    constraints: Optional[Constraints] = 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
+
+
+FieldsTypedDict = TypeAliasType(
+    "FieldsTypedDict",
+    Union[
+        FourTypedDict,
+        FiveTypedDict,
+        SevenTypedDict,
+        EightTypedDict,
+        Fields1TypedDict,
+        Fields2TypedDict,
+        Fields3TypedDict,
+        SixTypedDict,
+    ],
+)
+
+
+Fields = Annotated[
+    Union[
+        Annotated[Fields1, Tag("text")],
+        Annotated[Fields2, Tag("textarea")],
+        Annotated[Fields3, Tag("select")],
+        Annotated[Four, Tag("country")],
+        Annotated[Five, Tag("date")],
+        Annotated[Six, Tag("multiSelect")],
+        Annotated[Seven, Tag("number")],
+        Annotated[Eight, Tag("phone")],
+    ],
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]
+
+
+class ReferralFormDataTypedDict(TypedDict):
+    fields: List[FieldsTypedDict]
+
+
+class ReferralFormData(BaseModel):
+    fields: List[Fields]
+
+
 class PartnerEnrolledEventDataTypedDict(TypedDict):
     id: str
     r"""The partner's unique ID on Dub."""
@@ -140,6 +528,7 @@ class PartnerEnrolledEventDataTypedDict(TypedDict):
     r"""If the partner was banned from the program, this is the date of the ban."""
     banned_reason: NotRequired[Nullable[BannedReason]]
     r"""If the partner was banned from the program, this is the reason for the ban."""
+    referral_form_data: NotRequired[Nullable[ReferralFormDataTypedDict]]
     total_clicks: NotRequired[float]
     r"""The total number of clicks on the partner's links"""
     total_leads: NotRequired[float]
@@ -270,6 +659,10 @@ class PartnerEnrolledEventData(BaseModel):
     ] = UNSET
     r"""If the partner was banned from the program, this is the reason for the ban."""
 
+    referral_form_data: Annotated[
+        OptionalNullable[ReferralFormData], 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"""
 
@@ -342,92 +735,93 @@ class PartnerEnrolledEventData(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