vpil-openstack/setup-nodes.py at master · jhobbs/vpil-openstack · GitHub

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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
#!/usr/bin/env python3
import argparse
import json
from subprocess import check_output


def cmd(cmd_string, parse=True):
    maas_cmd = "maas root " + cmd_string
    print(maas_cmd)
    output = check_output(maas_cmd.split(" ")).decode("utf-8")
    if parse:
       return json.loads(output)
    else:
       return output


def get_nodes(excluded=None):
    if excluded is None:
        excluded = []
    node_list = cmd("nodes read")
    for node in node_list:
        if 'service_set' in node:
            continue
        if node['system_id'] in excluded:
            continue
        yield node


def clear_partitions(system_id, blockdevice):
    print("clearing partitions from %s" % (blockdevice['id_path']))
    for partition in blockdevice['partitions']:
        cmd("partition delete %s %s %s" % (
            system_id, blockdevice['id'], partition['id']), parse=False)
    if blockdevice['filesystem'] is not None:
        cmd("block-device unmount %s %s" % (system_id, blockdevice['id']))
        cmd("block-device unformat %s %s" % (system_id, blockdevice['id']))


def create_partition(system_id, blockdevice_id, size, fstype,
                     label, mount_point):
    results = cmd("partitions create %s %s size=%s" % (
        system_id, blockdevice_id, size))
    partition_id = results['id']
    cmd("partition format %s %s %s fstype=%s label=%s" % (
        system_id, blockdevice_id, partition_id, fstype, label))
    cmd("partition mount %s %s %s mount_point=%s" % (
        system_id, blockdevice_id, partition_id, mount_point))


def setup_root_blockdevice(system_id, blockdevice_id):
    cmd(
        "block-device set-boot-disk %s %s" % (system_id, blockdevice_id),
        parse=False)
    # just using the size that maas used automatically on the nvme device.
    create_partition(
        system_id, blockdevice_id, size=536870912,
        fstype='fat32', label='efi', mount_point='/boot/efi')
    # just hard coding the known left over side, could read from maas.
    create_partition(
        system_id, blockdevice_id, size=999662026752,
        fstype='ext4', label='root', mount_point='/')


def setup_storage(node):
    for blockdevice in node['blockdevice_set']:
        clear_partitions(node['system_id'], blockdevice)
        if blockdevice['path'] == '/dev/disk/by-dname/sda':
            setup_root_blockdevice(node['system_id'], blockdevice['id'])


def setup_networking(node):
    bond_children_ids = []
    for interface in node['interface_set']:
        if interface['type'] == 'bond':
            print("Found a bond on %s" % (node['system_id']))
            cmd("interface delete %s %s" % (node['system_id'], interface['id']),
                False)
        if interface['name'] in ['eno49', 'eno50']:
            bond_children_ids.append(interface['id'])
    cmd("interfaces create-bond %s parents=%s parents=%s bond_mode=802.3ad name=bond0 bond_xmit_hash_policy=layer2+3" % (
        node['system_id'], bond_children_ids[0], bond_children_ids[1]))


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--exclude",
        help="List of system IDs to exclude.",
        default=None, nargs="+")
    args = parser.parse_args()

    for node in get_nodes(args.exclude):
        setup_networking(node)
        setup_storage(node)


if __name__ == '__main__':
    main()