63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
from packaging import version
|
|
|
|
import netbox_agent.dmidecode as dmidecode
|
|
from netbox_agent.config import config
|
|
from netbox_agent.config import netbox_instance as nb
|
|
from netbox_agent.logging import logging # NOQA
|
|
from netbox_agent.vendors.dell import DellHost
|
|
from netbox_agent.vendors.generic import GenericHost
|
|
from netbox_agent.vendors.hp import HPHost
|
|
from netbox_agent.vendors.qct import QCTHost
|
|
from netbox_agent.vendors.supermicro import SupermicroHost
|
|
from netbox_agent.virtualmachine import VirtualMachine, is_vm
|
|
|
|
MANUFACTURERS = {
|
|
"Dell Inc.": DellHost,
|
|
"HP": HPHost,
|
|
"HPE": HPHost,
|
|
"Supermicro": SupermicroHost,
|
|
"Quanta Cloud Technology Inc.": QCTHost,
|
|
"Generic": GenericHost,
|
|
}
|
|
|
|
|
|
def run(config):
|
|
dmi = dmidecode.parse()
|
|
|
|
if config.virtual.enabled or is_vm(dmi):
|
|
if not config.virtual.cluster_name:
|
|
raise Exception(
|
|
"virtual.cluster_name parameter is mandatory because it's a VM"
|
|
)
|
|
server = VirtualMachine(dmi=dmi)
|
|
else:
|
|
manufacturer = dmidecode.get_by_type(dmi, "Chassis")[0].get("Manufacturer")
|
|
try:
|
|
server = MANUFACTURERS[manufacturer](dmi=dmi)
|
|
except KeyError:
|
|
server = GenericHost(dmi=dmi)
|
|
|
|
if version.parse(nb.version) < version.parse("3.7"):
|
|
print("netbox-agent is not compatible with Netbox prior to version 3.7")
|
|
return False
|
|
|
|
if (
|
|
config.register
|
|
or config.update_all
|
|
or config.update_network
|
|
or config.update_location
|
|
or config.update_inventory
|
|
or config.update_psu
|
|
):
|
|
server.netbox_create_or_update(config)
|
|
if config.debug:
|
|
server.print_debug()
|
|
return True
|
|
|
|
|
|
def main():
|
|
return 0 if run(config) else 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|