def __init__(self,
inventory=None,
names=None,
ssid=None,
wifipassword=None,
force_inv=False,
country=None):
# Get inventory
self.ssid = ssid
self.wifipasswd = wifipassword
if inventory is None:
names = Parameter.expand(names)
manager, workers = Host.get_hostnames(names)
if workers:
worker_base_name = ''.join(
[i for i in workers[0] if not i.isdigit()])
cluster_name = manager or worker_base_name
inventory = path_expand(
f'~/.cloudmesh/inventory-{cluster_name}.yaml')
if not os.path.exists(inventory) or force_inv:
if not manager:
Console.error("No inventory found. Can not create an "
"inventory without a "
"manager.")
return ""
Inventory.build_default_inventory(filename=inventory,
manager=manager,
workers=workers)
inv = Inventory(filename=inventory)
else:
inv = Inventory(filename=inventory)
self.inventory = inv
# Find managers and workers
managers = inv.find(service='manager')
if len(managers) > 0:
if not self.ssid:
self.ssid = get_ssid()
if self.ssid == "":
Console.info('Could not determine SSID, skipping wifi '
'config')
self.ssid = None
if not self.wifipasswd and self.ssid:
self.wifipasswd = getpass(f"Using --SSID={self.ssid}, please "
f"enter wifi password:")
workers = inv.find(service='worker')
# No inherenet need to distinguish the configs by service
configs = managers + workers
# Create dict for them for easy lookup
self.configs = dict((config['host'], config) for config in configs)
self.get_images()
self.country = country if country else Shell.locale().upper()
class Configure:
"""
This class serves to build cloud-init config files for entries in a cloudmesh inventory file.
This accepts two params for init:
inventory = INVENTORY_FILE
cluster = CLUSTER_NAME
If inventory arg is None, then default ~/.cloudmesh/inventory.yaml is used
if cluster arg is not None, then nodes are found by searching via the
"cluster" column in inventory
if cluster arg is None, then nodes are found by searching for "worker" and "manager" in the
"service" column in inventory
Usage:
config_generator = Configure()
Configure.build_user_data(name=NAME) returns a Userdata builder object
where NAME is the hostname of an entry in inventory.yaml with corresponding config options
"""
KEY_DIR = path_expand('~/.cloudmesh/cmburn')
def __init__(self,
inventory=None,
cluster=None,
debug=False,
download_images=False):
self.debug = debug
self.manager_public_key = None # Populated by self.generate_ssh_key
if inventory:
self.inventory = Inventory(inventory)
else:
self.inventory = Inventory()
if cluster is not None:
self.nodes = self.inventory.find(cluster=cluster)
else:
self.nodes = self.inventory.find(
service='manager') + self.inventory.find(service='worker')
self.manager_public_key = None
managers = self.inventory.find(service='manager')
workers = self.inventory.find(service='worker')
configs = managers + workers
# Create dict for them for easy lookup
self.configs = dict((config['host'], config) for config in configs)
if download_images:
self.get_images()
def get_images(self):
"""
Downloads all tags found in self.configs
"""
tags = set()
for config in self.configs.values():
try:
tags.add(config['tag'])
except KeyError as e:
Console.warning(
f'Could not find tag for {config["host"]}. Skipping')
banner("Downloading Images", figlet=True)
image = Image()
for tag in tags:
Console.info(f'Attempting to download {tag}')
res = image.fetch(tag=[tag])
if not res:
Console.error('Failed Image Fetch.')
raise Exception('Failed Image Fetch')
def build_user_data(self,
name=None,
with_defaults=True,
country=None,
add_manager_key=False,
upgrade=False,
with_bridge=False):
"""
Given a name, get its config from self.inventory and create a Userdata object
"""
if name is None:
raise Exception('name arg supplied is None')
elif not self.inventory.has_host(name):
raise Exception(
f'Could not find {name} in {self.inventory.filename}')
if country is not None and len(country) != 2:
raise Exception('Country code is not 2 characters.')
# Get the current configurations from inventory
hostname = self.inventory.get(name=name, attribute='host')
keyfile = self.inventory.get(name=name, attribute='keyfile')
if keyfile:
keys = readfile(keyfile).strip().split('\n')
else:
keys = None
if keys is None and add_manager_key:
keys = [self.manager_public_key]
elif add_manager_key:
keys.append(self.manager_public_key)
service = self.inventory.get(name=name, attribute='service')
# Build Userdata
user_data = Userdata()
if with_defaults:
user_data.with_locale().with_net_tools().with_packages(
packages=['avahi-daemon', 'libraspberrypi-bin'])
if upgrade:
user_data.with_package_update().with_package_upgrade()
if hostname:
user_data.with_hostname(hostname=hostname)
if keys:
# Disable password auth in favor of key auth
user_data.with_ssh_password_login(ssh_pwauth=False)
user_data.with_authorized_keys(keys=keys)
else:
user_data.with_default_user().with_ssh_password_login()
# Add known hosts
user_data.with_hosts(hosts=self.get_hosts_for(name=name))
if country:
user_data.with_set_wifi_country(country=country)
if service == 'manager' and self.manager_public_key:
# If public key is set, then we expect /boot/firmware/id_rsa and /boot/firmware/id_rsa.pub on burned card
user_data.with_runcmd(cmd='cat /boot/firmware/id_rsa.pub > /home/ubuntu/.ssh/id_rsa.pub')\
.with_runcmd(cmd='cat /boot/firmware/id_rsa > /home/ubuntu/.ssh/id_rsa')\
.with_fix_user_dir_owner(user='******')\
.with_runcmd(cmd='chmod 600 /home/ubuntu/.ssh/id_rsa')\
.with_runcmd(cmd='sudo rm /boot/firmware/id_rsa.pub')\
.with_runcmd(cmd='sudo rm /boot/firmware/id_rsa')
if with_bridge:
user_data.with_access_point_bridge_nftables()
if with_defaults:
# disable cloud-init on subsequent boots
user_data.with_runcmd(
cmd='sudo touch /etc/cloud/cloud-init.disabled')
if self.debug:
Console.info(f'User data for {name}:\n' + str(user_data))
return user_data
def build_network_data(self,
name=None,
ssid=None,
password=None,
with_defaults=True):
"""
Given a name, get its config from self.inventory and create a Networkdata object
"""
if name is None:
raise Exception('name arg supplied is None')
elif not self.inventory.has_host(name):
raise Exception(
f'Could not find {name} in {self.inventory.filename}')
if ssid or password:
if not ssid or not password:
raise Exception(
"ssid or password supplied with no corresponding ssid or password"
)
# Get the current configurations from inventory
eth0_ip = self.inventory.get(name=name, attribute='ip')
eth0_nameservers = self.inventory.get(name=name, attribute='dns')
eth0_gateway = self.inventory.get(name=name, attribute='router')
network_data = Networkdata()
if with_defaults:
network_data.with_defaults()
if eth0_ip:
network_data.with_ip(ip=eth0_ip)
if eth0_nameservers:
network_data.with_nameservers(nameservers=eth0_nameservers)
if eth0_gateway:
network_data.with_gateway(gateway=eth0_gateway)
if ssid and password:
Console.info(f'Providing WiFi access to {name}')
network_data.with_access_points(ssid=ssid, password=password)\
.with_dhcp4(interfaces='wifis', interface='wlan0', dhcp4=True)\
.with_optional(interfaces='wifis', interface='wlan0', optional=True)
if self.debug:
Console.info(f'Network data for {name}:\n' + str(network_data))
return network_data
def get_hosts_for(self, name=None):
"""
Given a hostname, return a list of ':' separated strings of the form:
ip:hostname
for all hosts with ips in the inventory. Also includes mapping for own hostname in the form of 127.0.0.1:{name}
For example, if inventory has worker001 with ip 10.1.1.1, worker002 with ip 10.1.1.2, worker003 with ip 10.1.1.3,
then:
self.get_hosts_for(name='worker001') returns ['127.0.0.1:worker001', '10.1.1.2:worker002', '10.1.1.3:worker003']
Do not rely on the order of the result here
"""
if name is None:
raise Exception('name arg supplied is None')
if not self.inventory.has_host(name):
raise Exception(
f'{name} could not be found in {self.inventory.filename}')
result = [f'127.0.0.1:{name}']
for node in self.nodes:
if node['ip'] and name != node['host']:
host = node['host']
ip = node['ip']
result += [f'{ip}:{host}']
return result
def generate_ssh_key(self, hostname):
Shell.execute('mkdir', f'-p {Configure.KEY_DIR}')
Shell.run(f'ssh-keygen -q -N "" -C "ubuntu@{hostname}" -f '
f'{Configure.KEY_DIR}/id_rsa')
priv_key = readfile(f'{Configure.KEY_DIR}/id_rsa').strip()
pub_key = readfile(f'{Configure.KEY_DIR}/id_rsa.pub').strip()
self.manager_public_key = pub_key
Shell.execute('rm', path_expand(f'{Configure.KEY_DIR}/id_rsa'))
Shell.execute('rm', path_expand(f'{Configure.KEY_DIR}/id_rsa.pub'))
return priv_key, pub_key
class Test_inventory:
def setup(self):
self.i = Inventory('~/.cloudmesh/test.yaml')
self.i.info()
def test_add(self):
HEADING(txt="cms inventory add")
cluster = "test_cluster"
label = "test{j}"
host = "red00{j}"
ip = "10.1.1.{j}"
dns = "8.8.8.8"
# Adding one by one
for j in range(1, 4):
self.i.add(
host=host.format(j=j),
cluster=cluster,
dns=dns,
label=label.format(j=j),
ip=ip.format(j=j),
)
for j in range(1, 4):
host_name = host.format(j=j)
assert self.i.has_host(host_name)
entry = self.i.data[host_name]
assert entry['cluster'] == cluster
assert entry['ip'] == ip.format(j=j)
assert entry['label'] == label.format(j=j)
assert entry['dns'] == dns
# Multi add
self.i.add(host=host.format(j='[4-7]'),
cluster=cluster,
ip=ip.format(j='[4-7]'))
for j in range(4, 8):
host_name = host.format(j=j)
assert self.i.has_host(host_name)
entry = self.i.data[host_name]
assert entry['cluster'] == cluster
assert entry['ip'] == ip.format(j=j)
self.i.save()
def test_list(self):
HEADING(txt="cms inventory list")
# for output in ['dict', 'yaml', 'csv', 'table']: # bug in 'csv' print in cloudmesh-common
for output in ['dict', 'yaml', 'table']:
banner(output)
print(self.i.list(format=output))
t = str(self.i.list('table'))
for order in self.i.order:
assert order in t
def test_set(self):
HEADING()
self.i.set("red002", "service", "worker")
assert self.i.get("red002", "service") == "worker"
self.i.save()
def test_find(self):
HEADING()
assert len(self.i.find(host="red002")) > 0
