Fix

Signed-off-by: Geoffroy Jamgotchian <[email protected]>

Author: geofjamg

pypowsybl/network/impl/network.py

@@ -127,7 +127,7 @@ def __getstate__(self) -> Dict[str, Any]:
                 'nominal_apparent_power': self._nominal_apparent_power}
 
     def __setstate__(self, state: Dict[str, Any]) -> None:
-        self._handle = _pp.load_network_from_binary_buffers([state['biidm'].getbuffer()], {}, None)
+        self._handle = _pp.load_network_from_binary_buffers([state['biidm'].getbuffer()], {}, [], None)
         self._per_unit = state['per_unit']
         self._nominal_apparent_power = state['nominal_apparent_power']
         self.__init_from_handle()

pypowsybl/network/impl/network_creation_util.py

@@ -256,7 +256,7 @@ def load_from_binary_buffers(buffers: List[io.BytesIO], parameters: Dict[str, st
                                                         None if report_node is None else report_node._report_node))  # pylint: disable=protected-access
 
 
-def load_from_string(file_name: str, file_content: str, parameters: Dict[str, str] = None,
+def load_from_string(file_name: str, file_content: str, parameters: Dict[str, str] = None, post_processors: List[str] = None,
                      reporter: ReportNode = None, report_node: ReportNode = None) -> Network:
     """
     Load a network from a string. File content should be in a supported format.
@@ -265,6 +265,7 @@ def load_from_string(file_name: str, file_content: str, parameters: Dict[str, st
        file_name:    file name
        file_content: file content
        parameters:   a dictionary of import parameters
+       post_processors: a list of import post processors (will be added to the ones defined by the platform config)
        reporter: deprecated, use report_node instead
        report_node: the reporter to be used to create an execution report, default is None (no report)
 
@@ -274,7 +275,7 @@ def load_from_string(file_name: str, file_content: str, parameters: Dict[str, st
     if reporter is not None:
         warnings.warn(DEPRECATED_REPORTER_WARNING, DeprecationWarning)
         report_node = reporter
-    if parameters is None:
-        parameters = {}
-    return Network(_pp.load_network_from_string(file_name, file_content, parameters,
+    return Network(_pp.load_network_from_string(file_name, file_content,
+                                                {} if parameters is None else parameters,
+                                                [] if post_processors is None else post_processors,
                                                 None if report_node is None else report_node._report_node))  # pylint: disable=protected-access

tests/test_network.py

@@ -61,6 +61,11 @@ def test_load_cgmes_two_zip():
     assert 3 == len(n.get_substations())
 
 
+def test_load_post_processor():
+    assert ['loadflowResultsCompletion', 'odreGeoDataImporter', 'replaceTieLinesByLines'] == pp.network.get_import_post_processors()
+    pp.network.load(DATA_DIR.joinpath('CGMES_Full.zip'), post_processors=['replaceTieLinesByLines'])
+
+
 def test_save_cgmes_zip():
     n = pp.network.create_eurostag_tutorial_example1_network()
     buffer = n.save_to_binary_buffer(format='CGMES')