You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
2.1 KiB
51 lines
2.1 KiB
#!/usr/bin/env python3 |
|
|
|
import os |
|
import sys |
|
import subprocess |
|
|
|
class SSHKey(object): |
|
def __init__(self, parser, parents): |
|
self.parser = parser |
|
|
|
self.parser['sshkey'] = self.parser['sub'].add_parser( |
|
'sshkey', |
|
help="Manage SSH keys", |
|
parents=[parents]) |
|
|
|
self.parser['sshkey'].add_argument('--user', |
|
help='Username on the host', |
|
required=False, |
|
default="root") |
|
self.parser['sshkey'].add_argument('--key', |
|
help='Name of the key', |
|
required=True) |
|
self.parser['sshkey'].add_argument('--key-dir', |
|
help='Directory holding keys', |
|
default=os.path.join(os.environ['HOME'], "vcs/ungleich-ssh-keys/current"), |
|
required=False) |
|
self.parser['sshkey'].add_argument('--host', |
|
help='Host to use', |
|
required=True) |
|
self.parser['sshkey'].add_argument('--delete', |
|
help='Delete key instead of adding', |
|
action="store_true") |
|
self.parser['sshkey'].set_defaults(func=self._manage_key) |
|
|
|
|
|
def _manage_key(self, args): |
|
if args.delete: |
|
print("Not yet supported to delete, sorry :-)") |
|
sys.exit(1) |
|
|
|
keyfile = os.path.join(args.key_dir, "{}.pub".format(args.key)) |
|
|
|
if not os.path.exists(keyfile): |
|
print("Key for {} does not exist in {}. Aborting".format(args.key, args.key_dir)) |
|
sys.exit(1) |
|
|
|
cmd = 'cat {} | ssh {}@{} "mkdir -p ~/.ssh; cat >> ~/.ssh/authorized_keys"'.format(keyfile, |
|
args.user, |
|
args.host) |
|
|
|
subprocess.run(cmd, shell=True)
|
|
|