report vendor on hp raid

netbox_agent/raid/hp.py

@@ -2,6 +2,7 @@ import re
 import subprocess
 
 from netbox_agent.raid.base import Raid, RaidController
+from netbox_agent.misc import get_vendor
 
 REGEXP_CONTROLLER_HP = re.compile(r'Smart Array ([a-zA-Z0-9- ]+) in Slot ([0-9]+)')
 
@@ -120,12 +121,18 @@ class HPRaidController(RaidController):
         key = next(iter(info_dict))
         for array, physical_disk in info_dict[key].items():
             for _, pd_attr in physical_disk.items():
+                model = pd_attr.get('Model', '').strip()
+                vendor = get_vendor(model)
+                if len(model.split()) > 1:
+                    vendor = get_vendor(model.split()[1])
                 ret.append({
-                    'Model': pd_attr.get('Model', '').strip(),
+                    'Model': model,
+                    'Vendor': vendor,
                     'SN': pd_attr.get('Serial Number', '').strip(),
                     'Size': pd_attr.get('Size', '').strip(),
                     'Type': 'SSD' if pd_attr.get('Interface Type') == 'Solid State SATA'
                     else 'HDD',
+                    '_src': self.__class__.__name__,
                 })
         return ret
 

netbox_agent/raid/storcli.py

@@ -46,6 +46,7 @@ class StorcliController(RaidController):
                 'SN': drive_attr.get('SN', '').strip(),
                 'Size': size,
                 'Type': media_type,
+                '_src': self.__class__.__name__,
                 })
         return ret