synapse table added to setting (#395)

aplbrain · Jan 9, 2023 · 79165a2 · 79165a2
1 parent b97f45e
commit 79165a2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/neuvue_project/neuvue/settings.py b/neuvue_project/neuvue/settings.py
@@ -214,6 +214,7 @@
 # Annotation Tables
 NEURON_TABLE = "nucleus_neuron_svm"
 CELL_CLASS_TABLE = "allen_soma_coarse_cell_class_model_v2"
+SYNAPSE_TABLE = "synapses_pni_2"
 DAYS_UNTIL_EXPIRED = 3
 CACHED_TABLES_PATH = os.path.join(STATIC_ROOT, "tables")
 

diff --git a/neuvue_project/workspace/neuroglancer.py b/neuvue_project/workspace/neuroglancer.py
@@ -756,7 +756,7 @@ def construct_synapse_state(root_ids: List, flags: dict = None):
         if flags["timestamp"] != "None":
             try:
                 pre_synapses = cave_client.materialize.query_table(
-                    "synapses_pni_2",
+                    settings.SYNAPSE_TABLE,
                     filter_in_dict={"pre_pt_root_id": int_root_ids},
                     select_columns=[
                         "ctr_pt_position",
@@ -769,7 +769,7 @@ def construct_synapse_state(root_ids: List, flags: dict = None):
                 raise Exception(f"Root ID {index} not found for this timestamp")
         else:
             pre_synapses = cave_client.materialize.query_table(
-                "synapses_pni_2",
+                settings.SYNAPSE_TABLE,
                 filter_in_dict={"pre_pt_root_id": int_root_ids},
                 select_columns=["ctr_pt_position", "pre_pt_root_id", "post_pt_root_id"],
             )
@@ -788,7 +788,7 @@ def construct_synapse_state(root_ids: List, flags: dict = None):
         if flags["timestamp"] != "None":
             try:
                 post_synapses = cave_client.materialize.query_table(
-                    "synapses_pni_2",
+                    settings.SYNAPSE_TABLE,
                     filter_in_dict={"post_pt_root_id": int_root_ids},
                     select_columns=[
                         "ctr_pt_position",
@@ -801,7 +801,7 @@ def construct_synapse_state(root_ids: List, flags: dict = None):
                 raise Exception(f"Root ID {index} not found for this timestamp")
         else:
             post_synapses = cave_client.materialize.query_table(
-                "synapses_pni_2",
+                settings.SYNAPSE_TABLE,
                 filter_in_dict={"post_pt_root_id": int_root_ids},
                 select_columns=["ctr_pt_position", "post_pt_root_id", "pre_pt_root_id"],
             )