Skip to content

Commit ab00c49

Browse files
committed
Require aiod_entry instead of concept
1 parent 05e8855 commit ab00c49

File tree

3 files changed

+7
-8
lines changed

3 files changed

+7
-8
lines changed

src/database/authorization.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -82,17 +82,14 @@ def register_user(kc_user: KeycloakUser, session: Session) -> User:
8282

8383

8484
def set_permission(
85-
user: KeycloakUser, resource: AIoDConcept, session: Session, *, type_: PermissionType
85+
user: KeycloakUser, resource: AIoDEntryORM, session: Session, *, type_: PermissionType
8686
):
8787
key = {
8888
"user_identifier": user._subject_identifier,
89-
"aiod_entry_identifier": resource.aiod_entry_identifier,
89+
"aiod_entry_identifier": resource.identifier,
9090
}
9191
permission = session.get(Permission, key)
9292
if permission is None:
93-
permission = Permission(
94-
user_identifier=user._subject_identifier,
95-
aiod_entry=resource.aiod_entry,
96-
)
93+
permission = Permission(user_identifier=user._subject_identifier, aiod_entry=resource)
9794
permission.type_ = type_
9895
session.add(permission)

src/routers/resource_router.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -418,7 +418,9 @@ def register_resource(
418418
try:
419419
resource = self.create_resource(session, resource_create)
420420
register_user(user, session)
421-
set_permission(user, resource, session, type_=PermissionType.ADMIN)
421+
set_permission(
422+
user, resource.aiod_entry, session, type_=PermissionType.ADMIN
423+
)
422424
session.commit()
423425
return self._wrap_with_headers({"identifier": resource.identifier})
424426
except Exception as e:

src/tests/authorization/test_authorization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -325,7 +325,7 @@ def register_asset(asset: AIoDConcept, /, *, owner: KeycloakUser, status: EntryS
325325
session.commit()
326326

327327
register_user(owner, session)
328-
set_permission(owner, asset, session, type_=PermissionType.ADMIN)
328+
set_permission(owner, asset.aiod_entry, session, type_=PermissionType.ADMIN)
329329

330330
asset.aiod_entry.status = status
331331
if status in [EntryStatus.SUBMITTED, EntryStatus.PUBLISHED, EntryStatus.REJECTED]:

0 commit comments

Comments
 (0)