tsndr/sync-ssh-keys
1
1
Fork 0
Code Pull Requests Activity

Add port option

Browse Source
This commit is contained in:
Toby
2022-04-02 01:21:15 +02:00
parent 0a20674bf5
commit 09ea7f77a3
1 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
sync.py
View File

@@ -5,23 +5,24 @@ import threading
import yaml import yaml
class task_thread(threading.Thread): class task_thread(threading.Thread):
def __init__(self, host, user, keys): def __init__(self, host, port, user, keys):
threading.Thread.__init__(self) threading.Thread.__init__(self)
self.host = host self.host = host
self.port = port
self.user = user self.user = user
self.keys = keys self.keys = keys
def run(self): def run(self):
update_keys(self.host, self.user, self.keys) update_keys(self.host, self.port, self.user, self.keys)
def read_config(): def read_config():
with open('config.yaml', 'r') as stream: with open('config.yaml', 'r') as stream:
return yaml.safe_load(stream) return yaml.safe_load(stream)
def update_keys(host, user, keys): def update_keys(host, port, user, keys):
try: try:
client = paramiko.SSHClient() client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy()) client.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy())
client.connect(host, username = user, timeout = 1) client.connect(host, username = user, port = port, timeout = 1)
client.exec_command('echo "###\n# Warning this file has been generated and will be overwritten!\n###\n' + '\n'.join(keys) + '" > ~/.ssh/authorized_keys2') client.exec_command('echo "###\n# Warning this file has been generated and will be overwritten!\n###\n' + '\n'.join(keys) + '" > ~/.ssh/authorized_keys2')
client.close() client.close()
print('✅ ' + user + '@' + host) print('✅ ' + user + '@' + host)
@@ -49,8 +50,10 @@ def main():
host_keys = list(set(host_keys)) # Filter duplicates host_keys = list(set(host_keys)) # Filter duplicates
if not host_keys: if not host_keys:
continue continue
if not 'port' in host:
host['port'] = 22
try: try:
thread = task_thread(host['host'], user_name, host_keys) thread = task_thread(host['host'], host['port'], user_name, host_keys)
thread.start() thread.start()
except: except:
print('❌ ' + user_name + '@' + host['host']) print('❌ ' + user_name + '@' + host['host'])