diff --git a/netbox_agent/misc.py b/netbox_agent/misc.py index 3ef9ff4..fe36410 100644 --- a/netbox_agent/misc.py +++ b/netbox_agent/misc.py @@ -9,13 +9,20 @@ def is_tool(name): def get_vendor(name): vendors = { 'ST': 'Seagate', - 'Crucial': 'Crucial', - 'Micron': 'Micron', - 'Intel': 'Intel', - 'Samsung': 'Samsung', + 'CRUCIAL': 'Crucial', + 'MICRON': 'Micron', + 'INTEL': 'Intel', + 'SAMSUNG': 'Samsung', + 'EH0': 'HP', 'HGST': 'HGST', + 'HUH': 'HGST', + 'MB': 'Toshiba', + 'MC': 'Toshiba', + 'MD': 'Toshiba', + 'MG': 'Toshiba', + 'WD': 'WDC' } for key, value in vendors.items(): - if name.startswith(key): + if name.upper().startswith(key): return value return name