fix export missing stream err

Seniatical · Apr 9, 2023 · 1d9d431 · 1d9d431
1 parent 03b27e8
commit 1d9d431
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/ezycore/drivers/core.py b/ezycore/drivers/core.py
@@ -93,11 +93,13 @@ def map_to_model(self, **kwds) -> None:
         """
 
     @abstractmethod
-    def export(self, stream: Iterator[Union[dict, Model]], include: set, exclude: set) -> None:
+    def export(self, location: str, stream: Iterator[Union[dict, Model]], include: set, exclude: set) -> None:
         """ Exports data from any object supporting the `__next__` method
 
         Parameters
         ----------
+        location: str
+            Location to export data to
         stream: Iterator[Union[:class:`dict`, :class:`Model`]]
             An object which returns values to export
         include: :class:`set`

diff --git a/ezycore/manager/core.py b/ezycore/manager/core.py
@@ -327,7 +327,7 @@ def populate_using_driver(self, location: str, driver: Driver, **driver_kwargs)
     def export_segment(self, location: str, driver: Driver = None, **driver_kwargs) -> None:
         seg = self.get_segment(location)
 
-        driver.export(seg, **driver_kwargs)
+        driver.export(location, seg, **driver_kwargs)
 
     ###########################################################################################
     ##