32 lines
1,013 B
Python
32 lines
1,013 B
Python
|
from dynamicweb.celery import app
|
||
|
|
from celery.utils.log import get_task_logger
|
||
|
|
from django.conf import settings
|
||
|
|
from cdist.integration import configure_hosts_simple
|
||
|
|
import cdist
|
||
|
|
import tempfile
|
||
|
|
import pathlib
|
||
|
|
|
||
|
|
logger = get_task_logger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
@app.task(bind=True, max_retries=settings.CELERY_MAX_RETRIES)
|
||
|
|
def save_ssh_key(hosts, keys):
|
||
|
|
"""
|
||
|
|
Saves ssh key into the VMs of a user using cdist
|
||
|
|
|
||
|
|
:param hosts: A list of hosts to be configured
|
||
|
|
:param keys: A list of keys to be added
|
||
|
|
"""
|
||
|
|
# Generate manifest to be used for configuring the hosts
|
||
|
|
with tempfile.NamedTemporaryFile() as tmp_manifest:
|
||
|
|
tmp_manifest.writelines(['__ssh_authorized_keys root \\',
|
||
|
|
' --key "{keys}"'.format(
|
||
|
|
keys='\n'.join(keys))])
|
||
|
|
|
||
|
|
f = pathlib.Path(tmp_manifest.name)
|
||
|
|
configure_hosts_simple(hosts,
|
||
|
|
tmp_manifest.name,
|
||
|
|
verbose=cdist.argparse.VERBOSE_TRACE)
|
||
|
|
|
||
|
|
|