talentro-commons 0.19.4__tar.gz → 0.19.5__tar.gz

{talentro_commons-0.19.4 → talentro_commons-0.19.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: talentro-commons
-Version: 0.19.4
+Version: 0.19.5
 Summary: This package contains all globally used code, services, models and data structures for Talentro
 License: Proprietary
 Author: Emiel van Essen

{talentro_commons-0.19.4 → talentro_commons-0.19.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "talentro-commons"
-version = "0.19.4"
+version = "0.19.5"
 description = "This package contains all globally used code, services, models and data structures for Talentro"
 authors = ["Emiel van Essen <emiel@marksmen.nl>"]
 license = "Proprietary"

{talentro_commons-0.19.4 → talentro_commons-0.19.5}/src/talentro/acquisition/dataclasses.py

@@ -120,7 +120,7 @@ class AdInfo(ResolvableCompanyModel):
     campaign: CampaignInfo
 
     @staticmethod
-    async def resolve_object(object_id: UUID, organization_id: str) -> "AdInfo | None":
+    async def resolve_object(object_id: UUID, organization_id: UUID) -> "AdInfo | None":
         cache = CacheService()
 
         cache_key = f"resolver:org:{organization_id}:ad_id:{object_id}"

{talentro_commons-0.19.4 → talentro_commons-0.19.5}/src/talentro/candidates/dataclasses.py

@@ -3,9 +3,11 @@ from typing import Optional
 from uuid import UUID
 
 from pydantic import BaseModel
-from talentro.acquisition.dataclasses import CampaignInfo
+from talentro.acquisition.dataclasses import CampaignInfo, AdInfo
 from talentro.candidates.models import Application as ApplicationModel, Document as DocumentModel, \
-    Candidate as CandidateModel
+    Candidate as CandidateModel, Source as SourceModel
+from talentro.services.caching import CacheService
+from talentro.services.clients import MSClient
 from talentro.vacancies.dataclasses import VacancyInfo
 
 
@@ -75,6 +77,56 @@ class DocumentInfo(BaseModel):
         )
 
 
+class SourceInfo(BaseModel):
+    id: UUID
+    created_at: datetime
+    updated_at: Optional[datetime]
+    organization: UUID
+
+    campaign: Optional[CampaignInfo]
+    ad: Optional[AdInfo]
+
+    application_count: Optional[int]
+
+    @classmethod
+    async def from_model(cls: "SourceInfo", model: SourceModel, application_count: int) -> 'SourceInfo':
+        if model.campaign_id:
+            campaign = await CampaignInfo.resolve_object(model.campaign_id, model.organization)
+        else:
+            campaign = None
+
+        if model.ad_id:
+            ad = await AdInfo.resolve_object(model.ad_id, model.organization)
+        else:
+            ad = None
+
+        return SourceInfo(
+            id=model.id,
+            created_at=model.created_at,
+            updated_at=model.updated_at,
+            organization=model.organization,
+            campaign=campaign,
+            ad=ad,
+            application_count=application_count,
+        )
+
+    @staticmethod
+    async def resolve_object(object_id: UUID, organization_id: UUID) -> "SourceInfo | None":
+        cache = CacheService()
+
+        cache_key = f"resolver:org:{organization_id}:source_id:{object_id}"
+
+        if value := await cache.get(cache_key):
+            return value
+
+        result = await MSClient.candidates().get(f"sources/{object_id}/resolve", headers={"X-Organization-ID": str(organization_id)})
+
+        if result.status_code != 200:
+            return None
+
+        return SourceInfo(**result.json())
+
+
 class ApplicationInfo(BaseModel):
     id: UUID
     created_at: datetime
@@ -84,8 +136,8 @@ class ApplicationInfo(BaseModel):
     status: str
     screening_answers: dict
 
+    source: Optional[SourceInfo]
     vacancy: Optional[VacancyInfo]
-    campaign: Optional[CampaignInfo]
     candidate: Optional[CandidateInfo]
 
     documents: list[DocumentInfo]
@@ -93,9 +145,9 @@ class ApplicationInfo(BaseModel):
     @classmethod
     async def from_model(cls: "ApplicationInfo", model: ApplicationModel) -> 'ApplicationInfo':
         if model.source:
-            campaign = await CampaignInfo.resolve_object(model.source.campaign_id, model.organization)
+            source = await SourceInfo.resolve_object(model.source.id, model.organization)
         else:
-            campaign = None
+            source = None
 
         if model.vacancy_id:
             vacancy = await VacancyInfo.resolve_object(model.vacancy_id, model.organization)
@@ -116,8 +168,8 @@ class ApplicationInfo(BaseModel):
             screening_answers=model.screening_answers,
             status=model.status,
 
+            source=source,
             vacancy=vacancy,
-            campaign=campaign,
             candidate=candidate,
 
             documents=[await DocumentInfo.from_model(document) for document in model.documents],

{talentro_commons-0.19.4 → talentro_commons-0.19.5}/src/talentro/candidates/models.py

@@ -57,7 +57,7 @@ class Application(CandidatesOrganizationModel, table=True):
 
 class Source(CandidatesOrganizationModel, table=True):
     source: str = Field(index=True)
-    event_id: Optional[str] = Field(index=True)
     campaign_id: Optional[UUID] = Field(index=True)
+    ad_id: Optional[UUID] = Field(index=True, nullable=True)
 
     applications: list["Application"] = Relationship(back_populates="source")