Fallback for destroyed processes

analysiscenter · Jul 22, 2024 · 8aaca70 · 8aaca70
1 parent 819668b
commit 8aaca70
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/nbtools/nbstat/resource_entry.py b/nbtools/nbstat/resource_entry.py
@@ -1,5 +1,6 @@
 """ ResourceEntry -- a dict-like class to hold all properties (Resources) of an entry. """
 from datetime import datetime
+import psutil
 
 from .resource import Resource
 from .utils import format_memory
@@ -75,7 +76,10 @@ def to_format_data(self, resource, terminal, **kwargs):
         elif resource == Resource.CPU:
             process = self[Resource.PROCESS]
             if process is not None:
-                data = process.cpu_percent()
+                try:
+                    data = process.cpu_percent()
+                except psutil.NoSuchProcess:
+                    data = 0.0
                 data = round(data)
 
                 style = terminal.bold if data > 30 else ''