|
|
from dataclasses import dataclass, field |
|
|
|
|
|
try: |
|
|
import trackio.utils as utils |
|
|
except ImportError: |
|
|
import utils |
|
|
|
|
|
|
|
|
@dataclass |
|
|
class RunSelection: |
|
|
choices: list[str] = field(default_factory=list) |
|
|
selected: list[str] = field(default_factory=list) |
|
|
locked: bool = False |
|
|
|
|
|
def update_choices( |
|
|
self, runs: list[str], preferred: list[str] | None = None |
|
|
) -> bool: |
|
|
if self.choices == runs: |
|
|
return False |
|
|
new_choices = set(runs) - set(self.choices) |
|
|
self.choices = list(runs) |
|
|
if self.locked: |
|
|
base = set(self.selected) | new_choices |
|
|
elif preferred: |
|
|
base = set(preferred) |
|
|
else: |
|
|
base = set(runs) |
|
|
self.selected = [run for run in self.choices if run in base] |
|
|
return True |
|
|
|
|
|
def select(self, runs: list[str]) -> list[str]: |
|
|
choice_set = set(self.choices) |
|
|
self.selected = [run for run in runs if run in choice_set] |
|
|
self.locked = True |
|
|
return self.selected |
|
|
|
|
|
def replace_group( |
|
|
self, group_runs: list[str], new_subset: list[str] | None |
|
|
) -> tuple[list[str], list[str]]: |
|
|
new_subset = utils.ordered_subset(group_runs, new_subset) |
|
|
selection_set = set(self.selected) |
|
|
selection_set.difference_update(group_runs) |
|
|
selection_set.update(new_subset) |
|
|
self.selected = [run for run in self.choices if run in selection_set] |
|
|
self.locked = True |
|
|
return new_subset, self.selected |
|
|
|
