oarepo-runtime 1.5.59__tar.gz → 1.5.60__tar.gz

{oarepo_runtime-1.5.59/oarepo_runtime.egg-info → oarepo_runtime-1.5.60}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oarepo-runtime
-Version: 1.5.59
+Version: 1.5.60
 Summary: A set of runtime extensions of Invenio repository
 Description-Content-Type: text/markdown
 License-File: LICENSE

oarepo_runtime-1.5.60/oarepo_runtime/records/systemfields/selectors.py

@@ -0,0 +1,104 @@
+from typing import Any, Callable, List
+
+
+class Selector:
+    def select(self, record) -> List[Any]:
+        return []
+
+
+class PathSelector(Selector):
+    def __init__(self, *paths):
+        self.paths = [x.split(".") for x in paths]
+
+    def select(self, record):
+        ret = []
+        for path in self.paths:
+            for rec in getter(record, path):
+                ret.append(rec)
+        return ret
+
+
+class FirstItemSelector(PathSelector):
+    def select(self, record):
+        for rec in super().select(record):
+            return [rec]
+        return []
+
+
+class FilteredSelector(Selector):
+    """
+    Selector which filters output of another selector
+    Example:
+        FilteredSelector(PathSelector("metadata.creators", "metadata.contributors"),
+                                         filter=lambda x: x["nameType"] == "personal", projection="affiliations")
+
+    selects affiliations of creators with nameType personal from following data
+
+    data = {
+        "metadata": {
+            "creators": [
+                {"name": "hugo", "affiliations": ["uni1", "uni2"], "nameType": "personal"},
+                {"name": "uni3", "nameType": "organizational"},
+            ]
+        }
+    }
+    """
+
+    def __init__(
+        self,
+        selector: Selector,
+        filter: Callable[[Any], bool],
+        projection: Callable[[Any], Any] | str = None,
+    ):
+
+        self.selector = selector
+        self.filter = filter
+        self.projection = projection
+
+    def select(self, record):
+        selected = self.selector.select(record)
+        selected = filter(self.filter, selected)
+        if self.projection:
+            ret = []
+            for select_element in selected:
+                if isinstance(self.projection, str):
+                    if isinstance(select_element, dict) and self.projection in select_element:
+                        result = select_element[self.projection]
+                    else:
+                        result = []
+                else:
+                    result = self.projection(select_element)
+                if isinstance(result, list):
+                    ret += result
+                else:
+                    ret.append(result)
+        else:
+            ret = list(selected)
+        return ret
+
+
+class MultiSelector(Selector):
+    """Selector concatenating outputs of multiple selectors"""
+
+    def __init__(self, *selectors: Selector):
+        self.selectors = selectors
+
+    def select(self, record):
+        ret = []
+        for selector in self.selectors:
+            ret += selector.select(record)
+        return ret
+
+
+def getter(data, path: List):
+    if len(path) == 0:
+        if isinstance(data, list):
+            yield from data
+        else:
+            yield data
+    elif isinstance(data, dict):
+        if path[0] in data:
+            yield from getter(data[path[0]], path[1:])
+    elif isinstance(data, list):
+        for item in data:
+            yield from getter(item, path)

{oarepo_runtime-1.5.59 → oarepo_runtime-1.5.60/oarepo_runtime.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oarepo-runtime
-Version: 1.5.59
+Version: 1.5.60
 Summary: A set of runtime extensions of Invenio repository
 Description-Content-Type: text/markdown
 License-File: LICENSE

{oarepo_runtime-1.5.59 → oarepo_runtime-1.5.60}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = oarepo-runtime
-version = 1.5.59
+version = 1.5.60
 description = A set of runtime extensions of Invenio repository
 authors = Alzbeta Pokorna
 readme = README.md

oarepo_runtime-1.5.59/oarepo_runtime/records/systemfields/selectors.py

@@ -1,39 +0,0 @@
-from typing import Any, List
-
-
-class Selector:
-    def select(self, record) -> List[Any]:
-        return []
-
-
-class PathSelector(Selector):
-    def __init__(self, *paths):
-        self.paths = [x.split(".") for x in paths]
-
-    def select(self, record):
-        ret = []
-        for path in self.paths:
-            for rec in getter(record, path):
-                ret.append(rec)
-        return ret
-
-
-class FirstItemSelector(PathSelector):
-    def select(self, record):
-        for rec in super().select(record):
-            return [rec]
-        return []
-
-
-def getter(data, path: List):
-    if len(path) == 0:
-        if isinstance(data, list):
-            yield from data
-        else:
-            yield data
-    elif isinstance(data, dict):
-        if path[0] in data:
-            yield from getter(data[path[0]], path[1:])
-    elif isinstance(data, list):
-        for item in data:
-            yield from getter(item, path)