np-workflows 1.6.93__tar.gz → 1.6.95__tar.gz

{np_workflows-1.6.93 → np_workflows-1.6.95}/PKG-INFO

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

{np_workflows-1.6.93 → np_workflows-1.6.95}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "np-workflows"
-version = "1.6.93"
+version = "1.6.95"
 description = "Ecephys and behavior workflows for the Mindscope Neuropixels team."
 authors = [
     { name = "Ben Hardcastle", email = "ben.hardcastle@alleninstitute.org" },

{np_workflows-1.6.93 → np_workflows-1.6.95}/src/np_workflows/experiments/dynamic_routing/widgets.py

@@ -91,24 +91,39 @@ 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"]
     )
-    n_files = len(tuple(vimba_dir.iterdir()))
-    t0 = time.time()
+
+    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()
+
+    timeout_s = 300
     print(
-        f"Take an image in Vimba Viewer and save it with the default name (eg should be unique)."
-        f"\n\n*This cell will wait for a new file to be created in {vimba_dir}, then copy it to the session folder with suffix {reminder!r}*"
+        f"Take an image in Vimba Viewer and save it {vimba_dir} with any name and .png suffix."
+        f"\n\nThis cell will wait for a new file or an existing file to be modified ({timeout_s = })*"
     )
-    timeout_s = 120
-    while len(files := tuple(vimba_dir.iterdir())) == n_files:
+    t0 = time.time()
+    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(
-        (p for p in files if p.is_file()), key=lambda f: f.stat().st_ctime
+    latest_image = max(new_file_stats, key=lambda k: new_file_stats[k])
+    dest = (
+        session.npexp_path
+        / f"{session.npexp_path.name}_{reminder}{latest_image.suffix}"
     )
-    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(f"New file detected:\n\t{latest_image.name}\nCopying to:\n\t{dest}")
+    shutil.copy2(latest_image, dest)
+    npxc.validate_or_overwrite(dest, latest_image)
     print("Done!")

{np_workflows-1.6.93 → np_workflows-1.6.95}/src/np_workflows.egg-info/PKG-INFO

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