refactor: Move EntityType definition to types

Author: bencap

src/mavedb/lib/permissions/core.py

@@ -1,10 +1,11 @@
-from typing import Any, Callable, Optional, Union
+from typing import Any, Callable, Optional
 
 from mavedb.lib.authentication import UserData
 from mavedb.lib.logging.context import save_to_logging_context
 from mavedb.lib.permissions.actions import Action
 from mavedb.lib.permissions.exceptions import PermissionException
 from mavedb.lib.permissions.models import PermissionResponse
+from mavedb.lib.types.permissions import EntityType
 from mavedb.models.collection import Collection
 from mavedb.models.experiment import Experiment
 from mavedb.models.experiment_set import ExperimentSet
@@ -22,16 +23,6 @@
     user,
 )
 
-# Define the supported entity types
-EntityType = Union[
-    Collection,
-    Experiment,
-    ExperimentSet,
-    ScoreCalibration,
-    ScoreSet,
-    User,
-]
-
 
 def has_permission(user_data: Optional[UserData], entity: EntityType, action: Action) -> PermissionResponse:
     """

src/mavedb/lib/types/permissions.py

@@ -0,0 +1,18 @@
+from typing import Union
+
+from mavedb.models.collection import Collection
+from mavedb.models.experiment import Experiment
+from mavedb.models.experiment_set import ExperimentSet
+from mavedb.models.score_calibration import ScoreCalibration
+from mavedb.models.score_set import ScoreSet
+from mavedb.models.user import User
+
+# Define the supported entity types
+EntityType = Union[
+    Collection,
+    Experiment,
+    ExperimentSet,
+    ScoreCalibration,
+    ScoreSet,
+    User,
+]