Coretex
list_dataset_parameter.py
1 # Copyright (C) 2023 Coretex LLC
2 
3 # This file is part of Coretex.ai
4 
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License as
7 # published by the Free Software Foundation, either version 3 of the
8 # License, or (at your option) any later version.
9 
10 # This program is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 # GNU Affero General Public License for more details.
14 
15 # You should have received a copy of the GNU Affero General Public License
16 # along with this program. If not, see <https://www.gnu.org/licenses/>.
17 
18 from typing import Any, List, Optional, Union
19 
20 from ..base_list_parameter import BaseListParameter
21 from ...utils import getDatasetType, fetchDataset
22 from ....project import ProjectType
23 from ....dataset import Dataset
24 
25 
26 def _parseSingleDataset(value: Union[int, str], type_: ProjectType) -> Dataset:
27  isLocal = isinstance(value, str)
28  datasetType = getDatasetType(type_, isLocal)
29 
30  dataset = fetchDataset(datasetType, value)
31  if dataset is None:
32  raise ValueError(f">> [Coretex] Failed to create dataset with type \"{datasetType.__name__}\"")
33 
34  return dataset
35 
36 
37 class ListDatasetParameter(BaseListParameter[List[Union[int, str]]]):
38 
39  @property
40  def listTypes(self) -> List[type]:
41  return [int, str]
42 
43  def parseValue(self, type_: ProjectType) -> Optional[Any]:
44  if self.value is None:
45  return None
46 
47  return [
48  _parseSingleDataset(datasetValue, type_)
49  for datasetValue in self.value
50  ]