zombie-squirrel 0.1.0__tar.gz → 0.1.1__tar.gz

{zombie_squirrel-0.1.0/src/zombie_squirrel.egg-info → zombie_squirrel-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zombie-squirrel
-Version: 0.1.0
+Version: 0.1.1
 Summary: Generated from aind-library-template
 Author: Allen Institute for Neural Dynamics
 License: MIT

zombie_squirrel-0.1.1/src/zombie_squirrel/__init__.py

@@ -0,0 +1,4 @@
+"""Init package"""
+__version__ = "0.1.1"
+
+from zombie_squirrel.squirrels import unique_project_names

{zombie_squirrel-0.1.0 → zombie_squirrel-0.1.1}/src/zombie_squirrel/squirrels.py

@@ -4,6 +4,7 @@ from typing import Any, Callable, Optional
 from zombie_squirrel.acorns import RedshiftAcorn, MemoryAcorn
 from aind_data_access_api.document_db import MetadataDbClient
 import os
+import logging
 
 # --- Backend setup ---------------------------------------------------
 
@@ -12,8 +13,10 @@ API_GATEWAY_HOST = "api.allenneuraldynamics.org"
 tree_type = os.getenv("TREE_SPECIES", "memory").lower()
 
 if tree_type == "redshift":
+    logging.info("Using Redshift acorn for caching")
     ACORN = RedshiftAcorn()
 else:
+    logging.info("Using in-memory acorn for caching")
     ACORN = MemoryAcorn()
 
 # --- Squirrel registry -----------------------------------------------------
@@ -31,13 +34,23 @@ def register_squirrel(name: str):
 
 # --- Squirrels -----------------------------------------------------
 
+NAMES = {
+    "upn": "unique-project-names",
+    "usi": "unique-subject-ids",
+}
 
-@register_squirrel("unique-project-names")
+
+@register_squirrel(NAMES["upn"])
 def unique_project_names(force_update: bool = False) -> list[str]:
-    df = ACORN.scurry("unique-project-names")
+    try:
+        df = ACORN.scurry(NAMES["upn"])
+    except Exception as e:
+        logging.warning(f"Error fetching from cache: {e}")
+        df = pd.DataFrame()
 
     if df.empty or force_update:
         # If cache is missing, fetch data
+        logging.info("Updating cache for unique project names")
         client = MetadataDbClient(
             host=API_GATEWAY_HOST,
             version="v2",
@@ -49,14 +62,14 @@ def unique_project_names(force_update: bool = False) -> list[str]:
             ]
         )
         df = pd.DataFrame(unique_project_names)
-        ACORN.hide("unique-project-names", df)
+        ACORN.hide(NAMES["upn"], df)
 
     return df["project_name"].tolist()
 
 
-@register_squirrel("unique-subject-ids")
+@register_squirrel(NAMES["usi"])
 def unique_subject_ids(force_update: bool = False) -> list[str]:
-    df = ACORN.scurry("unique-subject-ids")
+    df = ACORN.scurry(NAMES["usi"])
 
     if df.empty or force_update:
         # If cache is missing, fetch data
@@ -71,6 +84,6 @@ def unique_subject_ids(force_update: bool = False) -> list[str]:
             ]
         )
         df = pd.DataFrame(unique_subject_ids)
-        ACORN.hide("unique-subject-ids", df)
+        ACORN.hide(NAMES["usi"], df)
 
     return df["subject_id"].tolist()

{zombie_squirrel-0.1.0 → zombie_squirrel-0.1.1/src/zombie_squirrel.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zombie-squirrel
-Version: 0.1.0
+Version: 0.1.1
 Summary: Generated from aind-library-template
 Author: Allen Institute for Neural Dynamics
 License: MIT

zombie_squirrel-0.1.0/src/zombie_squirrel/__init__.py

@@ -1,4 +0,0 @@
-"""Init package"""
-__version__ = "0.1.0"
-
-from zombie_squirrel.squirrels import unique_project_names