feat: added qualifications check in POST /assignment endpoint

Author: Dzeranov

packages/examples/cvat/exchange-oracle/src/endpoints/authentication.py

@@ -37,6 +37,10 @@ class AuthorizationData(BaseModel):
     email: str
 
 
+class AssignmentAuthorizationData(AuthorizationData):
+    qualifications: list[str]
+
+
 AuthDataT = TypeVar("AuthDataT", bound=AuthorizationData)
 
 

packages/examples/cvat/exchange-oracle/src/endpoints/exchange.py

@@ -18,6 +18,7 @@
 from src.db import SessionLocal
 from src.db import engine as db_engine
 from src.endpoints.authentication import (
+    AssignmentAuthorizationData,
     AuthorizationData,
     AuthorizationParam,
     JobListAuthorizationData,
@@ -395,13 +396,17 @@ def _page_serializer(
     description="Start an assignment within the task for the annotator",
 )
 async def create_assignment(
-    data: AssignmentRequest, token: Annotated[AuthorizationData, AuthorizationParam]
+    data: AssignmentRequest,
+    token: Annotated[
+        AssignmentAuthorizationData, make_auth_dependency(AssignmentAuthorizationData)
+    ],
 ) -> AssignmentResponse:
     try:
         assignment_id = oracle_service.create_assignment(
             escrow_address=data.escrow_address,
             chain_id=data.chain_id,
             wallet_address=token.wallet_address,
+            qualifications=token.qualifications,
         )
     except oracle_service.UserHasUnfinishedAssignmentError as e:
         raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(e)) from e

packages/examples/cvat/exchange-oracle/src/services/cvat.py

@@ -973,7 +973,7 @@ def get_user_assignments_in_cvat_projects(
 
 def has_active_user_assignments(
     session: Session,
-    wallet_address: int,
+    wallet_address: str,
     escrow_address: str,
     chain_id: int,
 ) -> bool:

packages/examples/cvat/exchange-oracle/src/services/exchange.py

@@ -3,11 +3,12 @@
 
 import src.cvat.api_calls as cvat_api
 import src.services.cvat as cvat_service
+from src.chain.escrow import get_escrow_manifest
 from src.core.types import JobStatuses, Networks, ProjectStatuses, TaskTypes
 from src.db import SessionLocal
 from src.db.utils import ForUpdateParams
 from src.models.cvat import Job
-from src.utils.assignments import get_default_assignment_timeout
+from src.utils.assignments import get_default_assignment_timeout, parse_manifest
 from src.utils.requests import get_or_404
 from src.utils.time import utcnow
 
@@ -20,14 +21,26 @@ def __str__(self) -> str:
         )
 
 
-def create_assignment(escrow_address: str, chain_id: Networks, wallet_address: str) -> str | None:
+class UserQualificationError(Exception):
+    def __str__(self) -> str:
+        return "User doesn't have required qualifications."
+
+
+def create_assignment(
+    escrow_address: str, chain_id: Networks, wallet_address: str, qualifications: list[str]
+) -> str | None:
     with SessionLocal.begin() as session:
         user = get_or_404(
             cvat_service.get_user_by_id(session, wallet_address, for_update=True),
             wallet_address,
             object_type_name="user",
         )
 
+        manifest = parse_manifest(get_escrow_manifest(chain_id, escrow_address))
+
+        if not all(q in qualifications for q in manifest.qualifications):
+            raise UserQualificationError
+
         if cvat_service.has_active_user_assignments(
             session,
             wallet_address=wallet_address,