oarepo-runtime 1.2.2__tar.gz → 1.2.4__tar.gz

{oarepo-runtime-1.2.2/oarepo_runtime.egg-info → oarepo-runtime-1.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oarepo-runtime
-Version: 1.2.2
+Version: 1.2.4
 Summary: A set of runtime extensions of Invenio repository
 Description-Content-Type: text/markdown
 Provides-Extra: devs

{oarepo-runtime-1.2.2 → oarepo-runtime-1.2.4}/oarepo_runtime/datastreams/datastreams.py

@@ -9,6 +9,9 @@ from oarepo_runtime.datastreams.config import DATASTREAM_READERS, get_instance
 
 from .errors import TransformerError, WriterError
 import abc
+import logging
+
+log = logging.getLogger("datastreams")
 
 
 class StreamEntry:
@@ -128,12 +131,10 @@ class DataStream(AbstractDataStream):
             try:
                 writer.write(stream_entry)
             except WriterError as err:
-                self._log.error("Error in writer: ", err, repr(stream_entry.entry))
+                log.error("Error in writer: ", err, repr(stream_entry.entry))
                 stream_entry.errors.append(f"{writer.__class__.__name__}: {str(err)}")
             except Exception as err:
-                self._log.error(
-                    "Unexpected error in writer: ", err, repr(stream_entry.entry)
-                )
+                log.error("Unexpected error in writer: ", err, repr(stream_entry.entry))
                 stream_entry.errors.append(f"{writer.__class__.__name__}: {str(err)}")
 
         return stream_entry

{oarepo-runtime-1.2.2 → oarepo-runtime-1.2.4}/oarepo_runtime/datastreams/transformers.py

@@ -6,7 +6,7 @@ from .datastreams import StreamEntry
 class BaseTransformer(ABC):
     """Base transformer."""
 
-    def __init__(self) -> None:
+    def __init__(self, **kwargs) -> None:
         pass
 
     @abstractmethod
@@ -20,9 +20,12 @@ class BaseTransformer(ABC):
 class BatchTransformer(BaseTransformer):
     @abstractmethod
     def apply_batch(
-        self, stream_entry: List[StreamEntry], *args, **kwargs
+        self, stream_entries: List[StreamEntry], *args, **kwargs
     ) -> List[StreamEntry]:
         """Applies the transformation to the entry.
         :returns: A StreamEntry. The transformed entry.
                   Raises TransformerError in case of errors.
         """
+
+    def apply(self, stream_entry: StreamEntry, *args, **kwargs) -> StreamEntry:
+        return self.apply_batch([stream_entry], *args, **kwargs)

{oarepo-runtime-1.2.2 → oarepo-runtime-1.2.4}/oarepo_runtime/datastreams/writers/__init__.py

@@ -24,8 +24,11 @@ class BaseWriter(ABC):
 
 class BatchWriter(BaseWriter):
     @abstractmethod
-    def write_batch(self, entry: List[StreamEntry], *args, **kwargs):
+    def write_batch(self, entries: List[StreamEntry], *args, **kwargs):
         """Writes the input entry to the target output.
         :returns: nothing
                   Raises WriterException in case of errors.
         """
+
+    def write(self, entry: StreamEntry, *args, **kwargs):
+        return self.write_batch([entry], *args, **kwargs)

{oarepo-runtime-1.2.2 → oarepo-runtime-1.2.4}/oarepo_runtime/ext.py

@@ -1,4 +1,5 @@
 from .cli import oarepo as oarepo_cmd
+import oarepo_runtime.cf.cli  # noqa, just to register
 import oarepo_runtime.datastreams.cli  # noqa, just to register
 
 

{oarepo-runtime-1.2.2 → oarepo-runtime-1.2.4/oarepo_runtime.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oarepo-runtime
-Version: 1.2.2
+Version: 1.2.4
 Summary: A set of runtime extensions of Invenio repository
 Description-Content-Type: text/markdown
 Provides-Extra: devs

{oarepo-runtime-1.2.2 → oarepo-runtime-1.2.4}/setup.cfg

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