1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#!/usr/bin/env python
#-*-coding:utf8-*-
"""
@Author :  Villiam Sheng
@Group  :  Linux Group
@Date   :  2011-07-18
@Funtion:
             Update kvm host status ...
             get_nic: Get a week network flow
                 1,Get Seven days before flow,get maximum value!
                 2,Get Seven days before flow,get average value!
  
"""
import  os,sys,libvirt,socket,shutil,re
from   statvfs  import  F_BLOCKS,F_BAVAIL,F_BSIZE
  
class  kvm_os_status( object ):
     def  __init__( self ):
         self .vmm  =  {}
         try :
             self .conn  =  libvirt. open ( None )
         except  libvirt.libvirtError,e:
             print  e
  
     def  get_mem( self ):
         try :
             f  =  open ( '/proc/meminfo' , 'r' )
             for  i  in  f.readlines():
                 if  i.find( 'MemTotal:' ) ! =  - 1 :
                     total_mem = int (i.split( ':' )[ 1 ].split( 'kB' )[ 0 ]) / 1024
                     continue
             try :
                 exec_command  =  """ grep "memory" /data*/domains/*/*.xml 2>/dev/null |awk -F "<memory>" '{print $2}'|awk -F "</memory>" '{print $1}' """
                 mem  =  os.popen(exec_command).readlines()
                 act_mem  =  0
                 for  m  in  mem:
                     act_mem  + =  int (m)  /  1024
                 self .vmm[ 'free_mem' ]  =  total_mem  -  act_mem
             except  Exception,e:
                 pass
         except  Exception,e:
             pass
  
     def  get_mip( self ):
         try :
             exec_command  =  """cat /etc/sysconfig/network-scripts/ifcfg-br0 |grep "IPADDR"|awk -F"=" '{print $2}' 2>/dev/null"""
             mip  =  os.popen(exec_command).read().strip()
             sock = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
             try :
                 sock.connect((mip, 0 ))
             except  socket.error,e:
                 exec_command  =  """cat /etc/sysconfig/network-scripts/ifcfg-br1 |grep "IPADDR"|awk -F"=" '{print $2}'"""
                 nip  =  os.popen(exec_command).read().strip()
                 self .vmm[ 'mip' ]  =  nip
             self .vmm[ 'mip' ]  =  sock.getsockname()[ 0 ]
         except :
             pass
  
  
     def  get_disk( self ):
         extends  =  []
         disk = {}
         try :
             ext_disk  =  os.popen( "df -h |awk '{print $1,$6}'|grep -v 'tmpfs'|grep -v 'Filesystem'|grep -v 'sda1'|grep -v 'mfs'|grep -v 'T'" ).readlines()
         except  Exception,e:
             print  e
         if  ext_disk  = =  "":
             self .vmm[ 'free_disk' ]  =  vfs[F_BLOCKS] * vfs[F_BSIZE] / 1024 / 1024 / 1024
         else :
             free_disk  =  0
             for  disk  in  ext_disk:
                 try :
                     vfs  =  os.statvfs(disk.split()[ 1 ])
                 except  Exception,e:
                     print  e
                 full_space  =  vfs[F_BLOCKS] * vfs[F_BSIZE] / 1024 / 1024 / 1024
                 free_space  =  vfs[F_BAVAIL] * vfs[F_BSIZE] / 1024 / 1024 / 1024
                 imgs  =  os.popen( "ls -ls %s/domains/vm*/data*.img 2>/dev/null | awk '{print $1,$6}'"  %  disk.split()[ 1 ]).readlines()
                 if  imgs:
                     for  i  in  imgs:
                         t_size, f_size  =  i.strip().split()
                         free_space  - =  int (f_size) / 1024 / 1024 / 1024
                         if  int (t_size) ! =  0 :
                             free_space  + =  int (t_size) / 1024 / 1024
                 disk = {disk.split()[ 1 ]:free_space}
                 for  i  in  disk.keys():
                     free_disk  + =  disk[i]
             self .vmm[ 'free_disk' ]  =  free_disk
             a  =  os.popen( "cat /etc/issue|awk '{print $7}'|grep -v 'Kernel'|grep -v '^$'" ).readline()
             self .vmm[ 'os_type' ]  =  'RHEL%sx64' % a.strip()
  
     def  count( self ):
         self .vmm[ 'vmm_count' ]  =  0
         try :
             for  id  in  self .conn.listDomainsID():
                 dom  =  self .conn.lookupByID( id )
                 self .vmm[ 'vmm_count' ]  =  self .vmm[ 'vmm_count' ]  +  1
         except  libvirt.libvirtError,e:
             pass
         print  self .vmm
     def  work( self ):
         self .get_mem()
         self .get_disk()
         self .get_mip()
         self .count()
  
if  __name__  = =  "__main__" :
     st  =  kvm_os_status()
     st.work()