mirror of
https://github.com/bb-Ricardo/netbox-sync.git
synced 2026-01-20 07:50:16 -06:00
155 lines
4.3 KiB
Python
Executable File
155 lines
4.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
self_description = \
|
|
"""
|
|
Sync objects from various sources to Netbox
|
|
"""
|
|
|
|
|
|
from datetime import datetime
|
|
|
|
from module.common.misc import grab, get_relative_time
|
|
from module.common.cli_parser import parse_command_line
|
|
from module.common.logging import setup_logging
|
|
from module.common.configuration import get_config_file, open_config_file, get_config
|
|
from module.netbox.connection import NetBoxHandler
|
|
from module.netbox.inventory import NetBoxInventory
|
|
from module.netbox.object_classes import *
|
|
|
|
from module.sources import *
|
|
|
|
|
|
import pprint
|
|
|
|
__version__ = "0.0.1"
|
|
__version_date__ = "2020-10-01"
|
|
__author__ = "Ricardo Bartels <ricardo.bartels@telekom.de>"
|
|
__description__ = "NetBox Sync"
|
|
__license__ = "MIT"
|
|
__url__ = "https://github.com/bb-ricardo/netbox-sync"
|
|
|
|
|
|
default_log_level = "INFO"
|
|
default_config_file_path = "./settings.ini"
|
|
|
|
|
|
"""
|
|
ToDo:
|
|
* host "Management" interface is Primary
|
|
* return more then one object if found more then one and add somehow to returned objects. Maybe related?
|
|
* Add purge option
|
|
"""
|
|
|
|
def main():
|
|
|
|
start_time = datetime.now()
|
|
|
|
sources = list()
|
|
|
|
# parse command line
|
|
args = parse_command_line(self_description=self_description,
|
|
version=__version__,
|
|
version_date=__version_date__,
|
|
default_config_file_path=default_config_file_path)
|
|
|
|
# get config file path
|
|
config_file = get_config_file(args.config_file)
|
|
|
|
# get config handler
|
|
config_handler = open_config_file(config_file)
|
|
|
|
# get logging configuration
|
|
|
|
# set log level
|
|
log_level = default_log_level
|
|
# config overwrites default
|
|
log_level = config_handler.get("common", "log_level", fallback=log_level)
|
|
# cli option overwrites config file
|
|
log_level = grab(args, "log_level", fallback=log_level)
|
|
|
|
log_file = None
|
|
if bool(config_handler.getboolean("common", "log_to_file", fallback=False)) is True:
|
|
log_file = config_handler.get("common", "log_file", fallback=None)
|
|
|
|
# setup logging
|
|
log = setup_logging(log_level, log_file)
|
|
|
|
|
|
# now we are ready to go
|
|
log.info("Starting " + __description__)
|
|
log.debug(f"Using config file: {config_file}")
|
|
|
|
# initialize an empty inventory which will be used to hold and reference all objects
|
|
inventory = NetBoxInventory()
|
|
|
|
# get config for netbox handler
|
|
netbox_settings = get_config(config_handler, section="netbox", valid_settings=NetBoxHandler.settings)
|
|
|
|
# establish NetBox connection
|
|
NB_handler = NetBoxHandler(cli_args=args, settings=netbox_settings, inventory=inventory)
|
|
|
|
# instantiate source handlers and get attributes
|
|
log.info("Initializing sources")
|
|
sources = instanciate_sources(config_handler, inventory)
|
|
|
|
# all sources are unavailable
|
|
if len(sources) == 0:
|
|
do_error_exit("No working sources found. Exit.")
|
|
|
|
# retrieve all dependent object classes
|
|
netbox_objects_to_query = list()
|
|
for source in sources:
|
|
netbox_objects_to_query.extend(source.dependend_netbox_objects)
|
|
|
|
# request NetBox data
|
|
log.info("Querying necessary objects from Netbox. This might take a while.")
|
|
NB_handler.query_current_data(list(set(netbox_objects_to_query)))
|
|
log.info("Finished querying necessary objects from Netbox")
|
|
|
|
# resolve object relations within the initial inventory
|
|
inventory.resolve_relations()
|
|
|
|
# initialize basic data needed for syncing
|
|
NB_handler.inizialize_basic_data()
|
|
|
|
# loop over sources and patch netbox data
|
|
for source in sources:
|
|
source.apply()
|
|
|
|
# add/remove tags to/from all inventory items
|
|
inventory.tag_all_the_things(sources, NB_handler)
|
|
|
|
# update all IP addresses
|
|
#inventory.update_all_ip_addresses()
|
|
|
|
#for object in inventory.get_all_items(NBVMs):
|
|
# print(object.get_display_name())
|
|
|
|
"""
|
|
for object in inventory.get_all_items(NBIPAddresses):
|
|
|
|
try:
|
|
print(object)
|
|
except Exception as e:
|
|
print(e)
|
|
pprint.pprint(object.__dict__)
|
|
exit(2)
|
|
|
|
|
|
exit(0)
|
|
|
|
inventory.set_primary_ips()
|
|
# Optional tasks
|
|
if settings.POPULATE_DNS_NAME:
|
|
nb.set_dns_names()
|
|
"""
|
|
|
|
# update data in NetBox
|
|
NB_handler.update_instance()
|
|
|
|
# finish
|
|
log.info("Completed NetBox Sync in %s" % get_relative_time(datetime.now() - start_time))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|