Skip to content

Commit edcecfb

Browse files
committed
Fields: extract _legacy_file_info
1 parent ff13aa5 commit edcecfb

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

src/stagpy/step.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -342,15 +342,18 @@ def __contains__(self, item: Any) -> bool:
342342
return False
343343
return True
344344

345+
def _legacy_file_info(self, name: str) -> tuple[str, list[str]]:
346+
for filestem, list_fvar in self.files.items():
347+
if name in list_fvar:
348+
return filestem, list_fvar
349+
raise error.UnknownFieldVarError(name)
350+
345351
def _get_raw_data(
346352
self, name: str
347353
) -> tuple[list[str], tuple[dict[str, Any], NDArray[np.float64]] | None]:
348354
"""Find file holding data and return its content."""
349355
# try legacy first, then hdf5
350-
filestem = ""
351-
for filestem, list_fvar in self.files.items():
352-
if name in list_fvar:
353-
break
356+
filestem, list_fvar = self._legacy_file_info(name)
354357
parsed_data = None
355358
if self.step.isnap is None:
356359
return list_fvar, None

0 commit comments

Comments
 (0)