python通过ssh-powershell监控windows

#! /usr/bin/env python
-- coding: utf-8 --
Author:pako
Email:zealzpc@gmail.com
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()</pre>