host/usr/sbin/invirt-availability

   1 #!/usr/bin/env python
   2 """
   3 Gathers availability data for the VM host it's running on.
   4 """
   5
   6 from subprocess import PIPE, Popen, call
   7 import sys
   8 import yaml
   9
  10 # return the amount of memory in kilobytes represented by s
  11 def parseUnits(s):
  12     num, unit = s.split(' ')
  13     return int(num) * {'kb':1, 'mb':1024}[unit.lower()]
  14
  15 def main(argv):
  16     p = Popen(['/usr/sbin/xm', 'info'], stdout=PIPE)
  17     output = p.communicate()[0]
  18     if p.returncode != 0:
  19         raise RuntimeError("Command '%s' returned non-zero exit status %d"
  20                            % ('invirt-availability', p.returncode))
  21     xminfo = yaml.load(output, yaml.CSafeLoader)
  22
  23     # In kilobytes
  24     free_memory = int(xminfo['free_memory']) * 1024
  25
  26     f = open('/proc/xen/balloon', 'r')
  27     ballooninfo = yaml.load(f.read())
  28     f.close()
  29     currentallocation = parseUnits(ballooninfo['Current allocation'])
  30     minimumtarget = parseUnits(ballooninfo['Minimum target'])
  31
  32     p = Popen(['/usr/sbin/xm', 'info', '-c'], stdout=PIPE)
  33     output = p.communicate()[0]
  34     if p.returncode != 0:
  35         raise RuntimeError("Command '%s' returned non-zero exit status %d"
  36                            % ('invirt-availability', p.returncode))
  37     xminfoc = yaml.load(output, yaml.CSafeLoader)
  38
  39     # In kilobytes
  40     dom0minmem = int(xminfoc['dom0-min-mem']) * 1024
  41
  42     dom0_spare_memory = currentallocation - max(minimumtarget, dom0minmem)
  43     print int((free_memory + dom0_spare_memory)/1024)
  44     return 0
  45
  46 if __name__ == '__main__':
  47     sys.exit(main(sys.argv))