Remove multiple annotators in annotation table

holonote/annotate/annotator.py

@@ -122,7 +122,7 @@ class AnnotatorInterface(param.Parameterized):
 
     connector_class = SQLiteDB
 
-    def __init__(self, spec, *, init=True, **params):
+    def __init__(self, spec, **params):
         connector_kws = {'fields': params.get('fields')} if 'fields' in params else {}
         connector = params.pop('connector') if 'connector' in params else self.connector_class(**connector_kws)
 
@@ -133,12 +133,11 @@ def __init__(self, spec, *, init=True, **params):
             connector.fields = self.fields
         self._region = {}
         self._last_region = None
-        self.annotation_table = AnnotationTable()
-        self.annotation_table.register_annotator(self)
-        self.annotation_table.add_schema_to_conn(self.connector)
 
-        if init:
-            self.load()
+        self.annotation_table = AnnotationTable()
+        self.connector._create_column_schema(self.spec, self.fields)
+        self.connector._initialize(self.connector.column_schema)
+        self.annotation_table.load(self.connector, fields=self.connector.fields, spec=self.spec)
 
     @classmethod
     def normalize_spec(self, input_spec: dict[str, Any]) -> SpecDict:
@@ -178,10 +177,6 @@ def normalize_spec(self, input_spec: dict[str, Any]) -> SpecDict:
 
         return new_spec
 
-    def load(self):
-        self.connector._initialize(self.connector.column_schema)
-        self.annotation_table.load(self.connector, fields=self.connector.fields, spec=self.spec)
-
     @property
     def df(self) -> pd.DataFrame:
         return self.annotation_table.get_dataframe(spec=self.spec)
@@ -192,13 +187,6 @@ def get_dataframe(self, dims) -> pd.DataFrame:
     def refresh(self, clear=False):
         "Method to update display state of the annotator and optionally clear stale visual state"
 
-    def set_annotation_table(self, annotation_table) -> None: # FIXME! Won't work anymore, set_connector??
-        self._region = {}
-        self.annotation_table = annotation_table
-        self.annotation_table.register_annotator(self)
-        self.annotation_table._update_index()
-        self.snapshot()
-
     # Selecting annotations
 
     def select_by_index(self, *inds):
@@ -239,18 +227,14 @@ def _add_annotation(self, **fields):
 
         # Don't do anything if self.region is an empty dict
         if self.region and self.region != self._last_region:
-            if len(self.annotation_table._annotators)>1:
-                msg = 'Multiple annotation instances attached to the connector: Call add_annotation directly from the associated connector.'
-                raise AssertionError(msg)
             self.annotation_table.add_annotation(self._region, spec=self.spec, **fields)
             self._last_region = self._region.copy()
 
     def add_annotation(self, **fields):
         self._add_annotation(**fields)
 
     def update_annotation_region(self, index):
-        self.annotation_table.update_annotation_region(index)
-
+        self.annotation_table.update_annotation_region(self._region, index)
 
     def update_annotation_fields(self, index, **fields):
         self.annotation_table.update_annotation_fields(index, **fields)

holonote/annotate/connector.py

@@ -3,15 +3,14 @@
 import datetime as dt
 import sqlite3
 import uuid
+from typing import TYPE_CHECKING
 
 import numpy as np
 import pandas as pd
 import param
 
-try:
-    import sqlalchemy
-except ModuleNotFoundError:
-    sqlalchemy = None
+if TYPE_CHECKING:
+    from .typing import SpecDict
 
 
 class PrimaryKey(param.Parameterized):
@@ -233,6 +232,12 @@ def _incompatible_schema_check(self, expected_keys, columns, fields, region_type
                             + f'Missing {region_type!r} region columns {missing_region_columns}. '
                             + msg_suffix)
 
+    def _create_column_schema(self, spec: SpecDict, fields: list[str]) -> None:
+        field_dtypes = {col: str for col in fields} # FIXME - generalize
+        all_region_types = [{v["region"] for v in spec.values()}]
+        all_kdim_dtypes = [{k: v["type"] for k, v in spec.items()} ]
+        schema = self.generate_schema(self.primary_key, all_region_types, all_kdim_dtypes, field_dtypes)
+        self.column_schema = schema
 
 class SQLiteDB(Connector):
     """
@@ -361,7 +366,3 @@ def update_row(self, **updates): # updates as a dictionary OR remove posarg?
         query = f"UPDATE {self.table_name} SET " + set_updates + f" WHERE \"{self.primary_key.field_name}\" = ?;"
         self.cursor.execute(query, [*updates.values(), id_val])
         self.con.commit()
-
-    def add_schema(self, schema):
-        # TODO: Check if schema don't overwrite existing columns
-        self.column_schema = {**self.column_schema, **schema}

holonote/annotate/table.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import weakref
 from typing import TYPE_CHECKING, Any
 
 import pandas as pd
@@ -21,7 +20,7 @@ class AnnotationTable(param.Parameterized):
 
     index = param.List(default=[])
 
-    def __init__(self,  **params):
+    def __init__(self, **params):
         """
         Either specify annotation fields with filled field columns
         (via connector or dataframe) or declare the expected
@@ -38,8 +37,6 @@ def __init__(self,  **params):
         self._update_index()
         self._field_df_snapshot, self._region_df_snapshot = None, None
 
-        self._annotators = weakref.WeakValueDictionary()
-
     def load(self, connector=None, fields_df=None, primary_key_name=None, fields=None, spec=None):
         """
         Load the AnnotationTable from a connector or a fields DataFrame.
@@ -68,17 +65,7 @@ def load(self, connector=None, fields_df=None, primary_key_name=None, fields=Non
         self.clear_edits()
         self._update_index()
 
-    def register_annotator(self, annotator):
-        self._annotators[id(annotator)] = annotator
-
-
-    # FIXME: Multiple region updates
-    def update_annotation_region(self, index):
-        region = next(iter(self._annotators.values()))._region
-        if region == {}:
-            print('No new region selected. Skipping')
-            return
-
+    def update_annotation_region(self, region, index):
         value = region['value']
         mask = self._region_df[self._region_df._id == index]
         assert len(mask)==1, 'TODO: Handle multiple region updates for single index'
@@ -174,9 +161,7 @@ def commits(self, connector):
             kwargs = connector.transforms[operation](commit['kwargs'])
             getattr(connector,connector.operation_mapping[operation])(**kwargs)
 
-        for annotator in self._annotators.values():
-            annotator.annotation_table.clear_edits()
-
+        self.clear_edits()
         return commits
 
     def clear_edits(self, edit_type=None):
@@ -206,10 +191,6 @@ def add_annotation(self, regions: dict[str, Any], spec: SpecDict, **fields):
         self._edits.append({'operation':'insert', 'id':index_value})
         self._update_index()
 
-    # def refresh_annotators(self, clear=False):
-    #     for annotator in self._annotators.values():
-    #         annotator.refresh(clear=clear)
-
     def _add_annotation_fields(self, index_value, fields=None):
 
         index_name_set = set() if self._field_df.index.name is None else {self._field_df.index.name}
@@ -348,10 +329,3 @@ def load_annotation_table(self, conn: Connector, fields: list[str], spec: SpecDi
 
         self._update_index()
         self.clear_edits()
-
-    def add_schema_to_conn(self, conn: Connector) -> None:
-        field_dtypes = {col: str for col in conn.fields} # FIXME - generalize
-        all_region_types = [{v["region"] for v in an.spec.values()} for an in self._annotators.values()]
-        all_kdim_dtypes = [{k: v["type"] for k, v in an.spec.items()} for an in self._annotators.values()]
-        schema = conn.generate_schema(conn.primary_key, all_region_types, all_kdim_dtypes, field_dtypes)
-        conn.add_schema(schema)

holonote/app/__init__.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from .panel import PanelWidgets  # noqa: F401

holonote/tests/test_app.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import panel as pn
 
 from holonote.app import PanelWidgets