Vendor team member management features

app/exceptions/__init__.py

@@ -146,6 +146,7 @@ from .team import (
     MaxTeamMembersReachedException,
     TeamValidationException,
     InvalidInvitationDataException,
+    InvalidInvitationTokenException,
 )
 
 # Product exceptions
@@ -215,6 +216,7 @@ __all__ = [
     "MaxTeamMembersReachedException",
     "TeamValidationException",
     "InvalidInvitationDataException",
+    "InvalidInvitationTokenException",
 
     # Inventory exceptions
     "InventoryNotFoundException",
@@ -306,4 +308,4 @@ __all__ = [
     "InvalidAdminActionException",
     "BulkOperationException",
     "ConfirmationRequiredException",
-]
+]

app/exceptions/team.py

@@ -147,10 +147,10 @@ class InvalidRoleException(ValidationException):
     """Raised when role data is invalid."""
 
     def __init__(
-        self,
-        message: str = "Invalid role data",
-        field: Optional[str] = None,
-        details: Optional[Dict[str, Any]] = None,
+            self,
+            message: str = "Invalid role data",
+            field: Optional[str] = None,
+            details: Optional[Dict[str, Any]] = None,
     ):
         super().__init__(
             message=message,
@@ -164,10 +164,10 @@ class InsufficientPermissionsException(AuthorizationException):
     """Raised when user lacks required permissions for an action."""
 
     def __init__(
-        self,
-        required_permission: str,
-        user_id: Optional[int] = None,
-        action: Optional[str] = None,
+            self,
+            required_permission: str,
+            user_id: Optional[int] = None,
+            action: Optional[str] = None,
     ):
         details = {"required_permission": required_permission}
         if user_id:
@@ -202,10 +202,10 @@ class TeamValidationException(ValidationException):
     """Raised when team operation validation fails."""
 
     def __init__(
-        self,
-        message: str = "Team operation validation failed",
-        field: Optional[str] = None,
-        validation_errors: Optional[Dict[str, str]] = None,
+            self,
+            message: str = "Team operation validation failed",
+            field: Optional[str] = None,
+            validation_errors: Optional[Dict[str, str]] = None,
     ):
         details = {}
         if validation_errors:
@@ -223,10 +223,10 @@ class InvalidInvitationDataException(ValidationException):
     """Raised when team invitation data is invalid."""
 
     def __init__(
-        self,
-        message: str = "Invalid invitation data",
-        field: Optional[str] = None,
-        details: Optional[Dict[str, Any]] = None,
+            self,
+            message: str = "Invalid invitation data",
+            field: Optional[str] = None,
+            details: Optional[Dict[str, Any]] = None,
     ):
         super().__init__(
             message=message,
@@ -234,3 +234,31 @@ class InvalidInvitationDataException(ValidationException):
             details=details,
         )
         self.error_code = "INVALID_INVITATION_DATA"
+
+
+# ============================================================================
+# NEW: Add InvalidInvitationTokenException
+# ============================================================================
+
+class InvalidInvitationTokenException(ValidationException):
+    """Raised when invitation token is invalid, expired, or already used.
+
+    This is a general exception for any invitation token validation failure.
+    Use this when checking invitation tokens during the acceptance flow.
+    """
+
+    def __init__(
+            self,
+            message: str = "Invalid or expired invitation token",
+            invitation_token: Optional[str] = None
+    ):
+        details = {}
+        if invitation_token:
+            details["invitation_token"] = invitation_token
+
+        super().__init__(
+            message=message,
+            field="invitation_token",
+            details=details,
+        )
+        self.error_code = "INVALID_INVITATION_TOKEN"