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.dmi.get_by_type('Baseboard')[0].keys()

    def get_blade_slot(self):
        if self.is_blade():
            return 'Slot {}'.format(
                self.dmi.get_by_type('Baseboard')[0].get('Location In Chassis')
            )
        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.dmi.get_by_type('Chassis')[0]['Version']
        return self.get_product_name()

    def get_chassis_service_tag(self):
        if self.is_blade():
            return self.dmi.get_by_type('Chassis')[0]['Serial Number']
        return self.get_service_tag()