let --recordings accept paths to the recordings too

LAAC-LSCP · Sep 8, 2021 · a362d8d · a362d8d
1 parent 5b7bfb9
commit a362d8d
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/ChildProject/projects.py b/ChildProject/projects.py
@@ -485,6 +485,14 @@ def get_converted_recording_filename(
     def get_recordings_from_list(self, recordings: list):
         _recordings = self.recordings.copy()
 
+        # if the user provided paths,
+        # transform those paths into recording_filename values
+        if all(map(os.path.exists, recordings)):
+            recordings = [
+                os.path.relpath(recording, os.path.join(self.path, self.RAW_RECORDINGS))
+                for recording in recordings
+            ]
+
         if recordings is not None:
             _recordings = _recordings[
                 _recordings["recording_filename"].isin(recordings)

diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py
@@ -45,3 +45,13 @@ def test_whitelist():
         caught_value_error = True
 
     assert caught_value_error == True
+
+    recordings = project.get_recordings_from_list(
+        Pipeline.recordings_from_list(
+            [
+                "examples/valid_raw_data/recordings/raw/sound.wav",
+                "examples/valid_raw_data/recordings/raw/sound2.wav",
+            ]
+        )
+    )
+    assert recordings["recording_filename"].tolist() == ["sound.wav", "sound2.wav"]