Skip to content

Commit 56671f0

Browse files
committed
Add collection permissions to permissions API
1 parent 506ab14 commit 56671f0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/mavedb/routers/permissions.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from mavedb.lib.logging import LoggedRoute
1111
from mavedb.lib.logging.context import logging_context, save_to_logging_context
1212
from mavedb.lib.permissions import Action, has_permission
13+
from mavedb.models.collection import Collection
1314
from mavedb.models.experiment import Experiment
1415
from mavedb.models.experiment_set import ExperimentSet
1516
from mavedb.models.score_set import ScoreSet
@@ -25,6 +26,7 @@
2526

2627

2728
class ModelName(str, Enum):
29+
collection = "collection"
2830
experiment = "experiment"
2931
experiment_set = "experiment-set"
3032
score_set = "score-set"
@@ -52,6 +54,8 @@ async def check_permission(
5254
item = db.query(Experiment).filter(Experiment.urn == urn).one_or_none()
5355
elif model_name == ModelName.score_set:
5456
item = db.query(ScoreSet).filter(ScoreSet.urn == urn).one_or_none()
57+
elif model_name == ModelName.collection:
58+
item = db.query(Collection).filter(Collection.urn == urn).one_or_none()
5559

5660
if item:
5761
permission = has_permission(user_data, item, action).permitted

0 commit comments

Comments
 (0)