Skip to content

Commit

Permalink
fix: compliance with existing convention
Browse files Browse the repository at this point in the history
  • Loading branch information
leclairm committed Oct 28, 2024
1 parent 7f154f4 commit 3070234
Showing 1 changed file with 10 additions and 7 deletions.
17 changes: 10 additions & 7 deletions src/wcflow/parsing/_yaml_data_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -114,14 +114,15 @@ def convert_to_struct_time(cls, value: str | None) -> time.struct_time | None:
return None if value is None else time.strptime(value, "%H:%M:%S")


class ConfigData(_NamedBaseModel):
class _DataBaseModel(_NamedBaseModel):
"""
To create an instance of a data defined in a workflow file.
"""

type: str
src: str
format: str | None = None
available: bool = False

@field_validator("type")
@classmethod
Expand All @@ -133,15 +134,17 @@ def is_file_or_dir(cls, value: str) -> str:
return value


class ConfigAvailableData(ConfigData):
available: ClassVar[bool] = True
class ConfigAvailableData(_DataBaseModel):

available: bool = True

class ConfigGeneratedData(ConfigData):
available: ClassVar[bool] = False

class ConfigGeneratedData(_DataBaseModel):

class ConfigDataStore(BaseModel):
available: bool = False


class ConfigData(BaseModel):

available: list[ConfigAvailableData]
generated: list[ConfigGeneratedData]
Expand Down Expand Up @@ -271,7 +274,7 @@ class ConfigWorkflow(BaseModel):
end_date: datetime
cycles: list[ConfigCycle]
tasks: list[ConfigTask]
data: ConfigDataStore
data: ConfigData
data_dict: dict = {}
task_dict: dict = {}

Expand Down

0 comments on commit 3070234

Please sign in to comment.