def write_user_key(self, key):
key_filename = None
for cls, filename in self.DEFAULT_KEY_FILES:
if isinstance(key, cls):
key_filename = filename
if not key_filename:
raise UnsupportedSSHKeyError()
sshdir = self.ensure_ssh_dir()
filename = os.path.join(sshdir, key_filename)
key.write_private_key_file(filename)
def import_user_key(self, keyfile):
"""Imports an uploaded key file into Review Board.
``keyfile`` is expected to be an ``UploadedFile`` or a paramiko
``KeyFile``. If this is a valid key file, it will be saved in
the storage backend and the resulting key as an instance of
:py:mod:`paramiko.PKey` will be returned.
If a key of this name already exists, it will be overwritten.
Callers are expected to handle any exceptions. This may raise
IOError for any problems in writing the key file, or
paramiko.SSHException for any other problems.
This will raise UnsupportedSSHKeyError if the uploaded key is not
a supported type.
"""
# Try to find out what key this is.
for cls in self.SUPPORTED_KEY_TYPES:
key = None
try:
if not isinstance(keyfile, paramiko.PKey):
keyfile.seek(0)
key = cls.from_private_key(keyfile)
elif isinstance(keyfile, cls):
key = keyfile
except paramiko.SSHException:
# We don't have more detailed info than this, but most
# likely, it's not a valid key. Skip to the next.
continue
if key:
self._write_user_key(key)
return key
raise UnsupportedSSHKeyError()