Last active
July 21, 2021 22:35
-
-
Save arjunnambiartc/8954313 to your computer and use it in GitHub Desktop.
Login to multiple devices,run multiple commands using paramiko, store the results using xlswriter and send email using Python's smtplib library
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#! /usr/bin/python | |
import os | |
import getpass | |
import paramiko | |
import xlsxwriter | |
import socket | |
import re | |
import sys | |
import smtplib | |
from email import encoders | |
from email.mime.multipart import MIMEMultipart | |
from email.mime.base import MIMEBase | |
from email.MIMEText import MIMEText | |
username = raw_input('Enter username for device login:') | |
def enterPassword(): | |
while True: # repeat forever | |
password = getpass.getpass('Enter corresponding password:') | |
password_again = getpass.getpass('Confirm password:') | |
if password != password_again: | |
print 'Password and confirmation do not match.Please try again!!' | |
else: | |
return password | |
password = enterPassword() | |
recipient = raw_input('Enter your email-id where the results of the test will be mailed to you :') | |
sender = raw_input('Enter recipient email address':) | |
subject = raw_input('Enter subject for the mail':) | |
message = raw_input('Enter the message body:') | |
SMTP_SERVER = raw_input('Enter SMTP server FQDN':) | |
SMTP_PORT = 587 | |
print "Running the tests..this might take some time.." | |
# Opens file in read mode | |
f1 = open('hostfile','r') | |
f2 = open('commandfile','r') | |
# Creates list based on f1 | |
devices = f1.readlines() | |
commands = f2.readlines() | |
ssh = paramiko.SSHClient() | |
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) | |
data = [] | |
for device in devices: | |
column = device.split() | |
data.append([column[0]]) | |
print column[0] | |
for command in commands: | |
try: | |
conn=ssh.connect(column[1], username=username, password=password, timeout=4) | |
if conn is None: | |
stdin, stdout, stderr = ssh.exec_command(command) | |
data[-1].append(stdout.read()) | |
ssh.close() | |
except paramiko.AuthenticationException: | |
output = "Authentication Failed" | |
data[-1].append(output) | |
break | |
except paramiko.SSHException: | |
output = "Issues with SSH service" | |
data[-1].append(output) | |
break | |
except socket.error, e: | |
output = "Connection Error" | |
data[-1].append(output) | |
break | |
data[-1] = tuple(data[-1]) | |
f1.close() | |
f2.close() | |
#Create Workbook instance | |
book = xlsxwriter.Workbook('Workbook.xlsx') | |
sheet = book.add_worksheet('Sheet1') | |
#Define and format header | |
header_format = book.add_format({'bold':True , 'bg_color':'yellow'}) | |
for col, text in enumerate(header): | |
sheet.write(0, col, text, header_format) | |
# Now, let's write the contents | |
for row, data_in_row in enumerate(data): | |
for col, text in enumerate(data_in_row): | |
sheet.write(row + 1, col, text) | |
book.close() | |
# Send excelsheet results in a mail | |
def main(): | |
msg = MIMEMultipart() | |
msg['Subject'] = subject | |
msg['To'] = recipient | |
msg['From'] = sender | |
part = MIMEText('text', "plain") | |
part.set_payload(message) | |
msg.attach(part) | |
session = smtplib.SMTP(SMTP_SERVER) | |
fp = open('Workbook.xlsx', 'rb') | |
msgq = MIMEBase('audio', 'audio') | |
msgq.set_payload(fp.read()) | |
fp.close() | |
# Encode the payload using Base64 | |
encoders.encode_base64(msgq) | |
# Set the filename parameter | |
filename='Workbook.xlsx' | |
msgq.add_header('Content-Disposition', 'attachment', filename=filename) | |
msg.attach(msgq) | |
# Now send or store the message | |
qwertyuiop = msg.as_string() | |
session.sendmail(sender, recipient, qwertyuiop) | |
session.quit() | |
if __name__ == '__main__': | |
main() | |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I made the same thing with netmiko, a little different formatting but it is nice seeing other ways to do the same thing. Yours is very efficient thanks for sharing.