mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-23 20:01:53 +00:00
chore: extract build_access_denied_message in its own class and add
pytest_asyncio where needed Signed-off-by: Akram Ben Aissi <<akram.benaissi@gmail.com>>
This commit is contained in:
parent
b945525a9e
commit
5d179f532b
5 changed files with 21 additions and 19 deletions
|
|
@ -105,7 +105,17 @@ def is_action_allowed(
|
|||
return False
|
||||
|
||||
|
||||
def build_access_denied_message(action: str | None, resource: ProtectedResource | None, user: User | None) -> str:
|
||||
class AccessDeniedError(RuntimeError):
|
||||
def __init__(self, action: str | None = None, resource: ProtectedResource | None = None, user: User | None = None):
|
||||
self.action = action
|
||||
self.resource = resource
|
||||
self.user = user
|
||||
|
||||
message = _build_access_denied_message(action, resource, user)
|
||||
super().__init__(message)
|
||||
|
||||
|
||||
def _build_access_denied_message(action: str | None, resource: ProtectedResource | None, user: User | None) -> str:
|
||||
"""Build detailed error message for access denied scenarios."""
|
||||
if action and resource and user:
|
||||
resource_info = f"{resource.type}::{resource.identifier}"
|
||||
|
|
@ -119,13 +129,3 @@ def build_access_denied_message(action: str | None, resource: ProtectedResource
|
|||
message = "Insufficient permissions"
|
||||
|
||||
return message
|
||||
|
||||
|
||||
class AccessDeniedError(RuntimeError):
|
||||
def __init__(self, action: str | None = None, resource: ProtectedResource | None = None, user: User | None = None):
|
||||
self.action = action
|
||||
self.resource = resource
|
||||
self.user = user
|
||||
|
||||
message = build_access_denied_message(action, resource, user)
|
||||
super().__init__(message)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue