python通过ssh-powershell监控windows的方法
admin
2023-07-31 02:29:14
0

本文实例讲述了python通过ssh-powershell监控windows的方法。分享给大家供大家参考。具体分析如下:

对于服务器的监控来说,监控linux不管是自己动手写脚本还是用一些开源的工具比如nagios,zenoss什么的。但毕竟还是有些公司有windows做服务器的,相对linux来说,windows没有方便的shell,cmd下提供的命令对于监控来说远远没有linux方便。但是现在windows上如果安装了powershell(win7,2008自带),就比以前方便多了,linux上的命令基本都能在powershell里执行,比如查看进程还是ps.

自己封装了一个python通过ssh(通过pexpect模块)调用powershell的脚本,里面包快ps,netstat,ping检测,查看硬盘,cpu信息和负载,内存信息。通过创建ssh_win32类对象,然后调用它的方法,返回的都是解析好的python对象。

ssh_powershell.py:

#! /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()

希望本文所述对大家的Python程序设计有所帮助。

相关内容

热门资讯

500 行 Python 代码... 语法分析器描述了一个句子的语法结构,用来帮助其他的应用进行推理。自然语言引入了很多意外的歧义,以我们...
定时清理删除C:\Progra... C:\Program Files (x86)下面很多scoped_dir开头的文件夹 写个批处理 定...
65536是2的几次方 计算2... 65536是2的16次方:65536=2⁶ 65536是256的2次方:65536=256 6553...
Mobi、epub格式电子书如... 在wps里全局设置里有一个文件关联,打开,勾选电子书文件选项就可以了。
scoped_dir32_70... 一台虚拟机C盘总是莫名奇妙的空间用完,导致很多软件没法再运行。经过仔细检查发现是C:\Program...
pycparser 是一个用... `pycparser` 是一个用 Python 编写的 C 语言解析器。它可以用来解析 C 代码并构...
小程序支付时提示:appid和... [Q]小程序支付时提示:appid和mch_id不匹配 [A]小程序和微信支付没有进行关联,访问“小...
Prometheus+Graf... 一,Prometheus概述 1,什么是Prometheus?Prometheus是最初在Sound...
python绘图库Matplo... 本文简单介绍了Python绘图库Matplotlib的安装,简介如下: matplotlib是pyt...
微信小程序使用slider实现... 众所周知哈,微信小程序里面的音频播放是没有进度条的,但最近有个项目呢,客户要求音频要有进度条控制,所...