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

refactored

Browse Source
This commit is contained in:
Toby
2021-03-13 16:12:36 +01:00
parent bce2d3bcb3
commit da8a70297b
2 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
monitor.py
View File

@@ -4,19 +4,19 @@ import paramiko
import threading
import yaml
class taskThread(threading.Thread):
class task_thread(threading.Thread):
def __init__(self, host, keys):
threading.Thread.__init__(self)
self.host = host
self.keys = keys
def run(self):
updateKeys(self.host, self.keys)
update_keys(self.host, self.keys)
def readConfig():
def read_config():
with open('config.yaml', 'r') as stream:
return yaml.safe_load(stream)
def updateKeys(host, keys):
def update_keys(host, keys):
try:
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy())
@@ -29,7 +29,7 @@ def updateKeys(host, keys):
print('❌ ' + host)
def main():
config = readConfig()
config = read_config()
keys = []
@@ -38,7 +38,7 @@ def main():
for host in config['hosts']:
try:
thread = taskThread(host, keys)
thread = task_thread(host, keys)
thread.start()
except:
print('❌ ' + host)

12
sync.py
View File

@@ -4,19 +4,19 @@ import paramiko
import threading
import yaml
class taskThread(threading.Thread):
class task_thread(threading.Thread):
def __init__(self, host, keys):
threading.Thread.__init__(self)
self.host = host
self.keys = keys
def run(self):
updateKeys(self.host, self.keys)
update_keys(self.host, self.keys)
def readConfig():
def read_config():
with open('config.yaml', 'r') as stream:
return yaml.safe_load(stream)
def updateKeys(host, keys):
def update_keys(host, keys):
try:
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy())
@@ -28,7 +28,7 @@ def updateKeys(host, keys):
print('❌ ' + host)
def main():
config = readConfig()
config = read_config()
keys = []
@@ -37,7 +37,7 @@ def main():
for host in config['hosts']:
try:
thread = taskThread(host, keys)
thread = task_thread(host, keys)
thread.start()
except:
print('❌ ' + host)