component filename, default fmt ascii, revert sim path -> workspace

Author: wpbonelli

docs/examples/quickstart.py

@@ -13,7 +13,7 @@
 workspace = Path(__file__).parent / name
 time = ModelTime(perlen=[1.0], nstp=[1])
 grid = StructuredGrid(nlay=1, nrow=10, ncol=10)
-sim = Simulation(name=name, path=workspace, tdis=time)
+sim = Simulation(name=name, workspace=workspace, tdis=time)
 ims = Ims(parent=sim)
 gwf_name = "mymodel"
 gwf = Gwf(parent=sim, name=gwf_name, save_flows=True, dis=grid)

docs/examples/quickstart_expanded.py

@@ -31,7 +31,7 @@
 
 ws = "./mymodel"
 name = "mymodel"
-sim = Simulation(name=name, path=ws, exe="mf6")
+sim = Simulation(name=name, workspace=ws, exe="mf6")
 tdis = Tdis(sim)
 gwf = Gwf(sim, name=name, save_flows=True)
 dis = Dis(gwf, nrow=10, ncol=10)

flopy4/mf6/__init__.py

@@ -7,8 +7,8 @@
 
 def _default_filename(component: Component) -> str:
     """Default path for a component, based on its name."""
-    if hasattr(component, "filename") and component.filename is not None:
-        return component.filename
+    if (filename := component.filename):
+        return filename
     name = component.name  # type: ignore
     cls_name = component.__class__.__name__.lower()
     return f"{name}.{cls_name}"

flopy4/mf6/component.py

@@ -4,7 +4,7 @@
 from modflow_devtools.dfn import Dfn, Field
 from xattree import xattree
 
-from flopy4.mf6.spec import fields_dict, to_dfn_field
+from flopy4.mf6.spec import field, fields_dict, to_dfn_field
 from flopy4.uio import IO, Loader, Writer
 
 COMPONENTS = {}
@@ -27,6 +27,10 @@ class Component(ABC, MutableMapping):
     _load = IO(Loader)  # type: ignore
     _write = IO(Writer)  # type: ignore
 
+    # NOTE: this needs to be overridden in subclasses
+    # with required fields, otherwise attrs complains
+    filename: str = field(default=None)
+
     @classmethod
     def __attrs_init_subclass__(cls):
         COMPONENTS[cls.__name__.lower()] = cls
@@ -51,11 +55,11 @@ def __len__(self):
     def get_dfn(cls) -> Dfn:
         fields = {field_name: to_dfn_field(field) for field_name, field in fields_dict(cls).items()}
         blocks: dict[str, dict[str, Field]] = {}
-        for field_name, field in fields.items():
-            if (block := field.get("block", None)) is not None:
-                blocks.setdefault(block, {})[field_name] = field
+        for field_name, field_ in fields.items():
+            if (block := field_.get("block", None)) is not None:
+                blocks.setdefault(block, {})[field_name] = field_
             else:
-                blocks[field_name] = field
+                blocks[field_name] = field_
 
         return Dfn(
             name=cls.__name__.lower(),

flopy4/mf6/exchange.py

@@ -8,7 +8,7 @@
 
 @xattree
 class Exchange(Package):
-    exgtype: type = field()
-    exgfile: Path = field()
+    exgtype: type = field()  # type: ignore
+    exgfile: Path = field()  # type: ignore
     exgmnamea: Optional[str] = field(default=None)
     exgmnameb: Optional[str] = field(default=None)

flopy4/mf6/simulation.py

@@ -1,15 +1,16 @@
 from os import PathLike
 from pathlib import Path
-from typing import ClassVar
+from warnings import warn
 
 from flopy.discretization.modeltime import ModelTime
 from modflow_devtools.misc import cd, run_cmd
-from xattree import field, xattree
+from xattree import xattree
 
 from flopy4.mf6.component import Component
 from flopy4.mf6.exchange import Exchange
 from flopy4.mf6.model import Model
 from flopy4.mf6.solution import Solution
+from flopy4.mf6.spec import field
 from flopy4.mf6.tdis import Tdis
 
 
@@ -27,31 +28,39 @@ class Simulation(Component):
     exchanges: dict[str, Exchange] = field()
     solutions: dict[str, Solution] = field()
     tdis: Tdis = field(converter=convert_time)
-    # TODO: decorator for components bound
-    # to some directory or file path?
-    path: Path = field(default=None)
-    filename: ClassVar[str] = "mfsim.nam"
+    workspace: Path = field(default=None)
+    filename: str = field(default="mfsim.nam")
+
+    def __attrs_post_init__(self):
+        """Post-initialization hook to set up the simulation."""
+        super().__attrs_post_init__()
+        if self.filename != "mfsim.nam":
+            warn(
+                "Simulation filename must be 'mfsim.nam'.",
+                UserWarning,
+            )
+            self.filename = "mfsim.nam"
 
     @property
     def time(self) -> ModelTime:
         return self.tdis.to_time()
 
     def run(self, exe: str | PathLike = "mf6", verbose: bool = False) -> None:
         """Run the simulation using the given executable."""
-        if self.path is None:
+        if self.workspace is None:
             raise ValueError(f"Simulation {self.name} has no workspace path.")
-        with cd(self.path):
+        with cd(self.workspace):
             stdout, stderr, retcode = run_cmd(exe, verbose=verbose)
             if retcode != 0:
                 raise RuntimeError(
                     f"Simulation {self.name}: {exe} failed to run with returncode "  # type: ignore
                     f"{retcode}, and error message:\n\n{stdout + stderr} "
                 )
 
-    def load(self, format):
-        with cd(self.path):
+    def load(self, format="ascii"):
+        with cd(self.workspace):
             super().load(format)
 
-    def write(self, format):
-        with cd(self.path):
+    def write(self, format="ascii"):
+        with cd(self.workspace):
             super().write(format)

test/test_component.py

@@ -275,15 +275,15 @@ def test_ims_dfn():
 def test_write_ascii(tmp_path):
     time = ModelTime(perlen=[1.0], nstp=[1], tsmult=[1.0])
     grid = StructuredGrid(nlay=1, nrow=10, ncol=10)
-    sim = Simulation(tdis=time, path=tmp_path)
+    sim = Simulation(tdis=time, workspace=tmp_path)
     # TODO fix errors
     # gwf = Gwf(parent=sim, dis=grid)
     # ic = Ic(parent=gwf)
     # oc = Oc(parent=gwf)
     # npf = Npf(parent=gwf)
     # chd = Chd(parent=gwf, head={"*": {(0, 0, 0): 1.0, (0, 9, 9): 0.0}})
 
-    sim.write("ascii")
+    sim.write()
 
     files = list(Path(tmp_path).glob("*"))
     file_names = [f.name for f in files]