summaryrefslogtreecommitdiffstats
path: root/scripts/python/libvirt-xml2netbox-csv.py
blob: f9ba86682f731de4c788f7b6f05bdaab961f8411 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
#!/usr/bin/python3
import os
import xml.etree.ElementTree as xmlet
import pandas
import math

outfile = 'xxx.csv'

columns = [ "name", "status", "role", "cluster", "tenant", "platform", "vcpus", "memory", "disk", "comments" ]
rows = []

domaindir = 'xmls/domains'
diskdir = 'xmls/disks'

status = 'Active'
cluster = 'xxx'
tenant = 'xxx'
comment = 'Imported from libvirt. Manual verification pending.'

for domainxml in os.listdir(domaindir):
    domainparse = xmlet.parse(domaindir + "/" + domainxml)
    domainroot = domainparse.getroot()

    name = domainroot.find("name").text
    vcpus = domainroot.find("vcpu").text
    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)))))))
    
    while True:
        role_choice = input ("Assign a role for " + name + ":\n1) Internal Client\n2) Internal Server\n3) ???\n4) ???\n> ")
        if role_choice == "1":
            role = "Virtual Machine (Internal, Client)"
            break
        if role_choice == "2":
            role = "Virtual Machine (Internal, Server)"
            break
        if role_choice == "3":
            role = "Virtual Machine (Customer)"
            break
        if role_choice not in ["1", "2", "3"]:
            print("Invalid choice.")

    while True:
        platform_choice = input ("Assign platform for " + name + ":\n 1) openSUSE-x86_64\n2)OpenBSD-x86_64\n3)FreeBSD-x86_64\n> ")
        if platform_choice == "1":
            platform = "openSUSE-x86_64"
            break
        if platform_choice == "2":
            platform = "OpenBSD-x86_64"
            break
        if platform_choice == "3":
            platform = "FreeBSD-x86_64"
            break
        if platform_choice not in ["1", "2", "3"]:
            print("Invalid choice.")

    rows.append(
            {
                "name": name,
                "status": status,
                "role": role,
                "cluster": cluster,
                "tenant": tenant,
                "platform": platform,
                "vcpus": vcpus,
                "memory": memorysize,
                "disk": disksize,
                "comments": comment
            }
    )

convert = pandas.DataFrame(rows, columns=columns)
convert.to_csv(outfile, index=False)