Storage migration

Author: xjules

src/ert/storage/local_storage.py

@@ -448,6 +448,7 @@ def _save_index(self) -> None:
     @require_write
     def _migrate(self, version: int) -> None:
         from ert.storage.migration import (  # noqa: PLC0415
+            to2,
             to3,
             to4,
             to5,
@@ -502,7 +503,7 @@ def _migrate(self, version: int) -> None:
 
             elif version < _LOCAL_STORAGE_VERSION:
                 migrations = list(
-                    enumerate([to3, to4, to5, to6, to7, to8, to9, to10], start=1)
+                    enumerate([to2, to3, to4, to5, to6, to7, to8, to9, to10], start=1)
                 )
                 for from_version, migration in migrations[version - 1 :]:
                     print(f"* Updating storage to version: {from_version + 1}")

src/ert/storage/migration/to10.py

@@ -12,6 +12,6 @@ def migrate(path: Path) -> None:
         with open(experiment / "parameter.json", "w", encoding="utf-8") as fout:
             for param in parameters_json.values():
                 if param["_ert_kind"] == "GenKwConfig":
-                    del param["template_file"]
-                    del param["output_file"]
+                    param.pop("template_file", None)
+                    param.pop("output_file", None)
             fout.write(json.dumps(parameters_json, indent=4))

tests/ert/unit_tests/storage/snapshots/test_storage_migration/test_that_storage_matches/parameters

@@ -1 +1 @@
-{'BPR': GenKwConfig(name='BPR', forward_init=False, update=True, template_file='/home/eivind/Projects/ert/test-data/all_data_types/template.txt', output_file='params.txt', transform_function_definitions=[{'name': 'BPR', 'param_name': 'NORMAL', 'values': ['0', '1']}], forward_init_file=None), 'PORO': Field(name='PORO', forward_init=False, update=True, nx=2, ny=3, nz=4, file_format=<FieldFileFormat.GRDECL: 'grdecl'>, output_transformation=None, input_transformation=None, truncation_min=None, truncation_max=None, forward_init_file='data/poro%d.grdecl', output_file=PosixPath('poro.grdecl'), grid_file='/home/eivind/Projects/ert/test-data/all_data_types/refcase/CASE.EGRID', mask_file=''), 'TOP': SurfaceConfig(name='TOP', forward_init=False, update=True, ncol=2, nrow=3, xori=0.0, yori=0.0, xinc=1.0, yinc=1.0, rotation=0.0, yflip=1, forward_init_file='data/surf%d.irap', output_file='surf.irap', base_surface_path='data/basesurf.irap')}
+{'BPR': GenKwConfig(name='BPR', forward_init=False, update=True, transform_function_definitions=[{'name': 'BPR', 'param_name': 'NORMAL', 'values': ['0', '1']}], forward_init_file=None), 'PORO': Field(name='PORO', forward_init=False, update=True, nx=2, ny=3, nz=4, file_format=<FieldFileFormat.GRDECL: 'grdecl'>, output_transformation=None, input_transformation=None, truncation_min=None, truncation_max=None, forward_init_file='data/poro%d.grdecl', output_file=PosixPath('poro.grdecl'), grid_file='/home/eivind/Projects/ert/test-data/all_data_types/refcase/CASE.EGRID', mask_file=''), 'TOP': SurfaceConfig(name='TOP', forward_init=False, update=True, ncol=2, nrow=3, xori=0.0, yori=0.0, xinc=1.0, yinc=1.0, rotation=0.0, yflip=1, forward_init_file='data/surf%d.irap', output_file='surf.irap', base_surface_path='data/basesurf.irap')}