tuna-cmd: Catch IOError in --show_threads

for cases such as 'tuna -P | head'

Signed-off-by: Arnaldo Carvalho de Melo <acme@redhat.com>

1 files changed, 7 insertions, 3 deletions

diff --git a/tuna-cmd.py b/tuna-cmd.py
index 408808d..2c4b566 100755
--- a/tuna-cmd.py
+++ b/tuna-cmd.py
@@ -156,9 +156,13 @@ def ps(threads, cpus, show_uthreads, show_kthreads, affect_children):
 	cpuinfo = procfs.cpuinfo()
 	irqs = procfs.interrupts()
 	has_ctxt_switch_info = ps[1]["status"].has_key("voluntary_ctxt_switches")
-	ps_show_header(has_ctxt_switch_info)
-	ps_show(ps, affect_children, cpuinfo, irqs, threads, cpus,
-		show_uthreads, show_kthreads, has_ctxt_switch_info)
+	try:
+		ps_show_header(has_ctxt_switch_info)
+		ps_show(ps, affect_children, cpuinfo, irqs, threads, cpus,
+			show_uthreads, show_kthreads, has_ctxt_switch_info)
+	except IOError:
+		# 'tuna -P | head' for instance
+		pass
 
 def main():
 	try: