Spaces:
Runtime error
Runtime error
File size: 2,383 Bytes
b115d50 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
import json
from enum import Enum
from pathlib import Path
from typing import Dict
from steamship.base.error import SteamshipError
from steamship.base.model import CamelModel
from steamship.data.manifest import ConfigParameter, ConfigParameterType
class Config(CamelModel):
"""Base class Steamship Package and Plugin configuration objects."""
def __init__(self, **kwargs):
kwargs = {k: v for k, v in kwargs.items() if v is not None}
super().__init__(**kwargs)
def extend_with_dict(self, d: dict, overwrite: bool = False):
"""Sets the attributes on this object with provided keys and values."""
for key, val in (d or {}).items():
if hasattr(self, key) and (overwrite or getattr(self, key) is None):
setattr(self, key, val)
def extend_with_json_file(
self, path: Path, overwrite: bool = False, fail_on_missing_file: bool = True
):
"""Extends this config object's values with a JSON file from disk.
This is useful for applying late-bound defaults, such as API keys added to a deployment bundle."""
if not path.exists():
if fail_on_missing_file:
raise SteamshipError(
message=f"Attempted to extend Config object with {path}, but the file was not found."
)
return
with open(path) as f:
data = json.load(f)
if not isinstance(data, dict):
raise SteamshipError(
message=f"Attempted to extend Config object with {path}, but the file did not contain a JSON `dict` object."
)
self.extend_with_dict(data, overwrite)
@staticmethod
def strip_enum(default_value):
if issubclass(type(default_value), Enum):
return default_value.value
else:
return default_value
@classmethod
def get_config_parameters(cls) -> Dict[str, ConfigParameter]:
result = {}
for field_name, field in cls.__fields__.items():
description = field.field_info.description
type_ = ConfigParameterType.from_python_type(field.type_)
result[field_name] = ConfigParameter(
type=type_,
default=cls.strip_enum(field.default),
description=description,
)
return result
|