Fixed missing import

[invirt/packages/invirt-remote.git] / host / usr / sbin / invirt-availability

diff --git a/host/usr/sbin/invirt-availability b/host/usr/sbin/invirt-availability
index 93ef381..26ef5c9 100644 (file)
--- a/host/usr/sbin/invirt-availability
+++ b/host/usr/sbin/invirt-availability
@@ -6,6 +6,7 @@ Gathers availability data for the VM host it's running on.
 from subprocess import PIPE, Popen, call
 import sys
 import yaml
+import os
 
 # return the amount of memory in kilobytes represented by s
 def parseUnits(s):
@@ -19,7 +20,16 @@ def main(argv):
     The numbers in /proc/xen/balloon have nice units
     All math is done in kilobytes for consistency
     Output is in MB
+
+    Bail if /etc/invirt/nocreate exists
     """
+    try:
+        os.stat('/etc/invirt/nocreate')
+        print 0
+        return 0
+    except OSError:
+        pass
+
     p = Popen(['/usr/sbin/xm', 'info'], stdout=PIPE)
     output = p.communicate()[0]
     if p.returncode != 0: