class DatacenterValidator:
def __init__(self):
self.dc_table = DatacenterTable()
self.rack_table = RackTable()
def create_dc_validation(self, datacenter):
if len(datacenter.abbreviation) > 6:
return "The datacenter abbreviation must be 6 characters or less."
abbrev_pattern = re.compile("[a-zA-Z]+[a-zA-Z0-9]*")
if abbrev_pattern.fullmatch(datacenter.abbreviation) is None:
return "Datacenter abbreviation must start with a letter and only contain letters and numbers."
name_pattern = re.compile("[a-zA-Z]*[ a-zA-Z0-9-]*[A-Za-z0-9]")
if name_pattern.fullmatch(datacenter.name) is None:
return "Datacenter name must start with a letter and only contain letters, numbers, spaces, and hyphens and end in a letter or number."
if (self.dc_table.get_datacenter_id_by_abbrev(datacenter.abbreviation)
is not None):
return "A datacenter with the given abbreviation already exists. Please choose a different abbreviation."
if self.dc_table.get_datacenter_id_by_name(
datacenter.name) is not None:
return "A datacenter with the given name already exists. Please choose a different name."
return Constants.API_SUCCESS
def edit_dc_validation(self, datacenter, original_name):
if len(datacenter.abbreviation) > 6:
return "The datacenter abbreviation must be 6 characters or less."
abbrev_pattern = re.compile("[a-zA-Z]+[a-zA-Z0-9]*")
if abbrev_pattern.fullmatch(datacenter.abbreviation) is None:
return "Datacenter abbreviation must start with a letter and only contain letters and numbers."
name_pattern = re.compile("[a-zA-Z]*[ a-zA-Z0-9-]*[A-Za-z0-9]")
if name_pattern.fullmatch(datacenter.name) is None:
return "Datacenter name must start with a letter and only contain letters, numbers, spaces, and hyphens and end in a letter or number."
orig_id = self.dc_table.get_datacenter_id_by_name(original_name)
abbrev_id = self.dc_table.get_datacenter_id_by_abbrev(
datacenter.abbreviation)
if abbrev_id is not None and abbrev_id != orig_id:
return "A datacenter with the given abbreviation already exists. Please choose a different abbreviation."
name_id = self.dc_table.get_datacenter_id_by_name(datacenter.name)
if name_id is not None and name_id != orig_id:
return "A datacenter with the given name already exists. Please choose a different name."
return Constants.API_SUCCESS
def delete_dc_validation(self, dc_name):
dc_id = self.dc_table.get_datacenter_id_by_name(dc_name)
dc_racks = self.rack_table.get_rack_by_datacenter(dc_id)
if dc_racks is not None and dc_racks != []:
return (
"All racks in the datacenter must be removed before it can be deleted."
)
return Constants.API_SUCCESS
class StatsManager:
def __init__(self):
self.dc_table = DatacenterTable()
self.instance_table = InstanceTable()
self.model_table = ModelTable()
self.rack_table = RackTable()
self.space_by_rack = {}
self.space_by_vendor = {}
self.space_by_model = {}
self.space_by_owner = {}
self.rack_count = {}
self.rack_height = 42
def create_report(self, dc_name):
self.reset_counters()
if dc_name is None or dc_name == "":
rack_list = self.rack_table.get_all_racks()
else:
dc_id = self.dc_table.get_datacenter_id_by_name(dc_name)
rack_list = self.rack_table.get_rack_by_datacenter(dc_id)
num_racks = len(rack_list)
if num_racks == 0:
raise ValueError(
"Reports require existing racks. Please ensure that the datacenter exists and contains racks."
)
for rack in rack_list:
instance_list = self.instance_table.get_instances_by_rack(
rack.label, rack.datacenter_id)
self.iterate_instance(instance_list, rack.label)
total_space_used = 0
for key in self.space_by_rack:
total_space_used += self.space_by_rack[key]
self.space_by_rack[key] = round(
(self.space_by_rack[key] /
(self.rack_count[key] * self.rack_height)) * 100,
2,
)
all_space = num_racks * self.rack_height
percent_total_used = round((total_space_used / all_space) * 100, 2)
percent_total_free = 100 - percent_total_used
self.space_by_vendor = self.divide_dict_by_space(
self.space_by_vendor, all_space)
self.space_by_model = self.divide_dict_by_space(
self.space_by_model, all_space)
self.space_by_owner = self.divide_dict_by_space(
self.space_by_owner, all_space)
rack_positionsage_json = json.dumps(self.space_by_rack, sort_keys=True)
vendor_usage_json = json.dumps(self.space_by_vendor, sort_keys=True)
model_usage_json = json.dumps(self.space_by_model, sort_keys=True)
owner_usage_json = json.dumps(self.space_by_owner, sort_keys=True)
returnJSON = {
"totalUsage": percent_total_used,
"totalFree": percent_total_free,
"spaceUsage": rack_positionsage_json,
"vendorUsage": vendor_usage_json,
"modelUsage": model_usage_json,
"ownerUsage": owner_usage_json,
}
return returnJSON
def iterate_instance(self, instance_list, rack_label):
rack_space_used = 0
for instance in instance_list:
model = self.model_table.get_model(instance.model_id)
rack_space_used += model.height
if model.vendor in self.space_by_vendor:
self.space_by_vendor[model.vendor] += model.height
else:
self.space_by_vendor[model.vendor] = model.height
model_name = model.vendor + " " + model.model_number
if model_name in self.space_by_model:
self.space_by_model[model_name] += model.height
else:
self.space_by_model[model_name] = model.height
if instance.owner is None or instance.owner == "":
owner = "No owner listed"
else:
owner = instance.owner
if owner in self.space_by_owner:
self.space_by_owner[owner] += model.height
else:
self.space_by_owner[owner] = model.height
if rack_label in self.rack_count:
self.space_by_rack[rack_label] += rack_space_used
self.rack_count[rack_label] += 1
else:
self.space_by_rack[rack_label] = rack_space_used
self.rack_count[rack_label] = 1
def divide_dict_by_space(self, dictionary, total_space_used):
for key in dictionary:
dictionary[key] = round((dictionary[key] / total_space_used) * 100,
2)
return dictionary
def reset_counters(self):
self.space_by_rack = {}
self.space_by_vendor = {}
self.space_by_model = {}
self.space_by_owner = {}
self.rack_count = {}
return