mirror of
https://github.com/khuedoan/homelab.git
synced 2025-01-10 07:16:57 +07:00
95 lines
2.4 KiB
Python
Executable File
95 lines
2.4 KiB
Python
Executable File
#!/bin/python
|
|
|
|
import os
|
|
import time
|
|
import string
|
|
|
|
user = "root"
|
|
|
|
nodes = [
|
|
{
|
|
'name': 'Node 0',
|
|
'mac': '00-23-24-d1-f3-f0',
|
|
'ip': '192.168.1.110',
|
|
'hostname': 'node0.homelab.khuedoan.com'
|
|
},
|
|
{
|
|
'name': 'Node 1',
|
|
'mac': '00-23-24-d1-f4-d6',
|
|
'ip': '192.168.1.111',
|
|
'hostname': 'node1.homelab.khuedoan.com'
|
|
},
|
|
{
|
|
'name': 'Node 2',
|
|
'mac': '00-23-24-e7-04-60',
|
|
'ip': '192.168.1.112',
|
|
'hostname': 'node2.homelab.khuedoan.com'
|
|
},
|
|
{
|
|
'name': 'Node 3',
|
|
'mac': '00-23-24-d1-f5-69',
|
|
'ip': '192.168.1.113',
|
|
'hostname': 'node3.homelab.khuedoan.com'
|
|
}
|
|
]
|
|
|
|
def is_alive(node):
|
|
return os.system(f"ping -q -c 1 {node['ip']}") == 0
|
|
|
|
def is_ready(node):
|
|
return os.system(f"ssh -q -o StrictHostKeyChecking=no {user}@{node['ip']} exit") == 0
|
|
|
|
def generate_network_config(node):
|
|
with open('./kickstart/config/00-00-00-00-00-00.ks.template', 'r') as template_file:
|
|
template = string.Template(template_file.read())
|
|
config = template.substitute({
|
|
'NETWORK_DEVICE': 'eno1',
|
|
'IP': node['ip'],
|
|
'GATEWAY': '192.168.1.0',
|
|
'NETMASK': '255.255.255.0',
|
|
'HOSTNAME': node['hostname'],
|
|
'NAMESERVER_1': '8.8.8.8',
|
|
'NAMESERVER_2': '8.8.4.4'
|
|
})
|
|
|
|
with open(f"./kickstart/config/{node['mac']}.ks", 'w') as config_file:
|
|
config_file.write(config)
|
|
|
|
def poweroff(node):
|
|
if is_alive(node):
|
|
print(f"Poweroff {node['name']}")
|
|
os.system(f"ssh -q -o StrictHostKeyChecking=no {user}@{node['ip']} poweroff")
|
|
else:
|
|
print(f"Node {node['name']} is already dead!")
|
|
|
|
def wake(node):
|
|
os.system(f"wol {node['mac']}")
|
|
|
|
def forget(node):
|
|
os.system(f"ssh-keygen -R {node['ip']}")
|
|
os.system(f"ssh-keygen -R {node['hostname']}")
|
|
|
|
if __name__ == "__main__":
|
|
os.chdir(f"./infra/pxe-server")
|
|
|
|
for node in nodes:
|
|
generate_network_config(node)
|
|
|
|
os.system(f"docker-compose up -d --build")
|
|
|
|
for node in nodes:
|
|
poweroff(node)
|
|
forget(node)
|
|
|
|
time.sleep(10)
|
|
|
|
for node in nodes:
|
|
wake(node)
|
|
|
|
while not all(is_ready(node) for node in nodes):
|
|
os.system(f"docker-compose logs -t --tail={len(nodes)} nginx")
|
|
print("Waiting for all servers to start up...")
|
|
time.sleep(10)
|
|
|
|
os.system(f"docker-compose down")
|