|
8 | 8 | from cognite.client.utils.useful_types import SequenceNotStr
|
9 | 9 |
|
10 | 10 |
|
| 11 | +def parse_str_or_sequence(value: str | Sequence[str] | None) -> list[str] | None: |
| 12 | + if isinstance(value, str): |
| 13 | + return [value] |
| 14 | + elif isinstance(value, Sequence): |
| 15 | + return list(value) |
| 16 | + return None |
| 17 | + |
| 18 | + |
11 | 19 | class SimulatorIntegrationFilter(CogniteFilter):
|
12 | 20 | def __init__(
|
13 | 21 | self,
|
@@ -52,27 +60,11 @@ def __init__(
|
52 | 60 | model_revision_external_ids: str | Sequence[str] | None = None,
|
53 | 61 | ) -> None:
|
54 | 62 | self.model_external_ids = [model_external_ids] if isinstance(model_external_ids, str) else model_external_ids
|
55 |
| - self.simulator_integration_external_ids = ( |
56 |
| - [simulator_integration_external_ids] |
57 |
| - if isinstance(simulator_integration_external_ids, str) |
58 |
| - else simulator_integration_external_ids |
59 |
| - ) |
60 |
| - self.simulator_external_ids = ( |
61 |
| - [simulator_external_ids] if isinstance(simulator_external_ids, str) else simulator_external_ids |
62 |
| - ) |
63 |
| - self.routine_external_ids = ( |
64 |
| - [routine_external_ids] if isinstance(routine_external_ids, str) else routine_external_ids |
65 |
| - ) |
66 |
| - self.routine_revision_external_ids = ( |
67 |
| - [routine_revision_external_ids] |
68 |
| - if isinstance(routine_revision_external_ids, str) |
69 |
| - else routine_revision_external_ids |
70 |
| - ) |
71 |
| - self.model_revision_external_ids = ( |
72 |
| - [model_revision_external_ids] |
73 |
| - if isinstance(model_revision_external_ids, str) |
74 |
| - else model_revision_external_ids |
75 |
| - ) |
| 63 | + self.simulator_integration_external_ids = parse_str_or_sequence(simulator_integration_external_ids) |
| 64 | + self.simulator_external_ids = parse_str_or_sequence(simulator_external_ids) |
| 65 | + self.routine_external_ids = parse_str_or_sequence(routine_external_ids) |
| 66 | + self.routine_revision_external_ids = parse_str_or_sequence(routine_revision_external_ids) |
| 67 | + self.model_revision_external_ids = parse_str_or_sequence(model_revision_external_ids) |
76 | 68 | self.status = status
|
77 | 69 | self.run_type = run_type
|
78 | 70 |
|
|
0 commit comments