implemented split() from stam-tools

Author: proycon

src/annotationstore.rs

@@ -14,6 +14,7 @@ use crate::resources::PyTextResource;
 use crate::selector::PySelector;
 use crate::textselection::PyTextSelection;
 use stam::*;
+use stamtools::split::{split, SplitMode};
 use stamtools::view::{AnsiWriter, Highlight, HtmlWriter};
 
 #[pyclass(dict, module = "stam", name = "AnnotationStore")]
@@ -556,6 +557,23 @@ impl PyAnnotationStore {
             )),
         }
     }
+
+    #[pyo3(signature = (querystrings, retain))]
+    fn split<'py>(&mut self, querystrings: Vec<&str>, retain: bool) -> PyResult<()> {
+        let mode = if retain {
+            SplitMode::Retain
+        } else {
+            SplitMode::Delete
+        };
+        let mut queries: Vec<Query> = Vec::new();
+        for querystring in querystrings {
+            let query: Query = querystring
+                .try_into()
+                .map_err(|err| PyStamError::new_err(format!("{}", err)))?;
+            queries.push(query);
+        }
+        self.map_store_mut(|store| Ok(split(store, queries, mode, false)))
+    }
 }
 
 pub(crate) trait MapStore {

stam.pyi

@@ -365,6 +365,12 @@ class AnnotationStore:
             Collapse all tags on initial view (html only)
         """
 
+    def split(self, queries: List[str], retain: bool):
+        """
+        Splits an annotation store by either retaining (if `retain == True`) or removing (if `retain == False`) the items selected by the queries.
+        Queries must be STAMQL queries that select annotations, resources or datasets.
+        This deletes items from the store along with all their dependencies and comes with a reasonable performance overhead.
+        """