SHARE
TWEET

Untitled

a guest Jun 15th, 2017 59 Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. import pexpect
  2. import string
  3. import datetime
  4. import MySQLdb
  5. from pexpect import pxssh
  6. import time
  7. import datetime
  8. import threading
  9. from threading import Timer
  10. import warnings
  11. import re
  12. import logging
  13. import signal
  14. import sys
  15.  
  16. #set up the logger
  17. #set up object to access the database
  18. db = MySQLdb.connect("localhost","root","","CPUmonitor")
  19. print(db)
  20. cursor = db.cursor()
  21. startTime = time.time()
  22.  
  23. #delay between
  24. refreshRate = 3
  25.  
  26. class sshSession(object):
  27.     def __init__(self,cardName,ip,username,password):
  28.         self.cardName = cardName
  29.         self.ip = ip
  30.         self.username = username
  31.         self.password = password
  32.         self.handler = pxssh.pxssh() #create ssh handler
  33.         self._processDict  = {}#{'process_name':column_index}
  34.         #create a new table in the database
  35.         sql = "DROP TABLE IF EXISTS "+self.cardName+";"
  36.         #sql = """CREATE TABLE """+self.cardName+""" (time TIMESTAMP) IF NOT EXISTS """+self.cardName
  37.         cursor.execute(sql)
  38.         sql = "CREATE TABLE "+self.cardName+" (time FLOAT);"
  39.         cursor.execute(sql)
  40.         self.running = True
  41.    
  42.     def login(self): #logs in using the ssh handler created in __init__
  43.         self.handler.login(self.ip,self.username,self.password)
  44.         print('logged in to '+self.cardName)
  45.    
  46.     def logout(self):
  47.         self.handler.logout()
  48.    
  49.     def ping(self):
  50.         print("pinging")
  51.         self.running = False #stops execution of dataloop
  52.         #self.handler.logout()
  53.         self.handler.sendline('ping '+self.ip) #ping with a very long timeout
  54.         self.handler.prompt(timeout=-1) #expect the prompt with no timeout
  55.         print("card responded")
  56.         #executes this code if prompt found
  57.         #self.login() #login
  58.         self.running = True
  59.  
  60.     def dataLoop(self):
  61.         if(self.running):
  62.             self.refreshData()
  63.             time.sleep(refreshRate)
  64.             self.dataLoop()
  65.        
  66.     def refreshData(self): #read the output of top command
  67.         #change to top -d 1 -n 3 | awk '{ print $10","$13 }'
  68.         command = "top -d 1 -n "+str(refreshRate)+" | awk \'{ print $10 \" \" $13 }\'"
  69.         self.handler.sendline(command)
  70.         successful = self.handler.prompt(timeout=30)
  71.         if(not successful):
  72.             print(self.cardName+" timed out!!!!")
  73.             self.ping()
  74.  
  75.         output = self.handler.before
  76.         newOutput = output.strip()
  77.         #print(newOutput)
  78.         #print(repr(newOutput))
  79.         index = newOutput.rfind("\r\n \r\n")
  80.         #print(index)
  81.         newOutput = newOutput[index+5:]
  82.         newOutput = newOutput.replace('\\x','')
  83.         lines = newOutput.split('\r\n')
  84.  
  85.         data_to_be_inserted = [0]*(len(self._processDict)+1) #will become the row inserted into the db
  86.         data_to_be_inserted[0] = time.time()-startTime
  87.  
  88.         for line in lines:
  89.             delimiter = line.find(' ')
  90.             CPU = line[:delimiter]
  91.             process = line[delimiter+1:]
  92.             invalid_characters = ['.','\x1b','/','-','+',':',';','*','[',']']
  93.             for character in invalid_characters:
  94.                 process = process.replace(character,'')
  95.             if(process.isdigit()):
  96.                 process.append('a')
  97.             if process in self._processDict: #check if the process is already a column in the db
  98.                 column_index = self._processDict[process]
  99.                 data_to_be_inserted[column_index] = CPU  
  100.             else:
  101.                 column_index = len(self._processDict)+1
  102.                 self._processDict[process] = column_index
  103.                 data_to_be_inserted.append(CPU)
  104.                 #add a new column to the database with label of process
  105.                 sql = "ALTER TABLE "+self.cardName+" ADD "+process+" FLOAT DEFAULT 0;"
  106.                 cursor.execute(sql)
  107.         #put the data into the database
  108.         data_to_be_inserted = tuple(data_to_be_inserted)
  109.         #print(data_to_be_inserted)
  110.         sql = "INSERT INTO "+self.cardName+" VALUES "+str(data_to_be_inserted)
  111.         #print(sql)
  112.         cursor.execute(sql)
  113.  
  114. #open and read the config file
  115. config = open('config.txt','r')
  116. lines = config.readlines()
  117. config.close()
  118. #concatenate array into one long string
  119. text = ''.join(lines)
  120. #search for all matches to pattern
  121. regex = '(\S+)\n\s+IP: (\S+)\n\s+USER: (\S+)\n\s+PASSWORD: (\S+)'  
  122. regex = re.compile(regex)
  123. matches = re.findall(regex,text)
  124.  
  125. #use matches to create spawn and login children (ssh objects) and add them to the list of children
  126. childList = []
  127. for match in matches:
  128.     name,ip,user,password = match
  129.     #print(name)
  130.     temp = sshSession(name,ip,user,password)
  131.     temp.login()
  132.     childList.append(temp)
  133. print('doing something')
  134. #def thread(child):
  135. #    child.refreshData()
  136. #    time.sleep(refreshRate)
  137. #    thread(child)
  138.  
  139. threadList = []
  140. for child in childList:
  141.     #print('before before')
  142.     t = threading.Thread(target=child.dataLoop)
  143.     t.daemon = True
  144.     threadList.append(t)
  145.     #print('before')
  146.     t.start()
  147.     #print('after')
  148.  
  149. print('threads started')
  150.  
  151. #for child in childList:
  152. #    child.ping()
  153. #    print('trying to ping')
  154.  
  155. def signal_handler(signal,frame):
  156.     print('Ctrl-C pressed')
  157.     for child in childList:
  158.         child.logout()
  159.     print('logged out of children')
  160.     sys.exit(0)
  161. signal.signal(signal.SIGINT, signal_handler)
  162. print('after signal things')
  163. signal.pause()
  164. #for child in childList:
  165.     #child.dataLoop() #start dataloop threads for all children
  166.  
  167. #while(True):
  168. #    for child in childList:
  169. #        child.dataLoop() #will start timers in refreshData class that loop back to themselves
  170. #    time.sleep(1)
RAW Paste Data
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy. OK, I Understand
 
Top