diff --git a/repositories/hubspot_deals/deal_database_postgres_gateway.py b/repositories/hubspot_deals/deal_database_postgres_gateway.py index 29c1868ec..bfd917781 100644 --- a/repositories/hubspot_deals/deal_database_postgres_gateway.py +++ b/repositories/hubspot_deals/deal_database_postgres_gateway.py @@ -26,7 +26,8 @@ class DealDatabasePostgresGateway: self._session.commit() def job_no_for_deal(self, deal_id: str) -> Optional[str]: - raise NotImplementedError + deal = self._find_deal(deal_id) + return deal.job_no if deal is not None else None def _find_deal(self, deal_id: str) -> Optional[HubspotDealData]: statement = select(HubspotDealData).where(HubspotDealData.deal_id == deal_id)