Spaces:
Runtime error
Runtime error
File size: 4,702 Bytes
105b369 |
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 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 |
from pathlib import Path
from typing import Optional, List, Any, Dict
from pydantic import BaseModel, ConfigDict
from phi.workspace.settings import WorkspaceSettings
class PhiBase(BaseModel):
name: Optional[str] = None
group: Optional[str] = None
version: Optional[str] = None
env: Optional[str] = None
enabled: bool = True
# -*- Resource Control
skip_create: bool = False
skip_read: bool = False
skip_update: bool = False
skip_delete: bool = False
recreate_on_update: bool = False
# Skip create if resource with the same name is active
use_cache: bool = True
# Force create/update/delete implementation
force: Optional[bool] = None
# -*- Debug Mode
debug_mode: bool = False
# -*- Resource Environment
# Add env variables to resource where applicable
env_vars: Optional[Dict[str, Any]] = None
# Read env from a file in yaml format
env_file: Optional[Path] = None
# Add secret variables to resource where applicable
# secrets_dict: Optional[Dict[str, Any]] = None
# Read secrets from a file in yaml format
secrets_file: Optional[Path] = None
# Read secret variables from AWS Secrets
aws_secrets: Optional[Any] = None
# -*- Waiter Control
wait_for_create: bool = True
wait_for_update: bool = True
wait_for_delete: bool = True
waiter_delay: int = 30
waiter_max_attempts: int = 50
# -*- Save to output directory
# If True, save output to json files
save_output: bool = False
# The directory for the input files in the workspace directory
input_dir: Optional[str] = None
# The directory for the output files in the workspace directory
output_dir: Optional[str] = None
# -*- Dependencies
depends_on: Optional[List[Any]] = None
# -*- Workspace Settings
workspace_settings: Optional[WorkspaceSettings] = None
# -*- Cached Data
cached_env_file_data: Optional[Dict[str, Any]] = None
cached_secret_file_data: Optional[Dict[str, Any]] = None
model_config = ConfigDict(arbitrary_types_allowed=True, populate_by_name=True)
def get_group_name(self) -> Optional[str]:
return self.group or self.name
@property
def workspace_root(self) -> Optional[Path]:
return self.workspace_settings.ws_root if self.workspace_settings is not None else None
@property
def workspace_name(self) -> Optional[str]:
return self.workspace_settings.ws_name if self.workspace_settings is not None else None
@property
def workspace_dir(self) -> Optional[Path]:
if self.workspace_root is not None:
workspace_dir = self.workspace_settings.workspace_dir if self.workspace_settings is not None else None
if workspace_dir is not None:
return self.workspace_root.joinpath(workspace_dir)
return None
def set_workspace_settings(self, workspace_settings: Optional[WorkspaceSettings] = None) -> None:
if workspace_settings is not None:
self.workspace_settings = workspace_settings
def get_env_file_data(self) -> Optional[Dict[str, Any]]:
if self.cached_env_file_data is None:
from phi.utils.yaml_io import read_yaml_file
self.cached_env_file_data = read_yaml_file(file_path=self.env_file)
return self.cached_env_file_data
def get_secret_file_data(self) -> Optional[Dict[str, Any]]:
if self.cached_secret_file_data is None:
from phi.utils.yaml_io import read_yaml_file
self.cached_secret_file_data = read_yaml_file(file_path=self.secrets_file)
return self.cached_secret_file_data
def get_secret_from_file(self, secret_name: str) -> Optional[str]:
secret_file_data = self.get_secret_file_data()
if secret_file_data is not None:
return secret_file_data.get(secret_name)
return None
def set_aws_env_vars(self, env_dict: Dict[str, str], aws_region: Optional[str] = None) -> None:
from phi.constants import (
AWS_REGION_ENV_VAR,
AWS_DEFAULT_REGION_ENV_VAR,
)
if aws_region is not None:
# logger.debug(f"Setting AWS Region to {aws_region}")
env_dict[AWS_REGION_ENV_VAR] = aws_region
env_dict[AWS_DEFAULT_REGION_ENV_VAR] = aws_region
elif self.workspace_settings is not None and self.workspace_settings.aws_region is not None:
# logger.debug(f"Setting AWS Region to {aws_region} using workspace_settings")
env_dict[AWS_REGION_ENV_VAR] = self.workspace_settings.aws_region
env_dict[AWS_DEFAULT_REGION_ENV_VAR] = self.workspace_settings.aws_region
|