Make change backwards-compatible

astrofrog · Oct 22, 2014 · 82da9aa · 82da9aa
1 parent 5b019c5
commit 82da9aa
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/psrecord/main.py b/psrecord/main.py
@@ -117,8 +117,13 @@ def monitor(pid, logfile=None, plot=None, duration=None, interval=None, include_
         # Find current time
         current_time = time.time()
 
+        try:
+            pr_status = pr.status()
+        except TypeError:  # psutil < 2.0
+            pr_status = pr.status
+
         # Check if process status indicates we should exit
-        if pr.status() in [psutil.STATUS_ZOMBIE, psutil.STATUS_DEAD]:
+        if pr_status in [psutil.STATUS_ZOMBIE, psutil.STATUS_DEAD]:
             print("Process finished ({:.2} seconds)"
                   .format(current_time - start_time))
             break