diff options
Diffstat (limited to 'fuel/deploy/dea.py')
| -rw-r--r-- | fuel/deploy/dea.py | 42 |
1 files changed, 31 insertions, 11 deletions
diff --git a/fuel/deploy/dea.py b/fuel/deploy/dea.py index 0ab215d..5f306a2 100644 --- a/fuel/deploy/dea.py +++ b/fuel/deploy/dea.py @@ -1,16 +1,19 @@ import yaml +import io class DeploymentEnvironmentAdapter(object): def __init__(self): self.dea_struct = None - self.blade_ids = {} - self.blades = {} + self.blade_ids_per_shelves = {} + self.blades_per_shelves = {} self.shelf_ids = [] + self.networks = {} def parse_yaml(self, yaml_path): - with open(yaml_path) as yaml_file: + with io.open(yaml_path) as yaml_file: self.dea_struct = yaml.load(yaml_file) self.collect_shelf_and_blade_info() + self.collect_network_info() def get_no_of_blades(self): no_of_blades = 0 @@ -19,7 +22,13 @@ class DeploymentEnvironmentAdapter(object): return no_of_blades def get_server_type(self): - return self.dea_struct['server_type'] + return self.dea_struct['server']['type'] + + def get_server_info(self): + return (self.dea_struct['server']['type'], + self.dea_struct['server']['mgmt_ip'], + self.dea_struct['server']['username'], + self.dea_struct['server']['password']) def get_environment_name(self): return self.dea_struct['name'] @@ -27,17 +36,20 @@ class DeploymentEnvironmentAdapter(object): def get_shelf_ids(self): return self.shelf_ids - def get_blade_ids(self, shelf_id): - return self.blade_ids[shelf_id] + def get_blade_ids_per_shelf(self, shelf_id): + return self.blade_ids_per_shelves[shelf_id] + + def get_blade_ids_per_shelves(self): + return self.blade_ids_per_shelves def collect_shelf_and_blade_info(self): - self.blade_ids = {} - self.blades = {} + self.blade_ids_per_shelves = {} + self.blades_per_shelves = {} self.shelf_ids = [] for shelf in self.dea_struct['shelf']: self.shelf_ids.append(shelf['id']) - blade_ids = self.blade_ids[shelf['id']] = [] - blades = self.blades[shelf['id']] = {} + blade_ids = self.blade_ids_per_shelves[shelf['id']] = [] + blades = self.blades_per_shelves[shelf['id']] = {} for blade in shelf['blade']: blade_ids.append(blade['id']) blades[blade['id']] = blade @@ -49,4 +61,12 @@ class DeploymentEnvironmentAdapter(object): def is_compute_host(self, shelf_id, blade_id): blade = self.blades[shelf_id][blade_id] - return True if 'role' not in blade else False
\ No newline at end of file + return True if 'role' not in blade else False + + def collect_network_info(self): + self.networks = {} + for network in self.dea_struct['network']: + self.networks[network['name']] = network + + def get_networks(self): + return self.networks
\ No newline at end of file |