Be more robust if child processes no longer exist

astrofrog · Apr 25, 2024 · 47ffb26 · 47ffb26
1 parent 975784f
commit 47ffb26
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/psrecord/main.py b/psrecord/main.py
@@ -207,16 +207,16 @@ def monitor(pid, logfile=None, plot=None, duration=None, interval=None,
                     try:
                         current_cpu += get_percent(child)
                         current_mem = get_memory(child)
+                        current_mem_real += current_mem.rss / 1024. ** 2
+                        current_mem_virtual += current_mem.vms / 1024. ** 2
+                        if include_io:
+                            counters = child.io_counters()
+                            read_count += counters.read_count
+                            write_count += counters.write_count
+                            read_bytes += counters.read_bytes
+                            write_bytes += counters.write_bytes
                     except Exception:
                         continue
-                    current_mem_real += current_mem.rss / 1024. ** 2
-                    current_mem_virtual += current_mem.vms / 1024. ** 2
-                    if include_io:
-                        counters = child.io_counters()
-                        read_count += counters.read_count
-                        write_count += counters.write_count
-                        read_bytes += counters.read_bytes
-                        write_bytes += counters.write_bytes
 
 
             if logfile: