python通过ssh-powershell监控windows的方法

 #! /usr/bin/env python # -*- coding: utf-8 -*- import re from pexpect import * class ssh_win32: def __init__(self, user, host, password=None,systemroot='c',papath='',timeout=5,verbose=0): self.user = user#监控机器的username self.host = host#监控机器的ip self.verbose = verbose self.password = password#密码 self.timeout=timeout#执行命令的timeout self.systemroot=systemroot#windows 所安装的盘符 if not papath:#powershell.exe的路径 self.powershell_path=self.systemroot+':/WINDOWS/system32/WindowsPowerShell/v1.0/powershell.exe ' self.key = [ 'authenticity', 'assword:', '@@@@@@@@@@@@', 'Command not found.', EOF, ] self.f = open('ssh.out','w') def ssh(self,command): cmd='ssh -l %s %s %s'%(self.user,self.host,command) print "cmd:",cmd con=spawn(cmd,timeout=self.timeout) seen=con.expect(self.key) if seen == 0: con.sendline('yes') seen = con.expect(self.key) if seen == 1: #    if not self.password: #      self.password = getpass.getpass('Remote password: ') con.sendline(self.password) try: res=con.read() except Exception ,e: res=con.before #      print "res:",res return res def ssh_disk(self): cmd=self.powershell_path+"Get-WmiObject win32_logicaldisk" res=self.ssh(cmd) disk={} if res: res=res.split('No such file or directory')[-1].replace('\r','').split('\n') res=[c for c in res if c] #      print 'res:',res predisk='C' for d in res: #      print d key,value=d.split(':',1) #      print d #      print 'key:',key,'value:',value key=key.strip() value=value.strip() if key=='DeviceID' and value not in disk.keys(): predisk=value disk[predisk]={} disk[predisk][key]=value else: if key in ['FreeSpace','Size']: if value: value=int(value)/1024/1024/1024 disk[predisk][key]=value for d in disk.keys(): if disk[d]['DriveType']!='3': disk.pop(d) #    print 'disk:',disk return disk def ssh_cpu(self): cmd=self.powershell_path+'gwmi -computername localhost win32_Processor' res=self.ssh(cmd) res=res.split('No such file or directory')[-1].replace('\r','').split('\n') res=[r for r in res if r] #    print res cpu={} for i in res: #      print '='*10 #      print i i=i.split(':') #  print i if len(i)==2: key,value=i else: continue key=key.strip() value=value.strip() #      print 'key:',key #      print 'value:',value cpu[key]=value return cpu def ssh_memory(self): totalmem=self.powershell_path+'Get-WmiObject win32_OperatingSystem TotalVisibleMemorySize' freemem=self.powershell_path+'Get-WmiObject win32_OperatingSystem FreePhysicalMemory' memory={} for cmd in [totalmem,freemem]: res=self.ssh(cmd) if 'Win32_OperatingSystem' in res: res=res=res.replace('\r','').split('\n') res=[m for m in res if m][-1] print 'res:',res key,value=res.split(':') key=key.strip() value=value.strip() memory[key]=value else: print "not return data" return None return memory def ssh_ping(self,host): cmd='ping -n 1 %s'%host patt=r'.+?(\d*)% loss.*' res=self.ssh(cmd).replace('\r','').replace('\n','') print res m=re.match(patt,res) if m: lost_percent=m.group(1) print 'lost_percent:',lost_percent return int(lost_percent) else: return None def ssh_ps(self): cmd=self.powershell_path+'ps' res=self.ssh(cmd) ps=[] if '-- -----------' in res: res=res.replace('\r','').split('-- -----------')[-1].split('\n') res=[d for d in res if d.strip()] for p in res: process={} row=[para for para in p.split(' ') if para.strip()] process['handles']=row[0] process['npm']=row[1] process['pm']=row[2] process['ws']=row[3] process['vm']=row[4] process['cpu']=row[5] process['id']=row[6] process['process_name']=row[-1] ps.append(process) #      print ps return ps else: return None def ssh_netstat(self): cmd='netstat -ao' res=self.ssh(cmd) netstat=[] if 'PID' in res: res=res.replace('\r','').split('PID')[-1].split('\n') res=[d for d in res if d.strip()] for p in res: process={} row=[para for para in p.split(' ') if para.strip()] process['proto']=row[0] process['local_address']=row[1] process['foreign_address']=row[2] process['state']=row[3] process['pid']=row[-1] netstat.append(process) #      print netstat return netstat else: return None if __name__ == "__main__": cmd="c:/WINDOWS/system32/WindowsPowerShell/v1.0/powershell.exe ps" user='admin' host='192.168.123.105' password='123456' ssh=ssh_win32(user,host,password,systemroot='c',timeout=5) #  print ssh.ssh_cpu() #  print "\n\n\n\n" #  print ssh.ssh_disk() #  print "\n\n\n\n" #  print ssh.ssh_memory() #  print ssh.ssh_ping(host) #  print ssh.ssh_ps() #  print ssh.ssh_netstat()