ps fix

src/client.py

@@ -10,7 +10,11 @@ import psutil
 import time
 import time
 import re
 import re
 
 
+logging.basicConfig()
 logger = logging.getLogger(__name__)
 logger = logging.getLogger(__name__)
+level = os.environ.get('LOGLEVEL', 'INFO')
+logger.setLevel(level)
+
 
 
 
 
 class Client:
 class Client:
@@ -153,8 +157,8 @@ class Client:
         now = time.time()
         now = time.time()
         running_vms = []
         running_vms = []
 
 
-        for entry in psutil.get_process_list():
-            try:
+        try:
+            for entry in psutil.get_process_list():
                 if entry.name() in ('kvm', 'qemu-system-x86_64'):
                 if entry.name() in ('kvm', 'qemu-system-x86_64'):
                     parser = argparse.ArgumentParser()
                     parser = argparse.ArgumentParser()
                     parser.add_argument('-name')
                     parser.add_argument('-name')
@@ -162,31 +166,35 @@ class Client:
                     args, unknown = parser.parse_known_args(
                     args, unknown = parser.parse_known_args(
                         entry.cmdline()[1:])
                         entry.cmdline()[1:])
                     # for Red Hat style parametering of kvm
                     # for Red Hat style parametering of kvm
-                    args.name = re.sub(r"^guest=", "", args.name)
-                    args.name = re.sub(r",debug-threads=.*$", "", args.name)
+                    if isinstance(args.name, basestring):
+                        args.name = re.sub(r"^guest=", "", args.name)
+                        args.name = re.sub(r",debug-threads=.*$", "", args.name)
 
 
-                    process = self.processes.get(entry.pid, None)
-                    if not process or process.cmdline() != entry.cmdline():
-                        process = psutil.Process(entry.pid)
-                        logger.info('New process: %s', process)
-                        self.processes[entry.pid] = process
+                        process = self.processes.get(entry.pid, None)
+                    	if not process or process.cmdline() != entry.cmdline():
+                        	process = psutil.Process(entry.pid)
+                        	logger.info('New process: %s', process)
+                        	self.processes[entry.pid] = process
 
 
-                    mem_perc = (float(process.get_memory_info().rss)
+                    	mem_perc = (float(process.get_memory_info().rss)
                                 / (args.memory_size * 1024 ** 2) * 90)
                                 / (args.memory_size * 1024 ** 2) * 90)
-                    metrics.append('vm.%(name)s.memory.usage %(value)f '
+                    	metrics.append('vm.%(name)s.memory.usage %(value)f '
                                    '%(time)d' % {'name': args.name,
                                    '%(time)d' % {'name': args.name,
                                                  'value': mem_perc,
                                                  'value': mem_perc,
                                                  'time': now})
                                                  'time': now})
-                    cpu_perc = process.get_cpu_percent()
-                    metrics.append('vm.%(name)s.cpu.percent %(value)f '
+                    	cpu_perc = process.get_cpu_percent()
+                    	metrics.append('vm.%(name)s.cpu.percent %(value)f '
                                    '%(time)d' % {'name': args.name,
                                    '%(time)d' % {'name': args.name,
                                                  'value': cpu_perc,
                                                  'value': cpu_perc,
                                                  'time': now})
                                                  'time': now})
-                    running_vms.append(args.name)
-            except psutil.NoSuchProcess:
-                logger.warning('Process %d lost.', entry.pid)
-            except TypeError:
-                logger.warning('ParseError %s.', entry.cmdline())
+                        running_vms.append(args.name)
+                    else: 
+                        logger.warning('Pid: %d args.name is empty?: %s %s', entry.pid, entry.name(), entry.cmdline())
+
+        except psutil.NoSuchProcess:
+            logger.warning('Process %d lost.', entry.pid)
+        except TypeError:
+            logger.warning('ParseError %s.', entry.cmdline())
 
 
         interfaces = psutil.network_io_counters(pernic=True)
         interfaces = psutil.network_io_counters(pernic=True)
         for interface, data in interfaces.iteritems():
         for interface, data in interfaces.iteritems():