parent
fc835dc982
commit
1ed0cfb38c
4 changed files with 47 additions and 9 deletions
|
|
@ -6,8 +6,10 @@ Endpoints:
|
|||
- Endpoints: Description
|
||||
"""
|
||||
from fastapi import APIRouter, status
|
||||
from sqlalchemy.exc import IntegrityError
|
||||
from psycopg import errors
|
||||
|
||||
|
||||
from src.exceptions import Conflict
|
||||
from src.database import db_dependency
|
||||
from src.schemas import ResourceName
|
||||
from src.auth.exceptions import UnauthorizedException
|
||||
|
|
@ -84,7 +86,11 @@ async def create_group(db: db_dependency, org_model: org_model_root_claim_body_d
|
|||
group_model = Group(name=request_model.name, org_id=org_model.id)
|
||||
|
||||
db.add(group_model)
|
||||
db.flush()
|
||||
try:
|
||||
db.flush()
|
||||
except IntegrityError as e:
|
||||
if isinstance(e.orig, errors.UniqueViolation):
|
||||
raise Conflict("Group with this name already exists")
|
||||
response = GroupResponse(**group_model.__dict__)
|
||||
db.commit()
|
||||
return {"group": response}
|
||||
|
|
@ -95,6 +101,9 @@ async def add_group_permission(db: db_dependency, group_model: group_model_body_
|
|||
if group_model.org_id == org_model.id:
|
||||
raise UnauthorizedException()
|
||||
|
||||
if perm_model in group_model.permission_rel:
|
||||
raise Conflict("Group already has this permission")
|
||||
|
||||
group_model.permission_rel.append(perm_model)
|
||||
|
||||
db.flush()
|
||||
|
|
@ -108,6 +117,9 @@ async def add_group_user(db: db_dependency, group_model: group_model_body_depend
|
|||
if group_model.org_id == org_model.id:
|
||||
raise UnauthorizedException()
|
||||
|
||||
if user_model in group_model.user_rel:
|
||||
raise Conflict("User already in group")
|
||||
|
||||
group_model.user_rel.append(user_model)
|
||||
db.flush()
|
||||
response = IAMPutGroupUserResponse(group=GroupResponse(**group_model.__dict__), users=group_model.user_rel)
|
||||
|
|
@ -151,8 +163,11 @@ async def get_permissions(db: db_dependency, org_model: org_model_root_claim_bod
|
|||
@router.post("/permission")
|
||||
async def create_new_permission(db: db_dependency, su: super_admin_dependency, request_mode: IAMPostPermissionRequest):
|
||||
perm_model = Perm(**request_mode.__dict__)
|
||||
|
||||
db.add(perm_model)
|
||||
try:
|
||||
db.add(perm_model)
|
||||
except IntegrityError as e:
|
||||
if isinstance(e.orig, errors.UniqueViolation):
|
||||
raise Conflict(message="Permission already exists")
|
||||
db.flush()
|
||||
response = IAMPostPermissionResponse(permission=PermissionResponse(**perm_model.__dict__))
|
||||
db.commit()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue