np_codeocean 0.3.5__py3-none-any.whl → 0.3.6__py3-none-any.whl

np_codeocean/metadata/np.py

@@ -87,4 +87,4 @@ def get_current_rig_model(
         storage_directory,
         datetime.datetime.now(),
         rig_name,
-    )
+    )

np_codeocean/metadata/rigs.py

@@ -106,4 +106,4 @@ def is_retrofitted_rig(rig_name: str) -> bool:
     """Retrofitted rigs are behavior box rigs that have a speaker attached and
     have a different solenoid.
     """
-    return not rig_name.startswith("G")
+    return not rig_name.startswith("G")

np_codeocean/metadata/storage.py

@@ -1,78 +1,78 @@
-"""Stores a history of object changes in local storage. Simple updater and
- getter for iterations of a file.
-"""
-
-import datetime
-import json
-import logging
-import pathlib
-import shutil
-
-logger = logging.getLogger(__name__)
-
-
-def _generate_item_filename(
-    *tags: str,
-) -> str:
-    return "_".join(str(tag) for tag in tags) + "_rig.json"
-
-
-TIMESTAMP_FORMAT = "%Y-%m-%d-%H-%M-%S"
-
-
-def _item_path_sorter(path: pathlib.Path) -> datetime.datetime:
-    mtime_str = path.stem.replace("_rig", "").split("_")[-1]
-    return datetime.datetime.strptime(mtime_str, TIMESTAMP_FORMAT)
-
-
-def get_item(
-    storage_directory: pathlib.Path,
-    timestamp: datetime.datetime,
-    *_tags: str,
-) -> pathlib.Path | None:
-    """Gets latest item before or equal to timestamp."""
-    search_pattern = _generate_item_filename(*_tags, "*")
-    logger.info("Search pattern: %s" % search_pattern)
-    items = list(storage_directory.glob(search_pattern))
-    if not items:
-        logger.debug("No item found for tags: %s" % json.dumps(_tags))
-        return None
-
-    sorted_items = sorted(items, key=_item_path_sorter)
-    logger.debug("Fetched items: %s" % sorted_items)
-    logger.debug([_item_path_sorter(item) for item in sorted_items])
-    filtered = list(
-        filter(
-            lambda item: _item_path_sorter(item) <= timestamp,
-            sorted_items,
-        )
-    )
-    logger.debug("Filtered items: %s" % filtered)
-    if not filtered:
-        return None
-    return filtered[-1]
-
-
-def update_item(
-    storage_directory: pathlib.Path,
-    filepath: pathlib.Path,
-    timestamp: datetime.datetime,
-    *_tags: str,
-) -> pathlib.Path:
-    """
-    Notes
-    -----
-    - Also used to initialize a new base rig in local storage.
-    - If a rig with the same name doest not exist, a new rig will be created.
-    """
-    filename = _generate_item_filename(
-        *_tags,
-        timestamp.strftime(TIMESTAMP_FORMAT),
-    )
-
-    return pathlib.Path(
-        shutil.copy2(
-            filepath,
-            storage_directory / filename,
-        )
-    )
+"""Stores a history of object changes in local storage. Simple updater and
+getter for iterations of a file.
+"""
+
+import datetime
+import json
+import logging
+import pathlib
+import shutil
+
+logger = logging.getLogger(__name__)
+
+
+def _generate_item_filename(
+    *tags: str,
+) -> str:
+    return "_".join(str(tag) for tag in tags) + "_rig.json"
+
+
+TIMESTAMP_FORMAT = "%Y-%m-%d-%H-%M-%S"
+
+
+def _item_path_sorter(path: pathlib.Path) -> datetime.datetime:
+    mtime_str = path.stem.replace("_rig", "").split("_")[-1]
+    return datetime.datetime.strptime(mtime_str, TIMESTAMP_FORMAT)
+
+
+def get_item(
+    storage_directory: pathlib.Path,
+    timestamp: datetime.datetime,
+    *_tags: str,
+) -> pathlib.Path | None:
+    """Gets latest item before or equal to timestamp."""
+    search_pattern = _generate_item_filename(*_tags, "*")
+    logger.info("Search pattern: %s" % search_pattern)
+    items = list(storage_directory.glob(search_pattern))
+    if not items:
+        logger.debug("No item found for tags: %s" % json.dumps(_tags))
+        return None
+
+    sorted_items = sorted(items, key=_item_path_sorter)
+    logger.debug("Fetched items: %s" % sorted_items)
+    logger.debug([_item_path_sorter(item) for item in sorted_items])
+    filtered = list(
+        filter(
+            lambda item: _item_path_sorter(item) <= timestamp,
+            sorted_items,
+        )
+    )
+    logger.debug("Filtered items: %s" % filtered)
+    if not filtered:
+        return None
+    return filtered[-1]
+
+
+def update_item(
+    storage_directory: pathlib.Path,
+    filepath: pathlib.Path,
+    timestamp: datetime.datetime,
+    *_tags: str,
+) -> pathlib.Path:
+    """
+    Notes
+    -----
+    - Also used to initialize a new base rig in local storage.
+    - If a rig with the same name doest not exist, a new rig will be created.
+    """
+    filename = _generate_item_filename(
+        *_tags,
+        timestamp.strftime(TIMESTAMP_FORMAT),
+    )
+
+    return pathlib.Path(
+        shutil.copy2(
+            filepath,
+            storage_directory / filename,
+        )
+    )

np_codeocean/metadata/update.py

@@ -1,4 +1,3 @@
-
 import datetime
 import logging
 import pathlib
@@ -153,4 +152,4 @@ def update_rig(
         else:
             update_rig_modification_date(build_source, datetime.date.today())
 
-    return pathlib.Path(shutil.copy2(build_source, output_path))
+    return pathlib.Path(shutil.copy2(build_source, output_path))

np_codeocean/metadata/utils.py

@@ -148,4 +148,4 @@ def save_aind_model(
 #     """Retrofitted rigs are behavior box rigs that have a speaker attached and
 #      have a different solenoid.
 #     """
-#     return not rig_name.startswith("G")
+#     return not rig_name.startswith("G")