Skip to content

implement csv table with logs #18

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

implement csv table with logs #18

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
63 changes: 46 additions & 17 deletions SSHKeyDistribut0r/key_distribut0r.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,10 +8,11 @@
import re
import socket
import sys
import yaml
import csv

import paramiko
import scp
import yaml

logging.raiseExceptions = False

Expand All @@ -26,6 +27,9 @@
YAML_EXT = re.compile("^\\.ya?ml$")
JSON_EXT = re.compile("^\\.json$")

ERROR_STATUS = '✗ Error'
SUCCESS_STATUS = '✓ Success'


def remove_special_chars(original_string):
return ''.join(e for e in original_string if e.isalnum())
Expand Down Expand Up @@ -63,10 +67,21 @@ def read_config(config_file):
sys.exit(1)


def create_result_csv_table(messages):
messages.sort(key=lambda m: m[0] == ERROR_STATUS)
try:
with open('ssh_keys_distributor_result.csv', 'w', encoding='utf-8') as file:
writer = csv.writer(file, delimiter='|')
writer.writerows(messages)
except OSError as e:
print(e)


def main(args):
# Load config files
servers = read_config(args.server)
keys = read_config(args.keys)
messages = [('Status', 'Ip', 'Comment', 'Description')]

for server in servers:
if server['authorized_users']:
Expand All @@ -85,7 +100,9 @@ def main(args):
key_stream.write('%s\n' % key)

if args.dry_run:
server_info_log(server['ip'], server['comment'], ', '.join(server_users))
msg = server['ip'], server['comment'], ', '.join(server_users)
server_info_log(*msg)
messages.append((SUCCESS_STATUS, *msg))
else:
# Configure SSH client
ssh_client = paramiko.SSHClient()
Expand All @@ -102,25 +119,37 @@ def main(args):
scp_client.putfo(key_stream, '.ssh/authorized_keys')

key_stream.close()
server_info_log(server['ip'], server['comment'], ', '.join(server_users))
msg = server['ip'], server['comment'], ', '.join(server_users)
server_info_log(*msg)
messages.append((SUCCESS_STATUS, *msg))

except paramiko.ssh_exception.PasswordRequiredException:
server_error_log(
server['ip'],
server['comment'],
'The private key file is protected by a passphrase, which is currently not supported.'
)
msg = server['ip'], \
server['comment'], \
'The private key file is protected by a passphrase, which is currently not supported.'
server_error_log(*msg)
messages.append((ERROR_STATUS, msg))
except paramiko.ssh_exception.AuthenticationException:
server_error_log(
server['ip'],
server['comment'],
'Cannot connect to server because of an authentication problem.'
)
msg = server['ip'],\
server['comment'], \
'Cannot connect to server because of an authentication problem.'
server_error_log(*msg)
messages.append((ERROR_STATUS, *msg))
except scp.SCPException:
server_error_log(server['ip'], server['comment'], 'Cannot send file to server.')
msg = server['ip'], server['comment'], 'Cannot send file to server.'
server_error_log(*msg)
messages.append((ERROR_STATUS, *msg))
except (paramiko.ssh_exception.NoValidConnectionsError, paramiko.ssh_exception.SSHException):
server_error_log(server['ip'], server['comment'], 'Cannot connect to server.')
msg = server['ip'], server['comment'], 'Cannot connect to server.'
server_error_log(*msg)
messages.append((ERROR_STATUS, *msg))
except socket.timeout:
server_error_log(server['ip'], server['comment'], 'Cannot connect to server because of a timeout.')
msg = server['ip'], server['comment'], 'Cannot connect to server because of a timeout.'
server_error_log(*msg)
messages.append((ERROR_STATUS, *msg))
else:
server_error_log(server['ip'], server['comment'], 'No user mentioned in configuration file!')
msg = server['ip'], server['comment'], 'No user mentioned in configuration file!'
server_error_log(*msg)
messages.append((ERROR_STATUS, *msg))
create_result_csv_table(messages)