96 lines
3.5 KiB
Python
96 lines
3.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# File: generate-systemd-vms.py
|
||
|
|
#
|
||
|
|
# Description: This file generates the systemd.service files that run our VM's
|
||
|
|
#
|
||
|
|
# Package: AniNIX/Ubiqtorate
|
||
|
|
# Copyright: WTFPL
|
||
|
|
#
|
||
|
|
# Author: DarkFeather <darkfeather@aninix.net>
|
||
|
|
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import sys
|
||
|
|
import yaml
|
||
|
|
|
||
|
|
filepath="roles/Node/files/vm-definitions/"
|
||
|
|
|
||
|
|
def WriteVMFile(content,hosttype,hostclass):
|
||
|
|
### Create the service files for the hosts
|
||
|
|
# param content: the yaml content to parse
|
||
|
|
# param hosttype: managed or unmanaged
|
||
|
|
# param hostclass: the type of host as classified in the yaml
|
||
|
|
|
||
|
|
global filepath
|
||
|
|
|
||
|
|
for host in content['all']['children'][hosttype]['children'][hostclass]['hosts']:
|
||
|
|
|
||
|
|
cores = 0
|
||
|
|
memory = 0
|
||
|
|
vnc = 0
|
||
|
|
disks = ''
|
||
|
|
mac = ''
|
||
|
|
bridge = ''
|
||
|
|
|
||
|
|
# Make sure the host definition has all the critera
|
||
|
|
try:
|
||
|
|
cores = str(content['all']['children'][hosttype]['children'][hostclass]['hosts'][host]['cores'])
|
||
|
|
memory = str(content['all']['children'][hosttype]['children'][hostclass]['hosts'][host]['memory'])
|
||
|
|
vnc = str(content['all']['children'][hosttype]['children'][hostclass]['hosts'][host]['vnc'])
|
||
|
|
disks = ' '.join(content['all']['children'][hosttype]['children'][hostclass]['hosts'][host]['disks'])
|
||
|
|
mac = content['all']['children'][hosttype]['children'][hostclass]['hosts'][host]['mac']
|
||
|
|
bridge = content['all']['children'][hosttype]['children'][hostclass]['hosts'][host]['bridge']
|
||
|
|
except Exception as e:
|
||
|
|
print('Host ' + host + " doesn't have the attributes needed to be a VM -- skipping.")
|
||
|
|
print(e)
|
||
|
|
1 == 1
|
||
|
|
|
||
|
|
# Write the file.
|
||
|
|
with open(filepath+host+'-vm.service','w') as vmfile:
|
||
|
|
vmfile.write('[Unit]\n')
|
||
|
|
vmfile.write('Description=AniNIX/' + host + '\n')
|
||
|
|
vmfile.write('After=network.target\n')
|
||
|
|
vmfile.write('\n')
|
||
|
|
vmfile.write('[Service]\n')
|
||
|
|
vmfile.write('ExecStart=/usr/sbin/qemu-system-x86_64 -name AniNIX/' + host + ' -machine type=pc,accel=kvm')
|
||
|
|
if 'uefi' in content['all']['children'][hosttype]['children'][hostclass]['hosts'][host].keys(): vmfile.write(' -bios /usr/share/edk2-ovmf/x64/OVMF.fd')
|
||
|
|
vmfile.write(' -cpu qemu64 -smp ' + cores + ' ' + disks + ' -net nic,macaddr=' + mac + ',model=virtio -net bridge,br=' + bridge + ' -vga std -nographic -vnc :' + str(vnc) + ' -m size=' + str(memory) + 'G -device virtio-rng-pci\n')
|
||
|
|
vmfile.write('ExecReload=/bin/kill -HUP $MAINPID\n')
|
||
|
|
vmfile.write('KillMode=process\n')
|
||
|
|
vmfile.write('Restart=always\n')
|
||
|
|
vmfile.write('User=root\n')
|
||
|
|
vmfile.write('Group=root\n')
|
||
|
|
vmfile.write('\n')
|
||
|
|
vmfile.write('[Install]\n')
|
||
|
|
vmfile.write('WantedBy=multi-user.target\n')
|
||
|
|
print(host+'-vm.service')
|
||
|
|
|
||
|
|
def GenerateFiles(file):
|
||
|
|
### Open the file and parse it
|
||
|
|
# param file: the file to work on
|
||
|
|
|
||
|
|
global filepath
|
||
|
|
|
||
|
|
try:
|
||
|
|
shutil.rmtree(filepath)
|
||
|
|
except:
|
||
|
|
1 == 1
|
||
|
|
finally:
|
||
|
|
os.mkdir(filepath)
|
||
|
|
|
||
|
|
# Parse the yaml
|
||
|
|
with open(file, 'r') as stream:
|
||
|
|
content = yaml.safe_load(stream)
|
||
|
|
|
||
|
|
# Add service files for each host
|
||
|
|
WriteVMFile(content,'managed','virtual')
|
||
|
|
WriteVMFile(content,'unmanaged','ovas')
|
||
|
|
#WriteVMFile(content,'unmanaged','appliances')
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
if len(sys.argv) != 2:
|
||
|
|
print("You need to supply an inventory file.")
|
||
|
|
sys.exit(1)
|
||
|
|
GenerateFiles(sys.argv[1])
|
||
|
|
sys.exit(0)
|