Merge branch 'v2' into DEV-22316

# Conflicts:
#	pyproject.toml

Author: buschwaker

.github/workflows/autorelease.yml

@@ -8,16 +8,23 @@ jobs:
   autorelease:
     if: github.event.pull_request.merged == true
 
-    runs-on: ubuntu-20.04
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: [ '3.8.16' ]
+
     steps:
       - uses: actions/checkout@v3
 
       - name: Set up python
         uses: actions/setup-python@v4
         with:
-          python-version: 3.8
+          python-version: ${{ matrix.python-version }}
       - name: Set up PDM
         uses: pdm-project/setup-pdm@v3
+        with:
+          python-version: ${{ matrix.python-version }}
+          version: 2.20.1
 
       - name: Install dependencies
         run: pdm install -dG release

.github/workflows/publish.yml

@@ -6,13 +6,27 @@ on:
 
 jobs:
   build-and-publish:
-    runs-on: ubuntu-20.04
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: [ '3.8.16' ]
+
     permissions:
       id-token: write
+
     steps:
       - uses: actions/checkout@v3
+
+      - name: Set up python
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.python-version }}
+
       - name: Set up PDM
         uses: pdm-project/setup-pdm@v3
+        with:
+          python-version: ${{ matrix.python-version }}
+          version: 2.20.1
 
       - name: Publish package distributions to PyPI
         run: pdm publish

.github/workflows/python-linters.yml

@@ -18,6 +18,7 @@ jobs:
         uses: pdm-project/setup-pdm@v3
         with:
           python-version: ${{ matrix.python-version }}
+          version: 2.20.1
 
       - name: Install dependencies
         run: |

.github/workflows/python-tests.yml

@@ -19,6 +19,7 @@ jobs:
         uses: pdm-project/setup-pdm@v3
         with:
           python-version: ${{ matrix.python-version }}
+          version: 2.20.1
 
       - name: Install dependencies
         run: |

huntflow_api_client/entities/__init__.py

@@ -18,9 +18,11 @@
 from huntflow_api_client.entities.organization_settings import OrganizationSettings
 from huntflow_api_client.entities.production_calendars import ProductionCalendar
 from huntflow_api_client.entities.questionary import ApplicantsQuestionary
+from huntflow_api_client.entities.recommendation import Recommendation
 from huntflow_api_client.entities.regions import Region
 from huntflow_api_client.entities.rejection_reason import RejectionReason
 from huntflow_api_client.entities.resume import Resume
+from huntflow_api_client.entities.survey_type_a import SurveyTypeA
 from huntflow_api_client.entities.survey_type_q import SurveyTypeQ
 from huntflow_api_client.entities.tags import AccountTag, ApplicantTag
 from huntflow_api_client.entities.user_settings import UserSettings
@@ -53,14 +55,16 @@
     "MultiVacancy",
     "OrganizationSettings",
     "ProductionCalendar",
+    "Recommendation",
     "Region",
     "RejectionReason",
     "Resume",
+    "SurveyTypeA",
+    "SurveyTypeQ",
     "User",
     "UsersManagement",
     "UserSettings",
     "Vacancy",
     "VacancyRequest",
     "Webhook",
-    "SurveyTypeQ",
 )

huntflow_api_client/entities/applicants.py

@@ -12,10 +12,12 @@
     ApplicantUpdateRequest,
 )
 from huntflow_api_client.models.response.applicants import (
+    ApplicantCreateAgreementLinkResponse,
     ApplicantCreateResponse,
     ApplicantItem,
     ApplicantListResponse,
     ApplicantSearchByCursorResponse,
+    ApplicantSendAgreementResponse,
 )
 
 
@@ -186,3 +188,41 @@ async def search_by_cursor(
 
         response = await self._api.request("GET", path, params=params)
         return ApplicantSearchByCursorResponse.model_validate(response.json())
+
+    async def create_agreement_link(
+        self,
+        account_id: int,
+        applicant_id: int,
+    ) -> ApplicantCreateAgreementLinkResponse:
+        """
+        API method reference:
+            https://api.huntflow.ai/v2/docs#post-/accounts/-account_id-/applicants/-applicant_id-/agreement_link
+
+        :param account_id: Organization ID
+        :param applicant_id: Applicant ID
+        :return: Link to interact with agreement
+        """
+        response = await self._api.request(
+            "POST",
+            f"/accounts/{account_id}/applicants/{applicant_id}/agreement_link",
+        )
+        return ApplicantCreateAgreementLinkResponse.model_validate(response.json())
+
+    async def send_agreement_via_email(
+        self,
+        account_id: int,
+        applicant_id: int,
+    ) -> ApplicantSendAgreementResponse:
+        """
+        API method reference:
+            https://api.huntflow.ai/v2/docs#post-/accounts/-account_id-/applicants/-applicant_id-/agreement_email
+
+        :param account_id: Organization ID
+        :param applicant_id: Applicant ID
+        :return: Job_id and recipient email of agreement
+        """
+        response = await self._api.request(
+            "POST",
+            f"/accounts/{account_id}/applicants/{applicant_id}/agreement_email",
+        )
+        return ApplicantSendAgreementResponse.model_validate(response.json())

huntflow_api_client/entities/file.py

@@ -1,4 +1,4 @@
-from typing import BinaryIO, Optional, Union
+from typing import BinaryIO, Dict, Optional, Tuple, Union
 
 from huntflow_api_client.entities.base import BaseEntity
 from huntflow_api_client.models.request.file import UploadFileHeaders
@@ -12,6 +12,7 @@ async def upload(
         headers: UploadFileHeaders,
         file: Union[bytes, BinaryIO],
         preset: Optional[str] = None,
+        filename: Optional[str] = None,
     ) -> UploadResponse:
         """
         API method reference https://api.huntflow.ai/v2/docs#post-/accounts/-account_id-/upload
@@ -20,16 +21,24 @@ async def upload(
         :param file: File
         :param preset: Preset
         :param headers: Headers
+        :param filename: Filename
         :return: Additional data
         """
 
         data = {}
         if preset:
             data["preset"] = preset
+
+        files: Dict[str, Union[Union[bytes, BinaryIO], Tuple[str, Union[bytes, BinaryIO]]]] = {}
+        if filename:
+            files["file"] = (filename, file)
+        else:
+            files["file"] = file
+
         response = await self._api.request(
             "POST",
             f"/accounts/{account_id}/upload",
-            files={"file": file},
+            files=files,
             data=data,
             headers=headers.jsonable_dict(exclude_none=True, by_alias=True),
         )

huntflow_api_client/entities/organization_settings.py

@@ -1,4 +1,5 @@
 from huntflow_api_client.entities.base import BaseEntity
+from huntflow_api_client.models.response.interview_types import InterviewTypesListResponse
 from huntflow_api_client.models.response.organization_settings import (
     BaseSurveySchemaTypeWithSchemas,
     CloseReasonsListResponse,
@@ -53,3 +54,15 @@ async def get_applicant_survey_form(
             f"/accounts/{account_id}/surveys/type_a/{survey_id}",
         )
         return BaseSurveySchemaTypeWithSchemas.model_validate(response.json())
+
+    async def get_interview_types(self, account_id: int) -> InterviewTypesListResponse:
+        """
+        API method reference
+           https://api.huntflow.ai/v2/docs#get-/accounts/-account_id-/interview_types
+
+        :param account_id: Organization ID
+
+        :return: List of interview types
+        """
+        response = await self._api.request("GET", f"/accounts/{account_id}/interview_types")
+        return InterviewTypesListResponse.model_validate(response.json())

huntflow_api_client/entities/recommendation.py

@@ -0,0 +1,40 @@
+from typing import Dict, Optional, Union
+
+from huntflow_api_client.entities.base import BaseEntity, ListEntityMixin
+from huntflow_api_client.models.consts import RecommendationProcessingStatus
+from huntflow_api_client.models.response.recommendation import RecommendationListResponse
+
+
+class Recommendation(BaseEntity, ListEntityMixin):
+    async def list(
+        self,
+        account_id: int,
+        vacancy_id: int,
+        count: int = 30,
+        processing_status: RecommendationProcessingStatus = RecommendationProcessingStatus.ALL,
+        next_page_cursor: Optional[str] = None,
+    ) -> RecommendationListResponse:
+        """
+        API method reference
+            https://api.huntflow.ai/v2/docs#get-/accounts/-account_id-/recommendations/-vacancy_id-
+
+        :param account_id: Organization ID
+        :param vacancy_id: Vacancy ID
+        :param count: Number of items per page
+        :param next_page_cursor: Next page cursor
+        :param processing_status: Get all recommendations or processed/unprocessed only
+        :return: A list of applicants recommended for a vacancy
+        """
+        params: Dict[str, Union[str, int]]
+        if next_page_cursor is not None:
+            params = {"next_page_cursor": next_page_cursor}
+        else:
+            params = {"count": count}
+        params["processing_status"] = processing_status.value
+
+        response = await self._api.request(
+            "GET",
+            f"/accounts/{account_id}/recommendations/{vacancy_id}",
+            params=params,
+        )
+        return RecommendationListResponse.model_validate(response.json())

huntflow_api_client/entities/survey_type_a.py

@@ -0,0 +1,65 @@
+from huntflow_api_client.entities.base import BaseEntity, GetEntityMixin, ListEntityMixin
+from huntflow_api_client.models.response.survey import (
+    SurveyAnswerTypeAResponse,
+    SurveySchemasTypeAListResponse,
+    SurveySchemaTypeAResponse,
+)
+
+
+class SurveyTypeA(BaseEntity, GetEntityMixin, ListEntityMixin):
+    async def list(
+        self,
+        account_id: int,
+        active: bool = True,
+    ) -> SurveySchemasTypeAListResponse:
+        """
+        API method reference
+            https://api.huntflow.ai/v2/docs#get-/accounts/-account_id-/surveys/type_a
+
+        :param account_id: Organization ID
+        :param active: Shows only active schemas
+        :return: List of all applicant feedback forms in organization.
+        """
+        params = {"active": active}
+        response = await self._api.request(
+            "GET",
+            f"/accounts/{account_id}/surveys/type_a",
+            params=params,
+        )
+        return SurveySchemasTypeAListResponse.model_validate(response.json())
+
+    async def get(self, account_id: int, survey_id: int) -> SurveySchemaTypeAResponse:
+        """
+        API method reference
+            https://api.huntflow.ai/v2/docs#get-/accounts/-account_id-/surveys/type_a/-survey_id-
+
+        :param account_id: Organization ID
+        :param survey_id: Survey ID
+        :return: An applicant feedback forms in organization.
+        """
+        response = await self._api.request(
+            "GET",
+            f"/accounts/{account_id}/surveys/type_a/{survey_id}",
+        )
+        return SurveySchemaTypeAResponse.model_validate(response.json())
+
+    async def get_applicant_answer(
+        self,
+        account_id: int,
+        survey_id: int,
+        answer_id: int,
+    ) -> SurveyAnswerTypeAResponse:
+        """
+        API method reference
+            https://api.huntflow.ai/v2/docs#get-/accounts/-account_id-/surveys/type_a/-survey_id-/answers/-answer_id-
+
+        :param account_id: Organization ID
+        :param survey_id: Survey ID
+        :param answer_id: Answer ID
+        :return: Returns answer of applicant feedback form.
+        """
+        response = await self._api.request(
+            "GET",
+            f"/accounts/{account_id}/surveys/type_a/{survey_id}/answers/{answer_id}",
+        )
+        return SurveyAnswerTypeAResponse.model_validate(response.json())