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

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

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

zombie_squirrel-0.1.3/src/zombie_squirrel/__init__.py

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

{zombie_squirrel-0.1.1 → zombie_squirrel-0.1.3}/src/zombie_squirrel/acorns.py

@@ -4,10 +4,7 @@ import pandas as pd
 import os
 
 from aind_data_access_api.rds_tables import Client, RDSCredentials
-
-
-def prefix_table_name(table_name: str) -> str:
-    return "zs_" + table_name
+from zombie_squirrel.utils import prefix_table_name
 
 
 class Acorn(ABC):

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

@@ -5,6 +5,7 @@ from zombie_squirrel.acorns import RedshiftAcorn, MemoryAcorn
 from aind_data_access_api.document_db import MetadataDbClient
 import os
 import logging
+from zombie_squirrel.utils import prefix_table_name, rds_get_handle_empty
 
 # --- Backend setup ---------------------------------------------------
 
@@ -42,11 +43,7 @@ NAMES = {
 
 @register_squirrel(NAMES["upn"])
 def unique_project_names(force_update: bool = False) -> list[str]:
-    try:
-        df = ACORN.scurry(NAMES["upn"])
-    except Exception as e:
-        logging.warning(f"Error fetching from cache: {e}")
-        df = pd.DataFrame()
+    df = rds_get_handle_empty(ACORN, NAMES["upn"])
 
     if df.empty or force_update:
         # If cache is missing, fetch data
@@ -69,7 +66,7 @@ def unique_project_names(force_update: bool = False) -> list[str]:
 
 @register_squirrel(NAMES["usi"])
 def unique_subject_ids(force_update: bool = False) -> list[str]:
-    df = ACORN.scurry(NAMES["usi"])
+    df = rds_get_handle_empty(ACORN, NAMES["usi"])
 
     if df.empty or force_update:
         # If cache is missing, fetch data

zombie_squirrel-0.1.3/src/zombie_squirrel/utils.py

@@ -0,0 +1,20 @@
+"""Utility functions"""
+import pandas as pd
+import logging
+
+from zombie_squirrel.acorns import Acorn
+
+
+def prefix_table_name(table_name: str) -> str:
+    return "zs_" + table_name
+
+
+def rds_get_handle_empty(acorn: Acorn, table_name: str) -> pd.DataFrame:
+    """Utility function for testing purposes."""
+    try:
+        df = acorn.scurry(table_name)
+    except Exception as e:
+        logging.warning(f"Error fetching from cache: {e}")
+        df = pd.DataFrame()
+
+    return df

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

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

{zombie_squirrel-0.1.1 → zombie_squirrel-0.1.3}/src/zombie_squirrel.egg-info/SOURCES.txt

@@ -6,6 +6,7 @@ src/zombie_squirrel/__init__.py
 src/zombie_squirrel/acorns.py
 src/zombie_squirrel/squirrels.py
 src/zombie_squirrel/sync.py
+src/zombie_squirrel/utils.py
 src/zombie_squirrel.egg-info/PKG-INFO
 src/zombie_squirrel.egg-info/SOURCES.txt
 src/zombie_squirrel.egg-info/dependency_links.txt

zombie_squirrel-0.1.1/src/zombie_squirrel/__init__.py

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