62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
import os
|
||
|
|
import argparse
|
||
|
|
import sys
|
||
|
|
import yaml
|
||
|
|
|
||
|
|
|
||
|
|
BASE_DIR='generated_vars/networking'
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
parser = argparse.ArgumentParser(description='Generate local vars for networking.')
|
||
|
|
parser.add_argument('netname', metavar='netname', type=str, help='the netname')
|
||
|
|
parser.add_argument('hostname', metavar='hostname', type=str, help='the hostname')
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
NET_DIR = os.path.join(BASE_DIR, args.netname)
|
||
|
|
HOST_PATH = os.path.join(NET_DIR, args.hostname)
|
||
|
|
|
||
|
|
if not os.path.exists(NET_DIR):
|
||
|
|
os.makedirs(NET_DIR)
|
||
|
|
|
||
|
|
if os.path.exists(HOST_PATH):
|
||
|
|
with open(HOST_PATH, 'r') as f:
|
||
|
|
host_number = int(f.read())
|
||
|
|
print host_number
|
||
|
|
f.close()
|
||
|
|
else:
|
||
|
|
host_number = 0
|
||
|
|
defined_hosts = [f for f in os.listdir(NET_DIR) if os.path.isfile(os.path.join(NET_DIR, f))]
|
||
|
|
for host in defined_hosts:
|
||
|
|
with open(os.path.join(NET_DIR, host), 'r') as f:
|
||
|
|
ip = int(f.read())
|
||
|
|
if ip > host_number:
|
||
|
|
host_number = ip
|
||
|
|
f.close()
|
||
|
|
|
||
|
|
host_number += 1
|
||
|
|
|
||
|
|
with open(HOST_PATH, 'w') as f:
|
||
|
|
f.write(str(host_number))
|
||
|
|
f.close()
|
||
|
|
print host_number
|
||
|
|
|
||
|
|
host_vars_dir = 'host_vars/'
|
||
|
|
if not os.path.exists(host_vars_dir):
|
||
|
|
os.makedirs(host_vars_dir)
|
||
|
|
|
||
|
|
host_file = os.path.join(host_vars_dir, args.hostname)
|
||
|
|
if os.path.exists(host_file):
|
||
|
|
with open(host_file, 'rw+') as f:
|
||
|
|
data = yaml.load(f)
|
||
|
|
f.close()
|
||
|
|
os.remove(host_file)
|
||
|
|
else:
|
||
|
|
data = {}
|
||
|
|
data[args.netname] = { 'host_number': host_number }
|
||
|
|
|
||
|
|
with open(host_file, 'w') as f:
|
||
|
|
f.write(yaml.dump(data, explicit_start=True, default_flow_style=False))
|
||
|
|
f.close()
|