retab 0.0.41__py3-none-any.whl → 0.0.43__py3-none-any.whl

retab/resources/processors/automations/mailboxes.py

@@ -3,7 +3,7 @@ from pathlib import Path
 from typing import Any, List, Literal, Optional
 
 from pydantic import EmailStr, HttpUrl
-from pydantic_core import PydanticUndefined
+from ....types.standards import FieldUnset
 
 from ...._resource import AsyncAPIResource, SyncAPIResource
 from ....utils.mime import prepare_mime_document
@@ -21,28 +21,36 @@ class MailBoxesMixin:
         name: str,
         processor_id: str,
         webhook_url: str,
-        authorized_domains: list[str] = PydanticUndefined,  # type: ignore[assignment]
-        authorized_emails: list[EmailStr] = PydanticUndefined,  # type: ignore[assignment]
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        default_language: str = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
+        authorized_domains: list[str] = FieldUnset,
+        authorized_emails: list[EmailStr] = FieldUnset,
+        webhook_headers: dict[str, str] = FieldUnset,
+        default_language: str = FieldUnset,
+        need_validation: bool = FieldUnset,
     ) -> PreparedRequest:
         """Create a new email automation configuration."""
-        mailbox = Mailbox(
-            name=name,
-            processor_id=processor_id,
-            default_language=default_language,
-            webhook_url=webhook_url,
-            webhook_headers=webhook_headers,
-            need_validation=need_validation,
-            email=email,
-            authorized_domains=authorized_domains,
-            authorized_emails=authorized_emails,
-        )
-        return PreparedRequest(method="POST", url=self.mailboxes_base_url, data=mailbox.model_dump(mode="json"))
+        mailbox_dict: dict[str, Any] = {
+            'name': name,
+            'processor_id': processor_id,
+            'webhook_url': webhook_url,
+            'email': email,
+        }
+        if default_language is not FieldUnset:
+            mailbox_dict['default_language'] = default_language
+        if webhook_headers is not FieldUnset:
+            mailbox_dict['webhook_headers'] = webhook_headers
+        if need_validation is not FieldUnset:
+            mailbox_dict['need_validation'] = need_validation
+        if authorized_domains is not FieldUnset:
+            mailbox_dict['authorized_domains'] = authorized_domains
+        if authorized_emails is not FieldUnset:
+            mailbox_dict['authorized_emails'] = authorized_emails
+
+        mailbox = Mailbox(**mailbox_dict)
+        return PreparedRequest(method="POST", url=self.mailboxes_base_url, data=mailbox.model_dump(mode="json", exclude_unset=True))
 
     def prepare_list(
         self,
+        processor_id: str,
         before: str | None = None,
         after: str | None = None,
         limit: int = 10,
@@ -52,6 +60,7 @@ class MailBoxesMixin:
         webhook_url: Optional[str] = None,
     ) -> PreparedRequest:
         params = {
+            "processor_id": processor_id,
             "before": before,
             "after": after,
             "limit": limit,
@@ -65,36 +74,41 @@ class MailBoxesMixin:
 
         return PreparedRequest(method="GET", url=self.mailboxes_base_url, params=params)
 
-    def prepare_get(self, email: str) -> PreparedRequest:
-        return PreparedRequest(method="GET", url=f"{self.mailboxes_base_url}/{email}")
-
-    def prepare_get_from_id(self, mailbox_id: str) -> PreparedRequest:
-        return PreparedRequest(method="GET", url=f"{self.mailboxes_base_url}/from_id/{mailbox_id}")
+    def prepare_get(self, mailbox_id: str) -> PreparedRequest:
+        return PreparedRequest(method="GET", url=f"{self.mailboxes_base_url}/{mailbox_id}")
 
     def prepare_update(
         self,
-        email: str,
-        name: str = PydanticUndefined,  # type: ignore[assignment]
-        default_language: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_url: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
-        authorized_domains: list[str] = PydanticUndefined,  # type: ignore[assignment]
-        authorized_emails: list[str] = PydanticUndefined,  # type: ignore[assignment]
+        mailbox_id: str,
+        name: str = FieldUnset,
+        default_language: str = FieldUnset,
+        webhook_url: str = FieldUnset,
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
+        authorized_domains: list[str] = FieldUnset,
+        authorized_emails: list[str] = FieldUnset,
     ) -> PreparedRequest:
-        update_mailbox_request = UpdateMailboxRequest(
-            name=name,
-            default_language=default_language,
-            webhook_url=webhook_url,
-            webhook_headers=webhook_headers,
-            need_validation=need_validation,
-            authorized_domains=authorized_domains,
-            authorized_emails=authorized_emails,
-        )
-        return PreparedRequest(method="PUT", url=f"/v1/processors/automations/mailboxes/{email}", data=update_mailbox_request.model_dump(mode="json"))
-
-    def prepare_delete(self, email: str) -> PreparedRequest:
-        return PreparedRequest(method="DELETE", url=f"/v1/processors/automations/mailboxes/{email}", raise_for_status=True)
+        update_dict: dict[str, Any] = {}
+        if name is not FieldUnset:
+            update_dict['name'] = name
+        if default_language is not FieldUnset:
+            update_dict['default_language'] = default_language
+        if webhook_url is not FieldUnset:
+            update_dict['webhook_url'] = webhook_url
+        if webhook_headers is not FieldUnset:
+            update_dict['webhook_headers'] = webhook_headers
+        if need_validation is not FieldUnset:
+            update_dict['need_validation'] = need_validation
+        if authorized_domains is not FieldUnset:
+            update_dict['authorized_domains'] = authorized_domains
+        if authorized_emails is not FieldUnset:
+            update_dict['authorized_emails'] = authorized_emails
+
+        update_mailbox_request = UpdateMailboxRequest(**update_dict)
+        return PreparedRequest(method="PUT", url=f"/v1/processors/automations/mailboxes/{mailbox_id}", data=update_mailbox_request.model_dump(mode="json", exclude_unset=True))
+
+    def prepare_delete(self, mailbox_id: str) -> PreparedRequest:
+        return PreparedRequest(method="DELETE", url=f"/v1/processors/automations/mailboxes/{mailbox_id}", raise_for_status=True)
 
 
 class Mailboxes(SyncAPIResource, MailBoxesMixin):
@@ -110,11 +124,11 @@ class Mailboxes(SyncAPIResource, MailBoxesMixin):
         name: str,
         webhook_url: str,
         processor_id: str,
-        authorized_domains: list[str] = PydanticUndefined,  # type: ignore[assignment]
-        authorized_emails: list[EmailStr] = PydanticUndefined,  # type: ignore[assignment]
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        default_language: str = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
+        authorized_domains: list[str] = FieldUnset,
+        authorized_emails: list[EmailStr] = FieldUnset,
+        webhook_headers: dict[str, str] = FieldUnset,
+        default_language: str = FieldUnset,
+        need_validation: bool = FieldUnset,
     ) -> Mailbox:
         """Create a new email automation configuration.
 
@@ -144,12 +158,13 @@ class Mailboxes(SyncAPIResource, MailBoxesMixin):
         )
         response = self._client._prepared_request(request)
 
-        print(f"Mailbox {response['email']} created. Url: https://www.retab.dev/dashboard/processors/automations/{response['id']}")
+        print(f"Mailbox {response['email']} created. Url: https://www.retab.com/dashboard/processors/automations/{response['id']}")
 
         return Mailbox.model_validate(response)
 
     def list(
         self,
+        processor_id: str,
         before: str | None = None,
         after: str | None = None,
         limit: int = 10,
@@ -174,6 +189,7 @@ class Mailboxes(SyncAPIResource, MailBoxesMixin):
             ListMailboxes: List of mailbox configurations
         """
         request = self.prepare_list(
+            processor_id=processor_id,
             before=before,
             after=after,
             limit=limit,
@@ -185,29 +201,29 @@ class Mailboxes(SyncAPIResource, MailBoxesMixin):
         response = self._client._prepared_request(request)
         return ListMailboxes.model_validate(response)
 
-    def get(self, email: str) -> Mailbox:
+    def get(self, mailbox_id: str) -> Mailbox:
         """Get a specific email automation configuration.
 
         Args:
-            email: Email address of the mailbox
+            mailbox_id: ID of the mailbox
 
         Returns:
             Mailbox: The mailbox configuration
         """
-        request = self.prepare_get(email)
+        request = self.prepare_get(mailbox_id)
         response = self._client._prepared_request(request)
         return Mailbox.model_validate(response)
 
     def update(
         self,
-        email: str,
-        name: str = PydanticUndefined,  # type: ignore[assignment]
-        default_language: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_url: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
-        authorized_domains: List[str] = PydanticUndefined,  # type: ignore[assignment]
-        authorized_emails: List[str] = PydanticUndefined,  # type: ignore[assignment]
+        mailbox_id: str,
+        name: str = FieldUnset,
+        default_language: str = FieldUnset,
+        webhook_url: str = FieldUnset,
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
+        authorized_domains: List[str] = FieldUnset,
+        authorized_emails: List[str] = FieldUnset,
     ) -> Mailbox:
         """Update an email automation configuration.
 
@@ -225,7 +241,7 @@ class Mailboxes(SyncAPIResource, MailBoxesMixin):
             Mailbox: The updated mailbox configuration
         """
         request = self.prepare_update(
-            email=email,
+            mailbox_id=mailbox_id,
             name=name,
             default_language=default_language,
             webhook_url=webhook_url,
@@ -237,13 +253,13 @@ class Mailboxes(SyncAPIResource, MailBoxesMixin):
         response = self._client._prepared_request(request)
         return Mailbox.model_validate(response)
 
-    def delete(self, email: str) -> None:
+    def delete(self, mailbox_id: str) -> None:
         """Delete an email automation configuration.
 
         Args:
             email: Email address of the mailbox to delete
         """
-        request = self.prepare_delete(email)
+        request = self.prepare_delete(mailbox_id)
         self._client._prepared_request(request)
         return None
 
@@ -259,11 +275,11 @@ class AsyncMailboxes(AsyncAPIResource, MailBoxesMixin):
         name: str,
         webhook_url: str,
         processor_id: str,
-        authorized_domains: List[str] = PydanticUndefined,  # type: ignore[assignment]
-        authorized_emails: List[EmailStr] = PydanticUndefined,  # type: ignore[assignment]
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        default_language: str = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
+        authorized_domains: List[str] = FieldUnset,
+        authorized_emails: List[EmailStr] = FieldUnset,
+        webhook_headers: dict[str, str] = FieldUnset,
+        default_language: str = FieldUnset,
+        need_validation: bool = FieldUnset,
     ) -> Mailbox:
         request = self.prepare_create(
             email=email,
@@ -278,12 +294,13 @@ class AsyncMailboxes(AsyncAPIResource, MailBoxesMixin):
         )
         response = await self._client._prepared_request(request)
 
-        print(f"Mailbox {response['email']} created. Url: https://www.retab.dev/dashboard/processors/automations/{response['id']}")
+        print(f"Mailbox {response['email']} created. Url: https://www.retab.com/dashboard/processors/automations/{response['id']}")
 
         return Mailbox.model_validate(response)
 
     async def list(
         self,
+        processor_id: str,
         before: str | None = None,
         after: str | None = None,
         limit: int = 10,
@@ -292,28 +309,37 @@ class AsyncMailboxes(AsyncAPIResource, MailBoxesMixin):
         name: str | None = None,
         webhook_url: str | None = None,
     ) -> ListMailboxes:
-        request = self.prepare_list(before=before, after=after, limit=limit, order=order, email=email, name=name, webhook_url=webhook_url)
+        request = self.prepare_list(
+            processor_id=processor_id,
+            before=before,
+            after=after,
+            limit=limit,
+            order=order,
+            email=email,
+            name=name,
+            webhook_url=webhook_url,
+        )
         response = await self._client._prepared_request(request)
         return ListMailboxes.model_validate(response)
 
-    async def get(self, email: str) -> Mailbox:
-        request = self.prepare_get(email)
+    async def get(self, mailbox_id: str) -> Mailbox:
+        request = self.prepare_get(mailbox_id)
         response = await self._client._prepared_request(request)
         return Mailbox.model_validate(response)
 
     async def update(
         self,
-        email: str,
-        name: str = PydanticUndefined,  # type: ignore[assignment]
-        default_language: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_url: str = PydanticUndefined,  # type: ignore[assignment]
-        webhook_headers: dict[str, str] = PydanticUndefined,  # type: ignore[assignment]
-        need_validation: bool = PydanticUndefined,  # type: ignore[assignment]
-        authorized_domains: List[str] = PydanticUndefined,  # type: ignore[assignment]
-        authorized_emails: List[str] = PydanticUndefined,  # type: ignore[assignment]
+        mailbox_id: str,
+        name: str = FieldUnset,
+        default_language: str = FieldUnset,
+        webhook_url: str = FieldUnset,
+        webhook_headers: dict[str, str] = FieldUnset,
+        need_validation: bool = FieldUnset,
+        authorized_domains: List[str] = FieldUnset,
+        authorized_emails: List[str] = FieldUnset,
     ) -> Mailbox:
         request = self.prepare_update(
-            email=email,
+            mailbox_id=mailbox_id,
             name=name,
             default_language=default_language,
             webhook_url=webhook_url,
@@ -325,8 +351,8 @@ class AsyncMailboxes(AsyncAPIResource, MailBoxesMixin):
         response = await self._client._prepared_request(request)
         return Mailbox.model_validate(response)
 
-    async def delete(self, email: str) -> None:
-        request = self.prepare_delete(email)
+    async def delete(self, mailbox_id: str) -> None:
+        request = self.prepare_delete(mailbox_id)
         await self._client._prepared_request(request)
         return None
 
@@ -334,12 +360,12 @@ class AsyncMailboxes(AsyncAPIResource, MailBoxesMixin):
 class TestMailboxesMixin:
     def prepare_forward(
         self,
-        email: str,
+        mailbox_id: str,
         document: Path | str | IOBase | HttpUrl | MIMEData,
         verbose: bool = True,
     ) -> PreparedRequest:
         mime_document = prepare_mime_document(document)
-        return PreparedRequest(method="POST", url=f"/v1/processors/automations/mailboxes/tests/forward/{email}", data={"document": mime_document.model_dump()})
+        return PreparedRequest(method="POST", url=f"/v1/processors/automations/mailboxes/tests/forward/{mailbox_id}", data={"document": mime_document.model_dump()})
 
     def print_forward_verbose(self, email_data: EmailData) -> None:
         print("\nTEST EMAIL FORWARDING RESULTS:")
@@ -360,7 +386,7 @@ class TestMailboxesMixin:
 class TestMailboxes(SyncAPIResource, TestMailboxesMixin):
     def forward(
         self,
-        email: str,
+        mailbox_id: str,
         document: Path | str | IOBase | HttpUrl | MIMEData,
         verbose: bool = True,
     ) -> EmailData:
@@ -372,7 +398,7 @@ class TestMailboxes(SyncAPIResource, TestMailboxesMixin):
         Returns:
             DocumentExtractResponse: The simulated extraction response
         """
-        request = self.prepare_forward(email, document, verbose)
+        request = self.prepare_forward(mailbox_id, document, verbose)
         response = self._client._prepared_request(request)
 
         email_data = EmailData.model_validate(response)
@@ -385,11 +411,11 @@ class TestMailboxes(SyncAPIResource, TestMailboxesMixin):
 class AsyncTestMailboxes(AsyncAPIResource, TestMailboxesMixin):
     async def forward(
         self,
-        email: str,
+        mailbox_id: str,
         document: Path | str | IOBase | HttpUrl | MIMEData,
         verbose: bool = True,
     ) -> EmailData:
-        request = self.prepare_forward(email, document, verbose)
+        request = self.prepare_forward(mailbox_id, document, verbose)
         response = await self._client._prepared_request(request)
         email_data = EmailData.model_validate(response)
         if verbose: