Add globus storage search_for

machinable-org · Mar 19, 2024 · 4932e3b · 4932e3b
1 parent e257d5a
commit 4932e3b
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/docs/examples/globus-storage/globus.py b/docs/examples/globus-storage/globus.py
@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, List
 
 import os
 import time
@@ -211,3 +211,16 @@ def local_path(self, *append):
 
     def remote_path(self, *append):
         return os.path.join(self.config.remote_endpoint_directory, *append)
+
+    def search_for(self, experiment) -> List[str]:
+        response = self.transfer_client.operation_ls(
+            self.config.remote_endpoint_id,
+            path=self.remote_path(),
+            show_hidden=True,
+        )
+        found = []
+        for item in response:
+            if experiment.hash in item["name"]:
+                found.append(item["name"])
+
+        return found