cloud-api/src/iam/exceptions.py

"""
Exceptions related to the IAM module

Exceptions:
 - GroupNotFoundException: Takes an optional group_id int
 - PermNotFoundException: Takes an optional perm_id int
"""
from typing import Optional

from fastapi import HTTPException, status


class GroupNotFoundException(HTTPException):
	def __init__(self, group_id: Optional[int] = None) -> None:
		detail = "Group not found" if group_id is None else f"User with ID '{group_id}' was not found."
		super().__init__(
			status_code=status.HTTP_404_NOT_FOUND,
			detail=detail,
		)


class PermNotFoundException(HTTPException):
	def __init__(self, perm_id: Optional[int] = None) -> None:
		detail = "Permission not found" if perm_id is None else f"User with ID '{perm_id}' was not found."
		super().__init__(
			status_code=status.HTTP_404_NOT_FOUND,
			detail=detail,
		)