Serialize dataset type for archiving

autobidsportal/datalad.py

@@ -9,12 +9,7 @@
 from datalad.support.annexrepo import AnnexRepo
 from flask import current_app
 
-from autobidsportal.models import (
-    DataladDataset,
-    DatasetType,
-    Study,
-    db,
-)
+from autobidsportal.models import DataladDataset, DatasetType, Study, db
 
 
 class RiaDataset:
@@ -46,29 +41,17 @@ def __exit__(self, exc_type, exc_value, traceback):
         remove_finalized_dataset(self.path_dataset)
 
 
-def get_alias(study_id, dataset_type):
+def get_alias(study_id: int, dataset_type: DatasetType):
     """Generate an alias for a study in the RIA.
 
     Parameters
     ----------
-    study_id : int
+    study_id
         ID of the study.
-    dataset_type: DatasetType
+    dataset_type
         "tar2bids" or "cfmm2tar".
-
-    Raises
-    ------
-    TypeError
-        If dataset_type is not DatasetType
     """
-    if dataset_type is DatasetType.SOURCE_DATA:
-        text = "sourcedata"
-    elif dataset_type is DatasetType.RAW_DATA:
-        text = "rawdata"
-    elif dataset_type is DatasetType.DERIVED_DATA:
-        text = "deriveddata"
-    else:
-        raise TypeError
+    text = dataset_type.to_bids_str()
     return f"study-{study_id}_{text}"
 
 

autobidsportal/models.py

@@ -381,6 +381,15 @@ class DatasetType(Enum):
     RAW_DATA = 2
     DERIVED_DATA = 3
 
+    def to_bids_str(self) -> str:
+        """Produce a BIDS-style string for serialization."""
+        map_ = {
+            DatasetType.SOURCE_DATA: "sourcedata",
+            DatasetType.RAW_DATA: "rawdata",
+            DatasetType.DERIVED_DATA: "deriveddata",
+        }
+        return map_[self]
+
 
 class DataladDataset(db.Model):
     """A datalad dataset relating to a study."""

autobidsportal/routes.py

@@ -1005,6 +1005,7 @@ def list_globus_users():
         [
             {
                 "id": dataset.study_id,
+                "type": dataset.dataset_type.to_bids_str(),
                 "path": f"{path_base}/{dataset.ria_alias}",
                 "users": [
                     username.username