diff options
Diffstat (limited to 'scripts')
| -rwxr-xr-x | scripts/python/libvirt-xml2netbox-csv.py | 70 |
1 files changed, 50 insertions, 20 deletions
diff --git a/scripts/python/libvirt-xml2netbox-csv.py b/scripts/python/libvirt-xml2netbox-csv.py index f9ba866..bbdb101 100755 --- a/scripts/python/libvirt-xml2netbox-csv.py +++ b/scripts/python/libvirt-xml2netbox-csv.py @@ -1,21 +1,27 @@ #!/usr/bin/python3 +""" +Takes a directory with libvirt XML domain dumps and creates a NetBox importable CSV. Since the role and platform fields cannot be fed from the XML files, the user is given the option to pick those for each VM. + +Created and Last modified: 02/01/2022 by Georg Pfuetzenreuter <georg@lysergic.dev> +""" import os import xml.etree.ElementTree as xmlet import pandas import math -outfile = 'xxx.csv' +cluster = 'xxx' +tenant = 'xxx' + +outfile = cluster + '.csv' columns = [ "name", "status", "role", "cluster", "tenant", "platform", "vcpus", "memory", "disk", "comments" ] rows = [] -domaindir = 'xmls/domains' -diskdir = 'xmls/disks' +domaindir = 'xmls/' + cluster + '/domains' +diskdir = 'xmls/' + cluster + '/disks' -status = 'Active' -cluster = 'xxx' -tenant = 'xxx' -comment = 'Imported from libvirt. Manual verification pending.' +status = 'active' +comment = 'Imported from libvirt.' for domainxml in os.listdir(domaindir): domainparse = xmlet.parse(domaindir + "/" + domainxml) @@ -26,13 +32,19 @@ for domainxml in os.listdir(domaindir): memory = int(domainroot.find("memory").text) memorysize = round(memory*0.001024) diskxml = diskdir + "/" + name + ".disk.export.xml" - diskparse = xmlet.parse(diskxml) - diskroot = diskparse.getroot() - diskcapacity = int(diskroot.find("capacity").text) - disksize = round(diskcapacity / (math.pow(1024, (int(math.floor(math.log(diskcapacity, 1024))))))) + if os.path.exists(diskxml): + diskparse = xmlet.parse(diskxml) + diskroot = diskparse.getroot() + diskcapacity = int(diskroot.find("capacity").text) + disksize = round(diskcapacity / (math.pow(1024, (int(math.floor(math.log(diskcapacity, 1024))))))) + if not os.path.exists(diskxml): + print("No disk XML for " + name + ", assuming there is no VHD.") + disksize = "" while True: - role_choice = input ("Assign a role for " + name + ":\n1) Internal Client\n2) Internal Server\n3) ???\n4) ???\n> ") + role_choice = input ("Assign role for " + name + ":\n1) Internal Client\n2) Internal Server\n3) Public Client\n4) Public Server\n5) Customer\n6) Router\n7) Null\n> ") + if role_choice not in ["1", "2", "3", "4", "5", "6", "7"]: + print("Invalid choice.") if role_choice == "1": role = "Virtual Machine (Internal, Client)" break @@ -40,15 +52,27 @@ for domainxml in os.listdir(domaindir): role = "Virtual Machine (Internal, Server)" break if role_choice == "3": + role = "Virtual Machine (Public, Client)" + break + if role_choice == "4": + role = "Virtual Machine (Public, Server)" + break + if role_choice == "5": role = "Virtual Machine (Customer)" break - if role_choice not in ["1", "2", "3"]: - print("Invalid choice.") + if role_choice == "6": + role = "Virtual Machine (Router)" + break + if role_choice == "7": + role = "" + break while True: - platform_choice = input ("Assign platform for " + name + ":\n 1) openSUSE-x86_64\n2)OpenBSD-x86_64\n3)FreeBSD-x86_64\n> ") + platform_choice = input ("Assign platform for " + name + ":\n1) openSUSE Leap x86_64\n2) OpenBSD x86_64\n3) FreeBSD x86_64\n4) OPNsense x86_64\n5) Arch Linux x86_64\n7) Null\n> ") + if platform_choice not in ["1", "2", "3", "4", "5", "7"]: + print("Invalid choice.") if platform_choice == "1": - platform = "openSUSE-x86_64" + platform = "openSUSE-Leap-x86_64" break if platform_choice == "2": platform = "OpenBSD-x86_64" @@ -56,9 +80,16 @@ for domainxml in os.listdir(domaindir): if platform_choice == "3": platform = "FreeBSD-x86_64" break - if platform_choice not in ["1", "2", "3"]: - print("Invalid choice.") - + if platform_choice == "4": + platform = "OPNsense-x86_64" + break + if platform_choice == "5": + platform = "ArchLinux-x86_64" + break + if platform_choice == "7": + platform = "" + break + rows.append( { "name": name, @@ -76,4 +107,3 @@ for domainxml in os.listdir(domaindir): convert = pandas.DataFrame(rows, columns=columns) convert.to_csv(outfile, index=False) - |