Commit e031586d authored by Eric Duminil's avatar Eric Duminil
Browse files

Allow to filter with multiple strings

parent db4fd9ca
...@@ -189,9 +189,9 @@ class SimStadtResults(ABC): ...@@ -189,9 +189,9 @@ class SimStadtResults(ABC):
def get_all_by_extension(self, ext: str) -> list[Path]: def get_all_by_extension(self, ext: str) -> list[Path]:
return [f for f in sorted(self.output_files) if f.suffix == ext] return [f for f in sorted(self.output_files) if f.suffix == ext]
def get_unique_by_extension(self, ext: str, filter: str | None = None) -> Path: def get_unique_by_extension(self, ext: str, *filters: str) -> Path:
files_by_extension = self.get_all_by_extension(ext) files_by_extension = self.get_all_by_extension(ext)
if filter: for filter in filters:
files_by_extension = [f for f in files_by_extension if filter in f.name] files_by_extension = [f for f in files_by_extension if filter in f.name]
if len(files_by_extension) == 0: if len(files_by_extension) == 0:
raise ValueError(f"Workflow didn't seem to have returned any {ext} file!") raise ValueError(f"Workflow didn't seem to have returned any {ext} file!")
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment