fix: correctly pull acquisition start_time and type fields

AllenNeuralDynamics · Jan 14, 2025 · 3a3a9ba · 3a3a9ba
1 parent c7963b4
commit 3a3a9ba
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/src/aind_qc_portal/docdb/database.py b/src/aind_qc_portal/docdb/database.py
@@ -166,6 +166,8 @@ def get_project(project_name: str):
             "subject.genotype": 1,
             "session.session_type": 1,
             "session.session_start_time": 1,
+            "acquisition.session_type": 1,
+            "acquisition.session_start_time": 1,
             "data_description.data_level": 1,
             "data_description.project_name": 1,
             "rig.rig_id": 1,

diff --git a/src/aind_qc_portal/projects/dataset.py b/src/aind_qc_portal/projects/dataset.py
@@ -44,14 +44,28 @@ def _get_assets(self):
             if operator_list:
                 operator_list = list(operator_list)
 
+            if record.get('session', {}):
+                start_time = record.get('session', {}).get('session_start_time')
+            elif record.get('acquisition', {}):
+                start_time = record.get('acquisition', {}).get('session_start_time')
+            else:
+                start_time = None
+
+            if record.get('session', {}):
+                session_type = record.get('session', {}).get('session_type')
+            elif record.get('acquisition', {}):
+                session_type = record.get('acquisition', {}).get('session_type')
+            else:
+                session_type = None
+
             record_data = {
                 '_id': record.get('_id'),
                 'raw': record.get('data_description', {}).get('data_level') == 'raw',
                 'project_name': record.get('data_description', {}).get('project_name'),
                 'location': record.get('location'),
                 'name': record.get('name'),
-                'session_start_time': record.get('session', {}).get('session_start_time'),
-                'session_type': record.get('session', {}).get('session_type'),
+                'session_start_time': start_time,
+                'session_type': session_type,
                 'subject_id': subject_id,
                 'operator': operator_list,
                 'Status': qc.status().value if qc else "No QC",
@@ -74,6 +88,8 @@ def _get_assets(self):
         self._df.sort_values(by="timestamp", ascending=True, inplace=True)
         self._df.sort_values(by="subject_id", ascending=False, inplace=True)
 
+        self._df.to_csv('data.csv')
+
     @property
     def _data_filtered(self) -> pd.DataFrame:
         """Internal access method to get the full filtered dataframe