Add new methods.

Author: koval

huntflow_api_client/entities/user_settings.py

@@ -1,6 +1,12 @@
 from huntflow_api_client.entities.base import BaseEntity
+from huntflow_api_client.models.common import StatusResponse
+from huntflow_api_client.models.request.user_settings import (
+    ExchangeEmailAccountRequest,
+    OtherEmailAccountRequest,
+)
 from huntflow_api_client.models.response.user_settings import (
     CalendarAccountsListResponse,
+    EmailAccount,
     EmailAccountsListResponse,
 )
 
@@ -23,3 +29,87 @@ async def get_calendar_accounts(self) -> CalendarAccountsListResponse:
         """
         response = await self._api.request("GET", "/calendar_accounts")
         return CalendarAccountsListResponse.model_validate(response.json())
+
+    async def create_exchange_email_account(
+        self,
+        data: ExchangeEmailAccountRequest,
+    ) -> EmailAccount:
+        """
+        API method reference https://api.huntflow.ai/v2/docs#post-/email_accounts/exchange
+
+        :param data: Exchange user email account data
+        :return: Created Exchange user email account for API robot.
+        """
+        response = await self._api.request(
+            "POST",
+            "/email_accounts/exchange",
+            json=data.jsonable_dict(exclude_none=True),
+        )
+        return EmailAccount.model_validate(response.json())
+
+    async def update_exchange_email_account(
+        self,
+        data: ExchangeEmailAccountRequest,
+        account_email_id: int,
+    ) -> EmailAccount:
+        """
+        API method reference
+            https://api.huntflow.ai/v2/docs#put-/email_accounts/exchange/-account_email_id-
+
+        :param account_email_id: Email account ID
+        :param data: Exchange user email account data
+        :return: Updated Exchange user email account for API robot.
+        """
+        response = await self._api.request(
+            "PUT",
+            f"/email_accounts/exchange/{account_email_id}",
+            json=data.jsonable_dict(exclude_none=True),
+        )
+        return EmailAccount.model_validate(response.json())
+
+    async def create_other_email_account(
+        self,
+        data: OtherEmailAccountRequest,
+    ) -> EmailAccount:
+        """
+        API method reference https://api.huntflow.ai/v2/docs#post-/email_accounts/other
+
+        :param data: SMTP/POP3/IMAP user email account data
+        :return: Created SMTP/POP3/IMAP user email account for API robot.
+        """
+        response = await self._api.request(
+            "POST",
+            "/email_accounts/other",
+            json=data.jsonable_dict(exclude_none=True),
+        )
+        return EmailAccount.model_validate(response.json())
+
+    async def update_other_email_account(
+        self,
+        data: OtherEmailAccountRequest,
+        account_email_id: int,
+    ) -> EmailAccount:
+        """
+        API method reference
+            https://api.huntflow.ai/v2/docs#put-/email_accounts/other/-account_email_id-
+
+        :param account_email_id: Email account ID
+        :param data: SMTP/POP3/IMAP user email account data
+        :return: Updated SMTP/POP3/IMAP user email account for API robot.
+        """
+        response = await self._api.request(
+            "PUT",
+            f"/email_accounts/other/{account_email_id}",
+            json=data.jsonable_dict(exclude_none=True),
+        )
+        return EmailAccount.model_validate(response.json())
+
+    async def delete_email_account(self, account_email_id: int) -> StatusResponse:
+        """
+        API method reference
+            https://api.huntflow.ai/v2/docs#delete-/email_accounts/-account_email_id-
+
+        :param account_email_id: Email account ID
+        """
+        response = await self._api.request("DELETE", f"/email_accounts/{account_email_id}")
+        return StatusResponse.model_validate(response.json())

huntflow_api_client/models/consts.py

@@ -196,3 +196,13 @@ class RecommendationStatus(str, Enum):
 class InterviewType(str, Enum):
     USER = "user"
     INTERVIEW = "interview"
+
+
+class ExchangeAccessType(str, Enum):
+    DEFAULT = "DEFAULT"
+    IMPERSONATION = "IMPERSONATION"
+
+
+class EmailInboundType(str, Enum):
+    DEFAULT = "IMAP"
+    IMPERSONATION = "POP3"

huntflow_api_client/models/request/user_settings.py

@@ -0,0 +1,35 @@
+from typing import Optional
+
+from pydantic import AnyHttpUrl, EmailStr, Field, PositiveInt
+
+from huntflow_api_client.models.common import JsonRequestModel
+from huntflow_api_client.models.consts import EmailInboundType, ExchangeAccessType
+
+
+class ExchangeEmailAccountRequest(JsonRequestModel):
+    """
+    Model for EXCHANGE email connection
+    """
+
+    access_type: ExchangeAccessType = Field(..., description="Exchange access type")
+    email: EmailStr = Field(..., description="Email address")
+    ews_url: AnyHttpUrl = Field(..., description="Exchange server URL")
+    password: str = Field(..., description="Exchange password")
+    user: Optional[str] = Field(None, description="Exchange user name")
+
+
+class OtherEmailAccountRequest(JsonRequestModel):
+    """
+    Model for IMAP/POP3 & SMTP email connection
+    """
+
+    email: EmailStr = Field(..., description="Email address")
+    password: str = Field(..., description="Mailbox password")
+
+    inhost: str = Field(..., description="Inbound mail server URL without protocol")
+    inport: Optional[PositiveInt] = Field(None, description="Inbound mail port")
+    in_secure: bool = Field(False, description="Inbound mail server uses SSL")
+    outhost: str = Field(..., description="Outbound mail server URL without protocol")
+    outport: Optional[PositiveInt] = Field(None, description="Outbound mail port")
+    out_secure: bool = Field(False, description="Outbound mail server uses SSL")
+    type: EmailInboundType = Field(..., description="Type of inbound mail server")

pyproject.toml

@@ -1,7 +1,7 @@
 
 [project]
 name = "huntflow-api-client"
-version = "2.7.0"
+version = "2.8.0"
 description = "Huntflow API Client for Python"
 authors = [
     {name = "Developers huntflow", email = "developer@huntflow.ru"},