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

dub/models/operations/createpartnerlink.py

@@ -187,71 +187,70 @@ class CreatePartnerLinkLinkProps(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = [
-            "keyLength",
-            "externalId",
-            "tenantId",
-            "prefix",
-            "archived",
-            "tagIds",
-            "tagNames",
-            "comments",
-            "expiresAt",
-            "expiredUrl",
-            "password",
-            "proxy",
-            "title",
-            "description",
-            "image",
-            "video",
-            "rewrite",
-            "ios",
-            "android",
-            "doIndex",
-            "testVariants",
-            "testStartedAt",
-            "testCompletedAt",
-        ]
-        nullable_fields = [
-            "externalId",
-            "tenantId",
-            "comments",
-            "expiresAt",
-            "expiredUrl",
-            "password",
-            "title",
-            "description",
-            "image",
-            "video",
-            "ios",
-            "android",
-            "testVariants",
-            "testStartedAt",
-            "testCompletedAt",
-        ]
-        null_default_fields = []
-
+        optional_fields = set(
+            [
+                "keyLength",
+                "externalId",
+                "tenantId",
+                "prefix",
+                "archived",
+                "tagIds",
+                "tagNames",
+                "comments",
+                "expiresAt",
+                "expiredUrl",
+                "password",
+                "proxy",
+                "title",
+                "description",
+                "image",
+                "video",
+                "rewrite",
+                "ios",
+                "android",
+                "doIndex",
+                "testVariants",
+                "testStartedAt",
+                "testCompletedAt",
+            ]
+        )
+        nullable_fields = set(
+            [
+                "externalId",
+                "tenantId",
+                "comments",
+                "expiresAt",
+                "expiredUrl",
+                "password",
+                "title",
+                "description",
+                "image",
+                "video",
+                "ios",
+                "android",
+                "testVariants",
+                "testStartedAt",
+                "testCompletedAt",
+            ]
+        )
         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
 
@@ -298,37 +297,27 @@ class CreatePartnerLinkRequestBody(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = [
-            "partnerId",
-            "tenantId",
-            "url",
-            "key",
-            "comments",
-            "linkProps",
-        ]
-        nullable_fields = ["partnerId", "tenantId", "url", "comments"]
-        null_default_fields = []
-
+        optional_fields = set(
+            ["partnerId", "tenantId", "url", "key", "comments", "linkProps"]
+        )
+        nullable_fields = set(["partnerId", "tenantId", "url", "comments"])
         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

dub/models/operations/createreferralsembedtoken.py

@@ -189,71 +189,70 @@ class CreateReferralsEmbedTokenLinkProps(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = [
-            "keyLength",
-            "externalId",
-            "tenantId",
-            "prefix",
-            "archived",
-            "tagIds",
-            "tagNames",
-            "comments",
-            "expiresAt",
-            "expiredUrl",
-            "password",
-            "proxy",
-            "title",
-            "description",
-            "image",
-            "video",
-            "rewrite",
-            "ios",
-            "android",
-            "doIndex",
-            "testVariants",
-            "testStartedAt",
-            "testCompletedAt",
-        ]
-        nullable_fields = [
-            "externalId",
-            "tenantId",
-            "comments",
-            "expiresAt",
-            "expiredUrl",
-            "password",
-            "title",
-            "description",
-            "image",
-            "video",
-            "ios",
-            "android",
-            "testVariants",
-            "testStartedAt",
-            "testCompletedAt",
-        ]
-        null_default_fields = []
-
+        optional_fields = set(
+            [
+                "keyLength",
+                "externalId",
+                "tenantId",
+                "prefix",
+                "archived",
+                "tagIds",
+                "tagNames",
+                "comments",
+                "expiresAt",
+                "expiredUrl",
+                "password",
+                "proxy",
+                "title",
+                "description",
+                "image",
+                "video",
+                "rewrite",
+                "ios",
+                "android",
+                "doIndex",
+                "testVariants",
+                "testStartedAt",
+                "testCompletedAt",
+            ]
+        )
+        nullable_fields = set(
+            [
+                "externalId",
+                "tenantId",
+                "comments",
+                "expiresAt",
+                "expiredUrl",
+                "password",
+                "title",
+                "description",
+                "image",
+                "video",
+                "ios",
+                "android",
+                "testVariants",
+                "testStartedAt",
+                "testCompletedAt",
+            ]
+        )
         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
 
@@ -311,40 +310,37 @@ class Partner(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = [
-            "name",
-            "username",
-            "image",
-            "tenantId",
-            "groupId",
-            "country",
-            "description",
-            "linkProps",
-        ]
-        nullable_fields = ["name", "username", "image", "country", "description"]
-        null_default_fields = []
-
+        optional_fields = set(
+            [
+                "name",
+                "username",
+                "image",
+                "tenantId",
+                "groupId",
+                "country",
+                "description",
+                "linkProps",
+            ]
+        )
+        nullable_fields = set(["name", "username", "image", "country", "description"])
         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
 
@@ -362,6 +358,22 @@ class CreateReferralsEmbedTokenRequestBody(BaseModel):
 
     partner: Optional[Partner] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["partnerId", "tenantId", "partner"])
+        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 CreateReferralsEmbedTokenResponseBodyTypedDict(TypedDict):
     r"""The created public embed token."""

dub/models/operations/createtag.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 enum import Enum
 import pydantic
+from pydantic import model_serializer
 from typing import Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -43,3 +44,19 @@ class CreateTagRequestBody(BaseModel):
         ),
     ] = None
     r"""The name of the tag to create."""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name", "color", "tag"])
+        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/deactivatepartner.py

@@ -0,0 +1,65 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from dub.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+import pydantic
+from pydantic import model_serializer
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class DeactivatePartnerRequestBodyTypedDict(TypedDict):
+    partner_id: NotRequired[Nullable[str]]
+    r"""The ID of the partner to create a link for. Will take precedence over `tenantId` if provided."""
+    tenant_id: NotRequired[Nullable[str]]
+    r"""The ID of the partner in your system. If both `partnerId` and `tenantId` are not provided, an error will be thrown."""
+
+
+class DeactivatePartnerRequestBody(BaseModel):
+    partner_id: Annotated[OptionalNullable[str], pydantic.Field(alias="partnerId")] = (
+        UNSET
+    )
+    r"""The ID of the partner to create a link for. Will take precedence over `tenantId` if provided."""
+
+    tenant_id: Annotated[OptionalNullable[str], pydantic.Field(alias="tenantId")] = (
+        UNSET
+    )
+    r"""The ID of the partner in your system. If both `partnerId` and `tenantId` are not provided, an error will be thrown."""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["partnerId", "tenantId"])
+        nullable_fields = set(["partnerId", "tenantId"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+            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
+
+
+class DeactivatePartnerResponseBodyTypedDict(TypedDict):
+    r"""The deactivated partner"""
+
+    partner_id: str
+    r"""The ID of the deactivated partner."""
+
+
+class DeactivatePartnerResponseBody(BaseModel):
+    r"""The deactivated partner"""
+
+    partner_id: Annotated[str, pydantic.Field(alias="partnerId")]
+    r"""The ID of the deactivated partner."""