fc1dc13678
This commit added feature for test bootstrap failure and resume functionality Change-Id: I6a600a4c8acdc37a643e0d38ea02c93fc41e503b Signed-off-by: Abhishek jaiswal <abhishek.jaiswal@windriver.com>
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
import yaml
|
|
|
|
|
|
class DeploymentAssetsHandler:
|
|
"""
|
|
A class for processing YAML files, including reading, writing, and updating values.
|
|
"""
|
|
|
|
def __init__(self, local_file_yaml: str):
|
|
"""Initializes the YamlProcessor.
|
|
|
|
Args:
|
|
local_file_yaml (str): yaml file path
|
|
"""
|
|
self.local_file = local_file_yaml
|
|
self.data = self._read_yaml()
|
|
|
|
def _read_yaml(self) -> None:
|
|
"""Reads the YAML file from the local file."""
|
|
try:
|
|
with open(self.local_file, "r") as local_file_st:
|
|
data = yaml.safe_load(local_file_st)
|
|
except FileNotFoundError:
|
|
raise FileNotFoundError(f"File not found locally: {self.local_file}")
|
|
except yaml.YAMLError as e:
|
|
raise yaml.YAMLError(f"Error parsing YAML from local file: {e}")
|
|
return data
|
|
|
|
def write_yaml(self):
|
|
"""Write the YAML to the local file."""
|
|
try:
|
|
# Write to local file
|
|
with open(self.local_file, "w") as local_file_st:
|
|
yaml.dump(self.data, local_file_st, indent=2)
|
|
except Exception as e:
|
|
raise Exception(f"Error writing YAML file: {e}")
|
|
|
|
def inject_wrong_bootstrap_value(self):
|
|
"""Function to change docker registry value"""
|
|
# inject wrong values in Bootstrap values
|
|
self.data["docker_registries"]["k8s.gcr.io"]["url"] = "registry.central:9001/k8s.gcr.io.wrong"
|
|
self.write_yaml()
|