Auto-update: Thu Aug 8 18:08:09 PDT 2024
This commit is contained in:
parent
95441ae20e
commit
9ec5a15d2f
1 changed files with 7 additions and 3 deletions
|
@ -130,6 +130,7 @@ class Configuration(BaseModel):
|
|||
err(f"Error loading configuration: {str(e)}")
|
||||
raise
|
||||
|
||||
|
||||
@classmethod
|
||||
def _resolve_path(cls, path: Union[str, Path], default_dir: str) -> Path:
|
||||
base_path = Path(__file__).parent.parent
|
||||
|
@ -140,6 +141,7 @@ class Configuration(BaseModel):
|
|||
path = base_path / path
|
||||
return path
|
||||
|
||||
|
||||
def resolve_placeholders(self, data: Any, secrets_data: Dict[str, Any]) -> Any:
|
||||
if isinstance(data, dict):
|
||||
resolved_data = {k: self.resolve_placeholders(v, secrets_data) for k, v in data.items()}
|
||||
|
@ -153,10 +155,11 @@ class Configuration(BaseModel):
|
|||
elif isinstance(data, list):
|
||||
return [self.resolve_placeholders(v, secrets_data) for v in data]
|
||||
elif isinstance(data, str):
|
||||
return self.resolve_string_placeholders(data, secrets_data, self.HOME)
|
||||
return self.resolve_string_placeholders(data, secrets_data, Path(self.HOME))
|
||||
else:
|
||||
return data
|
||||
|
||||
|
||||
def resolve_string_placeholders(self, value: str, secrets_data: Dict[str, Any], home_dir: Path) -> Any:
|
||||
pattern = r'\{\{\s*([^}]+)\s*\}\}'
|
||||
matches = re.findall(pattern, value)
|
||||
|
@ -182,7 +185,8 @@ class Configuration(BaseModel):
|
|||
if isinstance(value, str) and (value.startswith(('/', '~')) or (':' in value and value[1] == ':')):
|
||||
return Path(value).expanduser()
|
||||
return value
|
||||
|
||||
|
||||
|
||||
@classmethod
|
||||
def create_dynamic_model(cls, **data):
|
||||
for key, value in data.items():
|
||||
|
@ -198,13 +202,13 @@ class Configuration(BaseModel):
|
|||
)
|
||||
return DynamicModel(**data)
|
||||
|
||||
|
||||
class Config:
|
||||
extra = "allow"
|
||||
arbitrary_types_allowed = True
|
||||
|
||||
|
||||
|
||||
|
||||
class DirConfig(BaseModel):
|
||||
HOME: Path = Path.home()
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue