qct.py

from netbox_agent.server import ServerBase


class QCTHost(ServerBase):
    def __init__(self, *args, **kwargs):
        super(QCTHost, self).__init__(*args, **kwargs)
        self.manufacturer = "QCT"

    def is_blade(self):
        return "Location In Chassis" in self.baseboard[0].keys()

    def get_blade_slot(self):
        if self.is_blade():
            return "Slot {}".format(self.baseboard[0].get("Location In Chassis").strip())
        return None

    def get_chassis_name(self):
        if not self.is_blade():
            return None
        return "Chassis {}".format(self.get_service_tag())

    def get_chassis(self):
        if self.is_blade():
            return self.chassis[0]["Version"].strip()
        return self.get_product_name()

    def get_chassis_service_tag(self):
        if self.is_blade():
            return self.chassis[0]["Serial Number"].strip()
        return self.get_service_tag()