41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
import argparse
|
|
|
|
from netbox_agent.dell.dell import DellHost
|
|
import netbox_agent.dmidecode as dmidecode
|
|
from netbox_agent.hp.hp import HPHost
|
|
from netbox_agent.logging import logging # NOQA
|
|
|
|
MANUFACTURERS = {
|
|
'Dell Inc.': DellHost,
|
|
'HP': HPHost,
|
|
'HPE': HPHost,
|
|
}
|
|
|
|
|
|
def run(args):
|
|
manufacturer = dmidecode.get_by_type('Chassis')[0].get('Manufacturer')
|
|
server = MANUFACTURERS[manufacturer](dmidecode)
|
|
if args.debug:
|
|
server.print_debug()
|
|
if args.register:
|
|
server.netbox_create()
|
|
if args.update:
|
|
server.netbox_update()
|
|
return True
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='Netbox agent command line')
|
|
parser.add_argument('-r', '--register', action='store_true',
|
|
help='Register server in Netbox')
|
|
parser.add_argument('-u', '--update', action='store_true',
|
|
help='Update server in Netbox')
|
|
parser.add_argument('-d', '--debug', action='store_true',
|
|
help='Print debug informations')
|
|
|
|
args = parser.parse_args()
|
|
return run(args)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|