You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

62 lines
1.7 KiB

from specs_parser import SpecsParser
from config import etcd_client as client
specs_parser = SpecsParser(exceptional_devices=["cpu"])
class Field(object):
def __init__(self, _name, _type, _value=None):
self.name = _name
self.value = _value
self.type = _type
self.__errors = []
def validation(self):
return True
def is_valid(self):
if self.value == KeyError:
self.add_error(f"'{self.name}' field is a required field")
else:
if not isinstance(self.value, self.type):
self.add_error(f"Incorrect Type for '{self.name}' field")
else:
self.validation()
if self.__errors:
return False
return True
def get_errors(self):
return self.__errors
def add_error(self, error):
self.__errors.append(error)
class VmUUIDField(Field):
def __init__(self, data):
self.uuid = data.get("uuid", KeyError)
super().__init__("uuid", str, self.uuid)
self.validation = self.vm_uuid_validation
def vm_uuid_validation(self):
r = client.get(f"/v1/vm/{self.uuid}")
if not r:
self.add_error(f"VM with uuid {self.uuid} does not exists")
class SpecsField(Field):
def __init__(self, data):
self.specs = data.get("specs", KeyError)
super().__init__("specs", dict, self.specs)
self.validation = self.specs_validation
def specs_validation(self):
if not specs_parser.transform_specs(self.specs):
self.add_error("Invalid unit - "
f"Please use following units {specs_parser.get_allowed_units()}")