OutputStorage -> OutputStorageConfig

Author: sveinugu

src/omnipy/api/protocols/public/config.py

@@ -23,7 +23,7 @@ class IsPrefectEngineConfig(IsEngineConfig, Protocol):
 
 class IsJobConfig(Protocol):
     """"""
-    output_storage: 'IsOutputStorage'
+    output_storage: 'IsOutputStorageConfig'
 
 
 class IsDataConfig(Protocol):
@@ -49,24 +49,24 @@ class IsRootLogConfig(Protocol):
     file_log_path: str
 
 
-class IsOutputStorageBase(Protocol):
+class IsOutputStorageConfigBase(Protocol):
     persist_data_dir_path: str
 
 
-class IsLocalOutputStorage(IsOutputStorageBase, Protocol):
+class IsLocalOutputStorageConfig(IsOutputStorageConfigBase, Protocol):
     ...
 
 
-class IsS3OutputStorage(IsOutputStorageBase, Protocol):
+class IsS3OutputStorageConfig(IsOutputStorageConfigBase, Protocol):
     endpoint_url: str
     access_key: str
     secret_key: str
     bucket_name: str
 
 
-class IsOutputStorage(Protocol):
+class IsOutputStorageConfig(Protocol):
     persist_outputs: ConfigPersistOutputsOptions
     restore_outputs: ConfigRestoreOutputsOptions
     protocol: ConfigOutputStorageProtocolOptions
-    local: IsLocalOutputStorage
-    s3: IsS3OutputStorage
+    local: IsLocalOutputStorageConfig
+    s3: IsS3OutputStorageConfig

src/omnipy/config/job.py

@@ -5,22 +5,22 @@
 from omnipy.api.enums import (ConfigOutputStorageProtocolOptions,
                               ConfigPersistOutputsOptions,
                               ConfigRestoreOutputsOptions)
-from omnipy.api.protocols.public.config import (IsLocalOutputStorage,
-                                                IsOutputStorage,
-                                                IsS3OutputStorage)
+from omnipy.api.protocols.public.config import (IsLocalOutputStorageConfig,
+                                                IsOutputStorageConfig,
+                                                IsS3OutputStorageConfig)
 
 
 def _get_persist_data_dir_path() -> str:
     return str(Path.cwd().joinpath(Path('outputs')))
 
 
 @dataclass
-class LocalOutputStorage:
+class LocalOutputStorageConfig:
     persist_data_dir_path: str = field(default_factory=_get_persist_data_dir_path)
 
 
 @dataclass
-class S3OutputStorage:
+class S3OutputStorageConfig:
     persist_data_dir_path: str = os.path.join('omnipy', 'outputs')
     endpoint_url: str = ''
     bucket_name: str = ''
@@ -29,16 +29,16 @@ class S3OutputStorage:
 
 
 @dataclass
-class OutputStorage:
+class OutputStorageConfig:
     persist_outputs: ConfigPersistOutputsOptions = \
         ConfigPersistOutputsOptions.ENABLE_FLOW_AND_TASK_OUTPUTS
     restore_outputs: ConfigRestoreOutputsOptions = \
         ConfigRestoreOutputsOptions.DISABLED
     protocol: ConfigOutputStorageProtocolOptions = ConfigOutputStorageProtocolOptions.LOCAL
-    local: IsLocalOutputStorage = field(default_factory=LocalOutputStorage)
-    s3: IsS3OutputStorage = field(default_factory=S3OutputStorage)
+    local: IsLocalOutputStorageConfig = field(default_factory=LocalOutputStorageConfig)
+    s3: IsS3OutputStorageConfig = field(default_factory=S3OutputStorageConfig)
 
 
 @dataclass
 class JobConfig:
-    output_storage: IsOutputStorage = field(default_factory=OutputStorage)
+    output_storage: IsOutputStorageConfig = field(default_factory=OutputStorageConfig)