np-workflows 1.6.92__py3-none-any.whl → 1.6.94__py3-none-any.whl

np_workflows/experiments/dynamic_routing/widgets.py

@@ -1,7 +1,13 @@
+import shutil
+import time
+
 import IPython.display
 import ipywidgets as ipw
+import np_config
+import np_services
 import np_session
 
+import np_workflows.shared.npxc as npxc
 from np_workflows.shared.base_experiments import DynamicRoutingExperiment
 
 # for widget, before creating a experiment --------------------------------------------- #
@@ -82,6 +88,39 @@ def workflow_select_widget(
 
 
 def photodoc_widget(session: np_session.Session, reminder: str) -> None:
+    vimba_dir = np_config.local_to_unc(
+        session.rig.mon, np_services.config_from_zk()["ImageVimba"]["data"]
+    )
+
+    def get_file_stats():
+        return {p: p.stat().st_mtime for p in vimba_dir.iterdir() if p.is_file()}
+
+    original_file_stats = get_file_stats()
+
     print(
-        f"Take an image in Vimba Viewer and save as:\n{session.npexp_path.name}_{reminder}.png"
+        f"Take an image in Vimba Viewer and save it {vimba_dir} with any name + .png suffix."
+        f"\n\nThis cell will wait for a new file or an existing file to be modified, then copy it as {reminder!r}*"
     )
+    t0 = time.time()
+    timeout_s = 120
+    while True:
+        time.sleep(1)
+        new_files = len(new_file_stats := get_file_stats()) != len(original_file_stats)
+        modified_files = any(
+            p
+            for p in new_file_stats
+            if new_file_stats[p] > original_file_stats.get(p, 0)
+        )
+        if new_files or modified_files:
+            break
+
+        if time.time() - t0 > timeout_s:
+            raise TimeoutError(
+                f"No new image file detected in Vimba folder after {timeout_s} seconds - aborting"
+            )
+    latest_image = max(new_file_stats, key=lambda k: new_file_stats[k])
+    print(f"New file detected:\n\t{latest_image.name}\nCopying to session folder")
+    new_name = f"{session.npexp_path.name}_{reminder}.png"
+    shutil.copy2(latest_image, session.npexp_path / new_name)
+    npxc.validate_or_overwrite(session.npexp_path / new_name, latest_image)
+    print("Done!")

{np_workflows-1.6.92.dist-info → np_workflows-1.6.94.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: np-workflows
-Version: 1.6.92
+Version: 1.6.94
 Summary: Ecephys and behavior workflows for the Mindscope Neuropixels team.
 Author-email: Ben Hardcastle <ben.hardcastle@alleninstitute.org>
 License: MIT

{np_workflows-1.6.92.dist-info → np_workflows-1.6.94.dist-info}/RECORD

@@ -2,7 +2,7 @@ np_workflows/__init__.py,sha256=5ZC3gO-VV7PfP13WenHePtvU2wlP68ar6j47pJVvUCM,231
 np_workflows/experiments/__init__.py,sha256=voWNXF2Kl3mmB4PwqnZ4DcD5zhvBq1j2xLUHnLe6So8,39
 np_workflows/experiments/dynamic_routing/__init__.py,sha256=M2oiBrVpy_Wrt7oC8J1wzHvKTYdA2K1UIb0-_7S6SuQ,123
 np_workflows/experiments/dynamic_routing/main.py,sha256=BHleH7NmomD_tMhVPVIpbebetABooOoVClFkggLQS10,2617
-np_workflows/experiments/dynamic_routing/widgets.py,sha256=kRtpnyKYyKcjM_4oNM2YeCLDNaxEwOUopyDcdK0gNj0,2815
+np_workflows/experiments/dynamic_routing/widgets.py,sha256=q9X0hitKO63wZhv-ryB97xuluU-EWyt2frfC5qi1-oU,4246
 np_workflows/experiments/openscope_P3/P3_workflow_widget.py,sha256=MmGgo-ir4Xy-PDjY3YRhxy5lATAeQ3F9bsuGtF6Bu5I,2361
 np_workflows/experiments/openscope_P3/__init__.py,sha256=g1SzGmPdWB4MPzx7Zaye4YQU89BDocywCfcAs2ITC5A,133
 np_workflows/experiments/openscope_P3/main_P3_pilot.py,sha256=39vcLIEVPoB85TdPQ-QlF0sT7Ej9R7oIQHpGJ-dt1Q0,7007
@@ -41,8 +41,8 @@ np_workflows/shared/__init__.py,sha256=YErshr5fY7jU8TJgWqEHz07tUHnEJEZ_mY31DIpQ5
 np_workflows/shared/base_experiments.py,sha256=KTxsltWtvNlHmlHJkIbMj7qon-RULAlb_Q224c2dtI0,35395
 np_workflows/shared/npxc.py,sha256=R6n6xrLGr31epoDX_99D_Wy7BYgmjjiZ6efl3UOoK3Y,6642
 np_workflows/shared/widgets.py,sha256=xF4PKZugYfyHtCPyYxwYgeH5jpB8vyoGDgenEodTTzo,27799
-np_workflows-1.6.92.dist-info/METADATA,sha256=Xb9Y8sY16RTF2eeDSrd1yNjRTX_vPnPqZlczpCMBzvk,3399
-np_workflows-1.6.92.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-np_workflows-1.6.92.dist-info/entry_points.txt,sha256=XBvlXc6-LK4_pCjspQZF_yBbopODJIquhZ8J4LhxlaI,41
-np_workflows-1.6.92.dist-info/top_level.txt,sha256=lndEdwvoxg_3fCYd2tyIUfcdvCFegVm3lWUdEf85u8M,13
-np_workflows-1.6.92.dist-info/RECORD,,
+np_workflows-1.6.94.dist-info/METADATA,sha256=VGccVYWAt-BY4bB4h26uMtvn8P5D8gqhJ3ALa8aFEXI,3399
+np_workflows-1.6.94.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+np_workflows-1.6.94.dist-info/entry_points.txt,sha256=XBvlXc6-LK4_pCjspQZF_yBbopODJIquhZ8J4LhxlaI,41
+np_workflows-1.6.94.dist-info/top_level.txt,sha256=lndEdwvoxg_3fCYd2tyIUfcdvCFegVm3lWUdEf85u8M,13
+np_workflows-1.6.94.dist-info/RECORD,,