
This copyright change reflects previous changes that Juerg made for pylint and pep8 cleanups. From: Juerg Haefliger <juerg.haefliger@hp.com> Date: Mon, 16 Jan 2012 10:45:12 +0100
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
# vi: ts=4 expandtab
|
|
#
|
|
# Copyright (C) 2009-2010 Canonical Ltd.
|
|
# Copyright (C) 2012 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# Author: Scott Moser <scott.moser@canonical.com>
|
|
# Author: Juerg Haefliger <juerg.haefliger@hp.com>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License version 3, as
|
|
# published by the Free Software Foundation.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import cloudinit.util as util
|
|
import subprocess
|
|
import traceback
|
|
|
|
|
|
def handle(_name, cfg, _cloud, log, args):
|
|
if len(args) != 0:
|
|
user = args[0]
|
|
ids = []
|
|
if len(args) > 1:
|
|
ids = args[1:]
|
|
else:
|
|
user = util.get_cfg_option_str(cfg, "user", "ubuntu")
|
|
ids = util.get_cfg_option_list_or_str(cfg, "ssh_import_id", [])
|
|
|
|
if len(ids) == 0:
|
|
return
|
|
|
|
cmd = ["sudo", "-Hu", user, "ssh-import-id"] + ids
|
|
|
|
log.debug("importing ssh ids. cmd = %s" % cmd)
|
|
|
|
try:
|
|
subprocess.check_call(cmd)
|
|
except subprocess.CalledProcessError as e:
|
|
log.debug(traceback.format_exc(e))
|
|
raise Exception("Cmd returned %s: %s" % (e.returncode, cmd))
|
|
except OSError as e:
|
|
log.debug(traceback.format_exc(e))
|
|
raise Exception("Cmd failed to execute: %s" % (cmd))
|