From 5f2bf1734c946708290cd8978ffa7a264a6acae9 Mon Sep 17 00:00:00 2001 From: Joe Banks Date: Wed, 22 May 2024 00:41:34 +0100 Subject: Add get_active_nomination API method for Talentpool --- bot/exts/recruitment/talentpool/_api.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/bot/exts/recruitment/talentpool/_api.py b/bot/exts/recruitment/talentpool/_api.py index 3c0092617..476ba6b80 100644 --- a/bot/exts/recruitment/talentpool/_api.py +++ b/bot/exts/recruitment/talentpool/_api.py @@ -62,6 +62,15 @@ class NominationAPI: nomination = Nomination.model_validate(data) return nomination + async def get_active_nomination(self, user_id: int) -> Nomination | None: + """Search for an active nomination for a user and return it.""" + nominations = await self.get_nominations(user_id=user_id, active=True) + + if len(nominations) >= 1: + return nominations[0] + + return None + async def get_nomination_reason(self, user_id: int, actor_id: int) -> tuple[Nomination, str] | None: """Search for a nomination & reason for a specific actor on a specific user.""" nominations = await self.get_nominations(user_id, True) -- cgit v1.2.3