rsi.py

#!/usr/bin/python
###############################################################
#
# Contrail rsi tool for collecting a node information
# Usage:
#    ./rsi.py <node ip>
#  Ex: 
#      ./rsi.py 10.219.90.82
#  Output:
#     collects specified commands( see scripts/ folder within)
#     as a compressed tar file 
#        saves it locally rsi-<ddMMMyy-hhmmss>.tgz
#     Ex: rsi-20Mar18-121953.tgz  
###############################################################

import os
import time
import sys
import getpass
import os.path
import paramiko
from paramiko import SSHClient
from scp import SCPClient

rsiVersion=0.1

archivedir='scripts'
archive='scripts.tar'

#TBD: traverse all tenants 
#for now, specific tenant information is captured
tenantcred='tenant.cred'

def checkPing(host):
    response = os.system("ping -c 1 -t2 " + host + "> /dev/null")
    # and then check the response...
    if response == 0:
        pingstatus = True
    else:
        pingstatus = False

    return pingstatus 

def checkServer(server, user, pwd, chkcmd):
 if not checkPing(server):
  print server, 'not reachable?'
  print 'exiting'
  exit(0)    
 proxy = None
 ssh = paramiko.SSHClient()
 ssh.load_system_host_keys()
 ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
 try:
   ssh.connect(server, username=user, password=pwd, timeout=2)
   stdin, stdout, stderr = ssh.exec_command(chkcmd)
 
   for line in stderr:
     if 'not' in line:
       print server, 'is not a contrail node?'
       ssh.close()
       return False
   c=0
   for line in stdout:
     c+=1
   if not c:
       print server, 'is not running contrail services?'
       ssh.close()
       return False
   print server, 'is a contrail node'
   return True
   
   return True
 except paramiko.AuthenticationException:
   print("Authentication failed?")
   return False
 except:
   print("unknown error")
   print server, 'is a contrail node?'
   return False
 ssh.close()

def cleanup():
#remove local scripts tarball
  os.system('rm -rf scripts.tar')
  os.system('rm -rf /root/rsi/scripts')
 
def runRsi(server, file, user, pwd):
 global tenantcred
 proxy = None
 ssh = paramiko.SSHClient()
 ssh.load_system_host_keys()
 ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
 ssh.connect(server, username=user, password=pwd)

 # SCPClient takes a paramiko transport as its only argument
 scp = SCPClient(ssh.get_transport())
 out = ssh.exec_command('mkdir -p rsi 2>>/tmp/err')
 out = ssh.exec_command('hostname')[1]
 hostname = out.readlines()[0].rstrip()
 targetfile='/root/rsi/'+file
 scp.put(file, targetfile)
 rsiFolderNamw = hostname+'-'+str(int(time.time()))

 if os.path.isfile(tenantcred):
   scp.put(file, tenantcred)
 
 
 stdout = ssh.exec_command('rm /root/rsi/Finished')
 stdin,stdout,stderr = ssh.exec_command('tar xf ' + targetfile + ' -C rsi/')
 stdout = ssh.exec_command('rsi/scripts/rsi.sh ' + rsiFolderNamw)[1]
 rsiFileName = rsiFolderNamw+'.tgz'
 remFileName= '/root/rsi/' + rsiFileName 
 stdoutlen = 0
 waitTime = 60
 #while not stdoutlen and retries < 20:
 done = False
 while waitTime:
    try:
       stdin, stdout, stderr = ssh.exec_command('ls -l ' + '/root/rsi/Finished')
       stdout.readlines()[0]  
    except IndexError:
       waitTime -=2
       time.sleep(2) 
    else:
      scp.get('rsi/'+rsiFileName, './'+rsiFileName)
      print 'RSI saved to: ', rsiFileName
      out = ssh.exec_command('rm /root/rsi/Finished')
      out = ssh.exec_command('rm -rf /root/rsi/' + rsiFolderNamw)
      out = ssh.exec_command('rm -rf /root/rsi/scripts.tar')
      out = ssh.exec_command('rm -rf /root/rsi/scripts')
      return True
 print 'RSI session hung?'
 return False

 #stderr = ssh.exec_command('bash test.sh')[2]

def checkScripts():
 #print 'checking'
 if not os.path.isdir(archivedir):
  return False
 #print archivedir, 'is found..'
 return True

if not checkScripts():
 print archivedir, 'is not found!'
 exit(0)
user='root'
server=''

print 'contrail-rsi version:', rsiVersion
print
if len(sys.argv) < 2:
   print 'Enter Node FQDN/Address:',
   server = raw_input()

else:
   server = sys.argv[1]

targetfile=int(time.time() )
user = 'root'
pwd = ''
#Comment this line
#pwd = 'Juniper'
if not pwd:
 pwd = getpass.getpass()
else:
 print 'Warning: password is hardcoded! use getpass()'
print 'logging to', server




os.system('tar cf ' + archive + ' scripts/')

if checkServer(server, user, pwd, 'contrail-status'):
  print "OK"
  status = False
  while not status:
     status = runRsi(server, archive, user, pwd)
else:
  print('exiting');

cleanup()